Update app.py

app.py

@@ -14,13 +14,13 @@ import logging
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Function to search YouTube videos using yt-dlp for better reliability
-def youtube_search(query, max_results=50):
+def youtube_search(query, max_results=10):
     ydl_opts = {
         'quiet': False,  # Set to False to get more detailed output from yt-dlp
-        'extract_flat': 'in_playlist',
         'logger': logging.getLogger(),  # Use the logging module to capture yt-dlp logs
         'simulate': True,
-        'noplaylist': True,  # To avoid playlist entries
+        'noplaylist': True,  # Avoid extracting playlists
+        'format': 'best',
     }
     search_url = f"ytsearch{max_results}:{query}"
 
@@ -35,11 +35,18 @@ def youtube_search(query, max_results=50):
                 logging.debug(f"Number of entries found: {len(result['entries'])}")
                 for entry in result['entries']:
                     video_id = entry.get('id')
-                    thumbnail_url = entry.get('thumbnail') if entry.get('thumbnail') else "https://via.placeholder.com/150"
+                    # 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_url, video_id, video_title))
+                        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}")
@@ -94,7 +101,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=3):
             search_query_input = gr.Textbox(label="Search YouTube", placeholder="Enter your search query here")
             search_button = gr.Button("Search")
-            search_output = gr.Gallery(label="Search Results", columns=5, height="1500px")
+            search_output = gr.Gallery(label="Search Results", columns=2, height="1000px", elem_id="gallery")
             error_output = gr.Textbox(label="Error Message", interactive=False, visible=False)
         
         with gr.Column(scale=2):
@@ -107,8 +114,10 @@ with gr.Blocks() as demo:
         gallery_items, error_message = youtube_search(query)
         if error_message:
             return [], error_message, gr.update(visible=True)
-        # Display videos even if the title or thumbnail is missing by using placeholders
-        gallery_items_display = [(item[0], item[1]) for item in gallery_items]  # Show thumbnail and video ID only
+        
+        # Prepare 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)
 
     # Update the selected video link field when a video is clicked in the gallery