# webapp/packages/api/user-service/models/demo.py
# webapp/packages/api/user-service/models/demo.py
from pydantic import BaseModel, Field
from pydantic.config import ConfigDict
from typing import Dict, Any, List, Optional
from .chat import ProviderConfig
from .agent import DeployedApi
import uuid
from datetime import datetime
from time_utils import naive_utc_now

class GenerateDemoCodeRequest(BaseModel):
    user_prompt: str = Field(..., alias="userPrompt")
    selected_apis: List[DeployedApi] = Field(..., alias="selectedApis")
    composer_model_config: ProviderConfig = Field(..., alias="modelConfig")
    built_in_tools: Optional[List[str]] = Field(default_factory=list, alias="builtInTools")
    model_config = ConfigDict(populate_by_name=True)

class GenerateDemoCodeResponse(BaseModel):
    html: str = ""
    css: str = ""
    js: str = ""
    thoughts: Optional[Any] = None


class CreateDemoAppRequest(BaseModel):
    name: str
    description: Optional[str] = None
    selected_apis: List[DeployedApi] = Field(..., alias="selectedApis")
    composer_model_config: ProviderConfig = Field(..., alias="modelConfig")
    user_prompt: str = Field(..., alias="userPrompt")
    generated_code: GenerateDemoCodeResponse = Field(..., alias="generatedCode")
    composer_thoughts: Optional[Any] = Field(None, alias="composerThoughts")
    model_config = ConfigDict(populate_by_name=True)


class DemoApp(CreateDemoAppRequest):
    id: str = Field(default_factory=lambda: str(uuid.uuid4()), alias="_id")
    rev: Optional[str] = Field(None, alias="_rev")
    created_at: datetime = Field(default_factory=naive_utc_now)
    updated_at: datetime = Field(default_factory=naive_utc_now)
    model_config = ConfigDict(populate_by_name=True)