from pydantic import BaseModel, Field

from app.enums.generation_status import GenerationType
from app.schemas.common import NaiveDatetime, NaiveDatetimeOptional


class OptimizeParams(BaseModel):
    project_id: str
    prompt: str = Field(..., max_length=500)
    gen_type: GenerationType = Field(GenerationType.video, description="生成类型：video-视频，image-图片")
    engine_id: str = Field(..., min_length=1, max_length=32, description="提词阶段选定并冻结的生成引擎ID")
    include_media_references: bool = Field(False, description="资源生成时是否携带本次提词附件；提词完成后不可修改")
    duration: int | None = Field(None, description="视频时长（秒），视频生成必填")
    aspect_ratio: str | None = Field(None, description="视频比例，视频生成必填")
    resolution: str | None = Field(None, description="视频目标分辨率，视频生成必填")
    image_size: str | None = Field(None, description="画面分辨率，图片生成使用")
    image_proportion: str | None = Field(None, description="图片比例，图片生成使用")
    image_px: str | None = Field(None, description="图片像素大小，图片生成使用")
    references: list[dict] | None = None
    idempotency_key: str | None = Field(None, max_length=64, description="幂等键，防止重复请求")


class OptimizeResult(BaseModel):
    optimized_prompt: str
    text_credits_cost: float
    text_tokens_used: int = 0
    record: "GenerationRecordOut"

class GenerationRecordOut(BaseModel):
    id: str
    project_id: str
    project_name: str
    original_prompt: str
    optimized_prompt: str | None = None
    gen_type: str = "video"
    duration: int | None = None
    aspect_ratio: str | None = None
    resolution: str | None = None
    image_size: str | None = None
    image_proportion: str | None = None
    image_px: str | None = None
    status: str
    pipeline_stage: str | None = None
    video_upscale_enabled: bool = False
    video_url: str | None = None
    video_cover_url: str | None = None
    image_url: str | None = None
    references: list[dict] | None = None
    engine_id: str | None = None
    engine_name: str | None = None
    engine_snapshot: dict | None = None
    include_media_references: bool = False
    config_complete: bool = False
    config_recoverable: bool = False
    config_fallback_hint: str | None = None
    can_generate: bool = False
    can_retry: bool = False
    should_poll: bool = False
    client_status: str = "ready"
    operation_phase: str = "prompt"
    text_credits_cost: float = 0.0
    text_tokens_used: int = 0
    credits_cost: float = 0.0
    video_tokens_used: int = 0
    image_tokens_used: int = 0
    error_message: str | None = None
    created_at: NaiveDatetime
    generated_at: NaiveDatetimeOptional = None

    model_config = {"from_attributes": True}


class GenerationRecordPageListOut(BaseModel):
    """生成记录分页返回API。"""
    page: int = Field(..., description="当前日期分组分页页码")
    page_size: int = Field(..., description="当前每页返回的日期分组数量")
    total: int = Field(..., description="当前生成日期下的生成成功记录总数")
    items: list[GenerationRecordOut] = Field(default_factory=list,)

class UpdatePromptRequest(BaseModel):
    optimized_prompt: str = Field(..., max_length=2000)


OptimizeResult.model_rebuild()