Update app.py

app.py

@@ -1,6 +1,6 @@
 import sys
 import time
-from fastapi import FastAPI, BackgroundTasks, Request, HTTPException, Security
+from fastapi import FastAPI, BackgroundTasks, Request, HTTPException, Security, render_template, jsonify
 from fastapi.security import APIKeyHeader
 from fastapi.responses import FileResponse
 from fastapi.concurrency import run_in_threadpool
@@ -27,6 +27,8 @@ import logging
 import gc
 from typing import Dict, Any
 import re
+import asyncio
+import cloudscraper
 
 tmp_dir = tempfile.gettempdir()
 BASE_URL = "https://chrunos-zam.hf.space"
@@ -174,6 +176,79 @@ def get_user_ip(request: Request) -> str:
     return request.client.host
 
 
+class ApiRotator:
+    def __init__(self, apis):
+        self.apis = apis
+        self.last_successful_index = None
+
+    def get_prioritized_apis(self):
+        if self.last_successful_index is not None:
+            # Move the last successful API to the front
+            rotated_apis = (
+                [self.apis[self.last_successful_index]] + 
+                self.apis[:self.last_successful_index] + 
+                self.apis[self.last_successful_index+1:]
+            )
+            return rotated_apis
+        return self.apis
+
+    def update_last_successful(self, index):
+        self.last_successful_index = index
+
+# In your function:
+api_rotator = ApiRotator([
+    "https://cblt.fariz.dev",
+    "https://cobalt-api.ayo.tf",
+    "http://34.107.254.11",
+    "https://dwnld.nichind.dev",
+    "https://cobalt-api.kwiatekmiki.com",
+    "https://yt.edd1e.xyz/"
+    
+])
+
+
+
+async def get_track_download_url(video_url: str) -> str:
+    apis = api_rotator.get_prioritized_apis()
+    session = cloudscraper.create_scraper()  # Requires cloudscraper package
+    headers = {
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
+    }
+
+    for i, api_url in enumerate(apis):
+        try:
+            logger.info(f"Attempting to get download URL from: {api_url}")
+            y_url = video_url
+            response = session.post(
+                api_url,
+                timeout=20,
+                json={"url": y_url, "videoQuality": "720"},
+                headers=headers
+            )
+            logger.info(f"Response status: {response.status_code}")
+            logger.info(f"Response content: {response.content}")
+
+            if response.headers.get('content-type', '').startswith('application/json'):
+                json_response = response.json()
+                error_code = json_response.get("error", {}).get("code", "")
+                
+                if error_code == "error.api.content.video.unavailable":
+                    logger.warning(f"Video unavailable error from {api_url}")
+                    break  # Only break for specific error
+                
+                if "url" in json_response:
+                    api_rotator.update_last_successful(i)
+                    return json_response["url"]
+                
+        except Exception as e:
+            logger.error(f"Failed with {api_url}: {str(e)}")
+            continue
+    
+    logger.error(f"No download URL found")
+    return {"error": "Download URL not found"}
+
 
 
 @app.post("/maxs")
@@ -233,9 +308,16 @@ async def download_high_quality_video(request: Request):
     }
 
     if is_youtube_url:
-        ydl_opts['cookiefile'] = cookiefile
+        dl_url = await get_track_download_url(video_url)
+        if dl_url and "http" in dl_url:
+            result = {"url": dl_url}
+            return jsonify(result)
+        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}.mp4"))