from pydantic import BaseModel, HttpUrl


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


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


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


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