from pydantic import BaseModel
from typing import List, Optional

class ShotParsing(BaseModel):
    shot_number: str
    description: str
    visual_notes: Optional[str] = None
    dialogue: Optional[str] = None

class SceneParsing(BaseModel):
    scene_number: str
    heading: str
    description: str
    shots: List[ShotParsing] = []

class ScriptAnalysisResponse(BaseModel):
    scenes: List[SceneParsing]