from pydantic import BaseModel, Field, ConfigDict
from typing import List

class AudioShot(BaseModel):
    """Audio design for a shot"""
    model_config = ConfigDict(extra='allow')
    
    shot_id: int
    audio_prompt: str = Field(description="SFX: [音效]; BGM: [配乐]; Ambience: [环境音]")


class AudioList(BaseModel):
    model_config = ConfigDict(extra='allow')
    
    shots: List[AudioShot]
