Update app.py

app.py

@@ -30,59 +30,88 @@ PROXIES = [
 # Function to search YouTube videos using yt-dlp for better reliability
 def youtube_search(query, max_results=10):
     cookies_file = "cookies.txt"  # You need to provide this file with cookies exported from YouTube
-    ydl_opts = {
-        'quiet': False,  # Set to False to get more detailed output from yt-dlp
-        'logger': logging.getLogger(),  # Use the logging module to capture yt-dlp logs
-        'simulate': True,
-        'noplaylist': True,  # Avoid extracting playlists
-        'format': 'best',
-        'proxy': random.choice(PROXIES),  # Randomly choose a proxy for each request
-        'http_headers': {
-            'User-Agent': ua.random  # Rotate user agents to make requests appear less like a bot
-        },
-    }
-
-    if os.path.exists(cookies_file):
-        ydl_opts['cookiefile'] = cookies_file
-        logging.debug("Using cookies for YouTube authentication.")
-
-    search_url = f"ytsearch{max_results}:{query}"
-    logging.debug(f"Starting YouTube search for query: {query}")
-
-    try:
-        # Introduce a random delay to avoid rate-limiting issues
-        time.sleep(random.uniform(2, 5))
-
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            result = ydl.extract_info(search_url, download=False)
-            gallery_items = []
-
-            if 'entries' in result:
-                logging.debug(f"Number of entries found: {len(result['entries'])}")
-                for entry in result['entries']:
-                    video_id = entry.get('id')
-                    # Fallback to YouTube static thumbnails if missing
-                    thumbnail_url = entry.get('thumbnail') if entry.get('thumbnail') else f"https://img.youtube.com/vi/{video_id}/hqdefault.jpg"
-                    video_title = entry.get('title', "Unknown Title")
-                    video_description = entry.get('description', "No description available.")
-
-                    if video_id:
-                        gallery_items.append({
-                            "thumbnail": thumbnail_url,
-                            "video_id": video_id,
-                            "title": video_title,
-                            "description": video_description
-                        })
-                        logging.debug(f"Added video: ID={video_id}, Thumbnail={thumbnail_url}, Title={video_title}")
-                    else:
-                        logging.debug(f"Missing video ID for entry: {entry}")
-            else:
-                logging.warning("No entries found in search result.")
-            return gallery_items, ""
-    except Exception as e:
-        error_message = f"Error during YouTube yt-dlp request: {e}"
-        logging.error(error_message)
+    proxies_attempted = 0
+    max_proxy_attempts = len(PROXIES)
+    success = False
+    gallery_items = []
+    error_message = ""
+
+    while not success and proxies_attempted <= max_proxy_attempts:
+        if proxies_attempted < max_proxy_attempts:
+            # Use a proxy
+            proxy = PROXIES[proxies_attempted]
+            logging.debug(f"Trying proxy: {proxy}")
+            ydl_opts = {
+                'quiet': False,  # Set to False to get more detailed output from yt-dlp
+                'logger': logging.getLogger(),  # Use the logging module to capture yt-dlp logs
+                'simulate': True,
+                'noplaylist': True,  # Avoid extracting playlists
+                'format': 'best',
+                'proxy': proxy,
+                'http_headers': {
+                    'User-Agent': ua.random  # Rotate user agents to make requests appear less like a bot
+                },
+            }
+        else:
+            # Fallback to direct access after all proxies fail
+            logging.debug("Falling back to direct connection without a proxy.")
+            ydl_opts = {
+                'quiet': False,
+                'logger': logging.getLogger(),
+                'simulate': True,
+                'noplaylist': True,
+                'format': 'best',
+                'http_headers': {
+                    'User-Agent': ua.random
+                },
+            }
+
+        if os.path.exists(cookies_file):
+            ydl_opts['cookiefile'] = cookies_file
+            logging.debug("Using cookies for YouTube authentication.")
+
+        search_url = f"ytsearch{max_results}:{query}"
+        logging.debug(f"Starting YouTube search for query: {query}")
+
+        try:
+            # Introduce a random delay to avoid rate-limiting issues
+            time.sleep(random.uniform(2, 5))
+
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                result = ydl.extract_info(search_url, download=False)
+
+                if 'entries' in result:
+                    logging.debug(f"Number of entries found: {len(result['entries'])}")
+                    for entry in result['entries']:
+                        video_id = entry.get('id')
+                        # Fallback to YouTube static thumbnails if missing
+                        thumbnail_url = entry.get('thumbnail') if entry.get('thumbnail') else f"https://img.youtube.com/vi/{video_id}/hqdefault.jpg"
+                        video_title = entry.get('title', "Unknown Title")
+                        video_description = entry.get('description', "No description available.")
+
+                        if video_id:
+                            gallery_items.append({
+                                "thumbnail": thumbnail_url,
+                                "video_id": video_id,
+                                "title": video_title,
+                                "description": video_description
+                            })
+                            logging.debug(f"Added video: ID={video_id}, Thumbnail={thumbnail_url}, Title={video_title}")
+                        else:
+                            logging.debug(f"Missing video ID for entry: {entry}")
+
+                    success = True
+                else:
+                    logging.warning("No entries found in search result.")
+
+        except Exception as e:
+            error_message = f"Error during YouTube yt-dlp request: {e}"
+            logging.error(error_message)
+            proxies_attempted += 1  # Increment the proxy attempt counter
+
+    if not success:
         return [], error_message
+    return gallery_items, ""
 
 # Function to display the video using the video URL
 def show_video(video_url):