Update app.py

app.py

@@ -2,7 +2,7 @@ import subprocess
 import sys
 import random
 import time
-import os  # <-- Added import for os module to handle file checks
+import os  # Added import for os module to handle file checks
 
 # Ensure compatible versions of httpx and httpcore are installed
 subprocess.check_call([sys.executable, "-m", "pip", "install", "httpx==0.18.2", "httpcore==0.13.6"])
@@ -33,24 +33,36 @@ def get_random_proxy():
 
 # 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
+    cookies_file = "cookies.txt"
     proxies_attempted = 0
     max_proxy_attempts = len(PROXIES)
     success = False
     gallery_items = []
     error_message = ""
 
-    while not success and proxies_attempted < max_proxy_attempts:
-        proxy = get_random_proxy()
-        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': f'http://{proxy}',
-        }
+    while not success and proxies_attempted < max_proxy_attempts + 1:  # +1 to allow fallback without a proxy
+        if proxies_attempted < max_proxy_attempts:
+            proxy = get_random_proxy()
+            logging.debug(f"Trying proxy: {proxy}")
+            ydl_opts = {
+                'quiet': False,
+                'logger': logging.getLogger(),
+                'simulate': True,
+                'noplaylist': True,
+                'format': 'best',
+                'proxy': f'http://{proxy}',
+                'socket_timeout': 30  # Extended timeout
+            }
+        else:
+            logging.debug("All proxies failed. Trying without a proxy.")
+            ydl_opts = {
+                'quiet': False,
+                'logger': logging.getLogger(),
+                'simulate': True,
+                'noplaylist': True,
+                'format': 'best',
+                'socket_timeout': 30  # Extended timeout
+            }
 
         if cookies_file and os.path.exists(cookies_file):
             ydl_opts['cookiefile'] = cookies_file
@@ -103,10 +115,10 @@ def youtube_search(query, max_results=10):
 def show_video(video_url):
     video_id = None
     patterns = [
-        r"youtube\\.com/watch\\?v=([^&?\\/]+)",
-        r"youtube\\.com/embed/([^&?\\/]+)",
-        r"youtube\\.com/v/([^&?\\/]+)",
-        r"youtu\\.be/([^&?\\/]+)"
+        r"youtube\.com/watch\?v=([^&?\/]+)",
+        r"youtube\.com/embed/([^&?\/]+)",
+        r"youtube\.com/v/([^&?\/]+)",
+        r"youtu\.be/([^&?\/]+)"
     ]
 
     for pattern in patterns:
@@ -151,7 +163,7 @@ with gr.Blocks() as demo:
         if error_message:
             return [], error_message, gr.update(visible=True)
 
-        gallery_items_display = [(item["thumbnail"], f"{item['title']}\\n{item['description']}", item["video_id"]) for item in gallery_items]
+        gallery_items_display = [(item["thumbnail"], f"{item['title']}\n{item['description']}", item["video_id"]) for item in gallery_items]
         
         return gallery_items_display, "", gr.update(visible=False)
 
@@ -169,4 +181,4 @@ with gr.Blocks() as demo:
     search_output.select(on_video_select, inputs=None, outputs=selected_video_link)
     play_video_button.click(play_video, inputs=selected_video_link, outputs=video_output)
 
-demo.launch()
+demo.launch()