from pydantic import BaseModel
from typing import Optional


class GenerateRequest(BaseModel):
    prompt: str
    video_strength: float
    seed_value: Optional[int] = None
    scale_status: bool = False
    rife_status: bool = False