from pydantic import BaseModel


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


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