Update app.py

app.py

@@ -7,7 +7,6 @@ from datetime import datetime, timedelta
 import time
 import asyncio
 import cloudscraper
-from fastapi import FastAPI, BackgroundTasks, HTTPException
 
 app = Flask(__name__)
 ytmusic = YTMusic()
@@ -88,14 +87,15 @@ async def get_track_download_url(track_id: str) -> str:
             logger.info(f"Response status: {response.status_code}")
             logger.info(f"Response content: {response.content}")
 
-            if response.status_code == 200:
-                download_url = response.json().get("url")
-                if download_url:
-                    api_rotator.update_last_successful(apis.index(api_url))
-                    logger.info(f"Successfully obtained download URL from {api_url}")
-                    return download_url
-        except requests.exceptions.RequestException as e:
-            logger.warning(f"Failed to get download URL from {api_url}: {str(e)}")
+            if response.headers.get('content-type', '').startswith('application/json'):
+                json_response = response.json()
+                if "video.unavailable" in json_response.get("error", {}).get("code", ""):
+                    break
+                
+                api_rotator.update_last_successful(apis.index(api_url))
+                return json_response.get("url")
+        except Exception as e:
+            logger.error(f"Failed with {api_url}: {str(e)}")
             continue
     
     logger.error(f"No download URL found")