Update app.py

app.py

@@ -472,6 +472,15 @@ async def test_download(request: Request):
     response = extract_video_info(video_url)
     return response
 
+
+async def get_audio_download_url(track_id: str, quality: str) -> str:
+    youtube_url = f'https://www.youtube.com/watch?v={track_id}'
+    if quality == 'mp3':
+        type = 'audio'
+    else:
+        type = 'video'
+    donwnload_url = f'https://chrunos-shadl.hf.space/yt/dl?url={youtube_url}&type={type}&quality={quality}'
+    return donwnload_url
     
 
 @app.post("/maxs")
@@ -531,7 +540,7 @@ async def download_high_quality_video(request: Request):
     }
 
     if is_youtube_url:
-        dl_url = await get_track_download_url(video_url)
+        dl_url = await get_audio_download_url(video_url)
         if dl_url and "http" in dl_url:
             return {"url": dl_url, "requests_remaining": rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)}
         else:
@@ -578,13 +587,13 @@ async def verify_api_key(api_key: str = Security(api_key_header)):
 
 @app.post("/audio")
 async def download_audio(
-    request: Request,
-    api_key: str = Security(verify_api_key)
+    request: Request
+    #api_key: str = Security(verify_api_key)
 ):
     data = await request.json()
     video_url = data.get('url')
-    cookiefile = "firefox-cookies.txt"
-    env_to_cookies_from_env("firefox-cookies.txt")
+    #cookiefile = "firefox-cookies.txt"
+    #env_to_cookies_from_env("firefox-cookies.txt")
 
     timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
     output_template = str(Path(global_download_dir) / f'%(title).70s_{timestamp}.%(ext)s')
@@ -595,7 +604,7 @@ async def download_audio(
         'quiet': True,
         'no_warnings': True,
         'noprogress': True,
-        'cookiefile': cookiefile,
+        #'cookiefile': cookiefile,
         'postprocessors': [{
             'key': 'FFmpegExtractAudio',
             'preferredcodec': 'mp3',
@@ -603,18 +612,26 @@ async def download_audio(
         }]
         
     }
-
-    await run_in_threadpool(lambda: yt_dlp.YoutubeDL(ydl_opts).download([video_url]))
+    if is_youtube_url:
+        dl_url = await get_audio_download_url(video_url)
+        if dl_url and "http" in dl_url:
+            return {"url": dl_url, "requests_remaining": rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)}
+        else:
+            return {
+                "error": "Failed to Fetch the video."
+                }
+    else:
+        await run_in_threadpool(lambda: yt_dlp.YoutubeDL(ydl_opts).download([video_url]))
     
-    downloaded_files = list(Path(global_download_dir).glob(f"*_{timestamp}.*"))
-    if not downloaded_files:
-        return {"error": "Download failed"}
-
-    downloaded_file = downloaded_files[0]
-    encoded_filename = urllib.parse.quote(downloaded_file.name)
-    download_url = f"{BASE_URL}/file/{encoded_filename}"
-    gc.collect()
-    return {"url": download_url}
+        downloaded_files = list(Path(global_download_dir).glob(f"*_{timestamp}.*"))
+        if not downloaded_files:
+            return {"error": "Download failed"}
+    
+        downloaded_file = downloaded_files[0]
+        encoded_filename = urllib.parse.quote(downloaded_file.name)
+        download_url = f"{BASE_URL}/file/{encoded_filename}"
+        gc.collect()
+        return {"url": download_url}
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)