from typing import Optional

from pydantic import BaseModel


class VideoGenerationRequest(BaseModel):
    prompt: str = ""
    negative_prompt: Optional[str] = ""
    image_asset_id: str  # ID ассета-картинки для source image
    duration: int = 5  # 5 or 10 seconds
    mode: str = "std"  # "std" or "pro"
    model_name: str = "kling-v2-1"
    cfg_scale: float = 0.5
    aspect_ratio: str = "16:9"
    linked_character_id: Optional[str] = None
    project_id: Optional[str] = None