from pydantic import BaseModel
from typing import Any, List, Literal
from .base import IDModel, TimestampModel

class PodcastBase(BaseModel):
    title: str
    podcast_transcript: List[Any]
    file_location: str = ""
    search_space_id: int

class PodcastCreate(PodcastBase):
    pass

class PodcastUpdate(PodcastBase):
    pass

class PodcastRead(PodcastBase, IDModel, TimestampModel):
    class Config:
        from_attributes = True

class PodcastGenerateRequest(BaseModel):
    type: Literal["DOCUMENT", "CHAT"]
    ids: List[int]
    search_space_id: int
    podcast_title: str = "SurfSense Podcast"