from pydantic import BaseModel, Field


class ApiImageGenerateRequest(BaseModel):
    """图片生成请求。支持全量 Volcano Ark SDK 参数。"""

    model: str = Field(..., description="模型名称, 如 doubao-seedream-5-0-260128")
    prompt: str = Field(..., description="图片描述提示词")
    size: str | None = Field("2K", description="图片尺寸: 2K | 4K 或 2048x2048")
    response_format: str | None = Field("url", description="返回格式: url | b64_json")
    watermark: bool | None = Field(False, description="是否添加水印")
    image: list[str] | None = Field(None, description="参考图片URL列表")
    output_format: str | None = Field(None, description="输出格式: jpeg | png | webp")
    sequential_image_generation: str | None = Field(
        None, description="组图模式: auto 开启"
    )
    generation_count: int | None = Field(1, ge=1, le=5, description="生成数量: 1-5")


class ApiImageGenerateDataItem(BaseModel):
    """单张图片结果。"""

    url: str | None = None
    b64_json: str | None = None
    size: str | None = None
    output_format: str | None = None


class ApiImageGenerateResponse(BaseModel):
    """图片生成响应（同步返回）。"""

    created: int
    data: list[ApiImageGenerateDataItem]
    model: str