from pydantic import BaseModel, HttpUrl


class DownloadRequest(BaseModel):
    url: HttpUrl
    quality: int


class DownloadResponse(BaseModel):
    video: str
    audio: str


class DownloadResponseV2(BaseModel):
    video: str


class TaskCreateResponse(BaseModel):
    task_id: str
    status: str


class TaskStatusResponse(TaskCreateResponse):
    result: DownloadResponse | None = None