Update app.py

app.py

@@ -8,41 +8,30 @@ import gradio as gr
 import requests
 from bs4 import BeautifulSoup
 import re
+import yt_dlp
 
-# Function to search YouTube videos using web scraping
+# Function to search YouTube videos using yt-dlp for better reliability
 def youtube_search(query, max_results=50):
-    search_url = f"https://www.youtube.com/results?search_query={query}"
-    headers = {
-        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
+    ydl_opts = {
+        'quiet': True,
+        'extract_flat': 'in_playlist',
     }
+    search_url = f"ytsearch{max_results}:{query}"
 
     try:
-        # Make the HTTP request to YouTube search results page
-        response = requests.get(search_url, headers=headers)
-        response.raise_for_status()  # Raise an error for bad responses (4xx or 5xx)
-
-        # Parse the HTML content
-        soup = BeautifulSoup(response.text, "html.parser")
-        video_elements = soup.find_all("a", href=True, class_="yt-uix-tile-link")
-        
-        # Extract up to 'max_results' video details
-        gallery_items = []
-        for idx, video in enumerate(video_elements):
-            if idx >= max_results:
-                break
-
-            video_id = video['href'].split('=')[-1]
-            video_title = video.get_text(strip=True)
-            thumbnail_url = f"https://img.youtube.com/vi/{video_id}/mqdefault.jpg"
-
-            # Append tuple (thumbnail, video ID)
-            gallery_items.append((thumbnail_url, video_id))
-
-        return gallery_items
-
-    except requests.exceptions.RequestException as e:
-        # Print the error message to help debug issues
-        print(f"Error during YouTube web scraping request: {e}")
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            result = ydl.extract_info(search_url, download=False)
+            gallery_items = []
+
+            if 'entries' in result:
+                for entry in result['entries']:
+                    video_id = entry.get('id')
+                    thumbnail_url = entry.get('thumbnail')
+                    if video_id and thumbnail_url:
+                        gallery_items.append((thumbnail_url, video_id))
+            return gallery_items
+    except Exception as e:
+        print(f"Error during YouTube yt-dlp request: {e}")
         return []
 
 # Function to display the video using the video URL