from typing import Optional

from pydantic import BaseModel, Field

from .config import DEFAULT_CFG, DEFAULT_STEPS


class GenerationItem(BaseModel):
    id: str
    created_at: str = Field(..., description="ISO8601 timestamp")
    transcript: str
    prompt: str
    image_url: str
    steps: int
    cfg: float
    seed: Optional[int] = None


class HistoryPayload(BaseModel):
    items: list[GenerationItem]


class DeleteImageResponse(BaseModel):
    ok: bool = True
    id: str
    image_deleted: bool


class InternalGenerateRequest(BaseModel):
    token: str
    raw_audio_path: str
    steps: int = DEFAULT_STEPS
    cfg: float = DEFAULT_CFG
    seed: Optional[int] = None