Update app.py

app.py

@@ -251,13 +251,31 @@ async def get_track_download_url(video_url: str) -> str:
 async def extract_video_info(video_url: str) -> str:
     api_url = f'https://yt-dl-web.vercel.app/api/info?url={video_url}'
     session = cloudscraper.create_scraper()
-    response = session.get(api_url, timeout=20)
-    logger.info(response)
-    if response:
-        json_response = response.json()
-        return json_response
-    else:
-        return {"error": "Download URL not found"}
+    try:
+        response = session.get(api_url, timeout=20)
+        logger.info(response)
+        if response.status_code == 200:
+            json_response = response.json()
+            result = []
+            # 检查 formats 列表是否存在且不为空
+            if 'formats' in json_response:
+                for format_item in json_response['formats']:
+                    format_url = format_item.get('url')
+                    format_id = format_item.get('format_id')
+                    if format_id and format_url:
+                        result.append({
+                            "url": format_url,
+                            "format_id": format_id
+                        })
+                title = json_response.get('title')
+                return result
+            else:
+                return {"error": "No formats available"}
+        else:
+            return {"error": f"Request failed with status code {response.status_code}"}
+    except Exception as e:
+        logger.error(f"An error occurred: {e}")
+        return {"error": str(e)}
 
 
 @app.post("/test")