Update app.py

app.py

@@ -11,6 +11,8 @@ from pydantic import BaseModel
 from urllib.parse import urlparse, parse_qs
 from collections import defaultdict
 import threading
+import aiohttp
+
 
 app = Flask(__name__)
 ytmusic = YTMusic()
@@ -212,7 +214,22 @@ async def get_audio_download_url(track_id: str, quality: str) -> str:
     return donwnload_url
     
     
-
+async def get_download_url(track_id, quality):
+    url = "https://chrunos-ytdl2.hf.space/download"
+    youtube_url = f'https://www.youtube.com/watch?v={track_id}'
+    async with aiohttp.ClientSession() as session:
+        data = {"url": youtube_url}
+        try:
+            async with session.post(url, json=data) as response:
+                if response.status == 200:
+                    result = await response.json()
+                    return result.get('download_url')
+                else:
+                    print(f"请求失败，状态码: {response.status}")
+                    return None
+        except aiohttp.ClientError as e:
+            print(f"发生客户端错误: {e}")
+            return None
 
 
 @app.route('/track_dl', methods=['POST'])
@@ -230,7 +247,7 @@ async def track_dl():
                 "premium": "https://chrunos.com/premium-shortcuts/"
             }), 400
             
-        dl_url = await get_audio_download_url(track_id, quality)
+        dl_url = await get_download_url(track_id, quality)
         logger.info(dl_url)
         
         if dl_url and "http" in dl_url: