from typing import Optional

from pydantic import BaseModel, Field


class KlingRequest(BaseModel):
    image: str = Field(..., description='Base64-encoded image')
    prompt: Optional[str] = None
    duration: Optional[str] = Field(None, description='5 or 10 seconds')
    aspect_ratio: Optional[str] = None


class MinimaxRequest(BaseModel):
    prompt: str = Field(..., min_length=1, max_length=4000)
    first_frame_image: Optional[str] = Field(None, description='Base64-encoded image')
    subject_reference: Optional[str] = Field(None, description='Base64-encoded reference image')


class SeedanceRequest(BaseModel):
    prompt: str = Field(..., min_length=1, max_length=4000)
    image: Optional[str] = Field(None, description='Base64-encoded image')
    duration: Optional[str] = None
    resolution: Optional[str] = None