from fastapi import APIRouter
from app.schemas import TaskCreateResponse
from app.schemas import DownloadRequest
from app.tasks import download_youtube

router = APIRouter()


@router.post("/", response_model=TaskCreateResponse)
async def download_video(data: DownloadRequest):
    task = download_youtube.delay(data.url, data.quality)
    return TaskCreateResponse(task_id=task.id, status=task.status)