diff --git a/app/api/v2/endpoints/youtube.py b/app/api/v2/endpoints/youtube.py index 7f44a38..cde2255 100644 --- a/app/api/v2/endpoints/youtube.py +++ b/app/api/v2/endpoints/youtube.py @@ -18,5 +18,5 @@ async def video_resolutions(url: str): return {"resolutions": YouTubeService.resolutions(url)} @router.get("/size") -async def video_size(data: DownloadRequest): +async def video_size(data: DownloadRequest) -> dict[str, int]: return {"size": YouTubeService.filesize(data)} diff --git a/app/services.py b/app/services.py index 30dc848..d84bc8e 100644 --- a/app/services.py +++ b/app/services.py @@ -46,8 +46,7 @@ class YouTubeService: manager = YtDlpManager((data.url)) video_size = manager.best_video(data.quality).filesize audio_size = manager.best_audio().filesize - return {"filesize": video_size + audio_size} - + return video_size + audio_size class Files: @staticmethod