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):
    print("Get Request. Try to set task")
    task = download_youtube.delay(url=str(data.url), quality=data.quality)
    print("Set task. Return task response")
    return TaskCreateResponse(task_id=task.id, status=task.status)