Update app.py

app.py

@@ -24,7 +24,11 @@ API_KEY_NAME = "X-API-Key"
 api_key_header = APIKeyHeader(name=API_KEY_NAME, auto_error=False)
 
 # Constants
-INVIDIOUS_INSTANCE = "https://iv.melmac.space"
+INVIDIOUS_INSTANCES = [
+    "https://invidious.privacydev.net",
+    "https://invidious.reallyaweso.me",
+    "https://invidious.adminforge.de"
+]
 API_KEY = os.environ.get("API_KEY")
 
 # S3 Configuration
@@ -45,72 +49,75 @@ def get_random_user_agent() -> str:
     return random.choice(user_agents)
 
 async def search_and_get_videos(query: str) -> List[Dict]:
-    url = f"{INVIDIOUS_INSTANCE}/api/v1/search?q={query}&type=video"
-    try:
-        async with aiohttp.ClientSession() as session:
-            async with session.get(url, headers={"User-Agent": get_random_user_agent()}) as response:
-                response.raise_for_status()
-                search_results = await response.json()
-                videos = [
-                    {
-                        "id": video.get("videoId"),
-                        "title": video.get("title"),
-                        "thumbnail": video["videoThumbnails"][0]["url"]
-                        if video.get("videoThumbnails")
-                        else "",
-                    }
-                    for video in search_results
-                ][:2]
-                return videos
-    except aiohttp.ClientError as e:
-        logger.error(f"Error performing video search: {e}")
-        return []
-
-async def get_youtube_audio(video_id: str, max_retries: int = 3) -> Dict:
-    for attempt in range(max_retries):
+    for instance in INVIDIOUS_INSTANCES:
+        url = f"{instance}/api/v1/search?q={query}&type=video"
         try:
-            url = f"{INVIDIOUS_INSTANCE}/api/v1/videos/{video_id}"
-            
             async with aiohttp.ClientSession() as session:
-                async with session.get(url) as response:
+                async with session.get(url, headers={"User-Agent": get_random_user_agent()}) as response:
                     response.raise_for_status()
-                    video_data = await response.json()
-                    
-                    audio_format = next((format for format in video_data.get('adaptiveFormats', []) 
-                                         if format.get('type', '').startswith('audio/mp4')), None)
-                    
-                    if audio_format:
-                        audio_url = audio_format.get('url')
-                        if audio_url:
-                            try:
-                                async with session.get(audio_url) as audio_response:
-                                    audio_content = await audio_response.read()
-                                    
-                                    with tempfile.NamedTemporaryFile(delete=False, suffix='.m4a') as temp_file:
-                                        temp_file.write(audio_content)
-                                        temp_file_path = temp_file.name
-                                    
-                                    return {'success': True, 'temp_file_path': temp_file_path}
-                            except aiohttp.ServerDisconnectedError:
-                                if attempt == max_retries - 1:
-                                    logger.error(f"Max retries reached for video ID {video_id}")
-                                    return {'success': False, 'error': "Max retries reached"}
-                                await asyncio.sleep(1 * (attempt + 1))
-                                continue
-                    
-                    logger.warning(f"No suitable audio format found for video ID {video_id}")
-                    return {'success': False, 'error': "No suitable audio format found"}
+                    search_results = await response.json()
+                    videos = [
+                        {
+                            "id": video.get("videoId"),
+                            "title": video.get("title"),
+                            "thumbnail": video["videoThumbnails"][0]["url"]
+                            if video.get("videoThumbnails")
+                            else "",
+                        }
+                        for video in search_results
+                    ][:2]
+                    return videos
         except aiohttp.ClientError as e:
-            logger.error(f"Network error fetching YouTube audio for video ID {video_id}: {e}")
-        except json.JSONDecodeError:
-            logger.error(f"Error decoding JSON response for video ID {video_id}")
-        except Exception as e:
-            logger.error(f"Unexpected error fetching YouTube audio for video ID {video_id}: {e}")
+            logger.error(f"Error performing video search on {instance}: {e}")
+    logger.error("All Invidious instances failed")
+    return []
+
+async def get_youtube_audio(video_id: str, max_retries: int = 3) -> Dict:
+    for instance in INVIDIOUS_INSTANCES:
+        for attempt in range(max_retries):
+            try:
+                url = f"{instance}/api/v1/videos/{video_id}"
+                
+                async with aiohttp.ClientSession() as session:
+                    async with session.get(url) as response:
+                        response.raise_for_status()
+                        video_data = await response.json()
+                        
+                        audio_format = next((format for format in video_data.get('adaptiveFormats', []) 
+                                             if format.get('type', '').startswith('audio/mp4')), None)
+                        
+                        if audio_format:
+                            audio_url = audio_format.get('url')
+                            if audio_url:
+                                try:
+                                    async with session.get(audio_url) as audio_response:
+                                        audio_content = await audio_response.read()
+                                        
+                                        with tempfile.NamedTemporaryFile(delete=False, suffix='.m4a') as temp_file:
+                                            temp_file.write(audio_content)
+                                            temp_file_path = temp_file.name
+                                        
+                                        return {'success': True, 'temp_file_path': temp_file_path}
+                                except aiohttp.ServerDisconnectedError:
+                                    if attempt == max_retries - 1:
+                                        logger.error(f"Max retries reached for video ID {video_id} on {instance}")
+                                        break
+                                    await asyncio.sleep(1 * (attempt + 1))
+                                    continue
+                        
+                        logger.warning(f"No suitable audio format found for video ID {video_id} on {instance}")
+                        break
+            except aiohttp.ClientError as e:
+                logger.error(f"Network error fetching YouTube audio for video ID {video_id} on {instance}: {e}")
+            except json.JSONDecodeError:
+                logger.error(f"Error decoding JSON response for video ID {video_id} on {instance}")
+            except Exception as e:
+                logger.error(f"Unexpected error fetching YouTube audio for video ID {video_id} on {instance}: {e}")
             if attempt == max_retries - 1:
-                return {'success': False, 'error': str(e)}
+                break
             await asyncio.sleep(1 * (attempt + 1))
     
-    return {'success': False, 'error': "Failed to fetch audio after multiple attempts"}
+    return {'success': False, 'error': "Failed to fetch audio after multiple attempts on all instances"}
 
 def extract_text_from_document(file: UploadFile) -> dict:
     try: