Update app.py

app.py

@@ -102,9 +102,13 @@ def youtube_api_search(query, max_results=1000):
             # Update params with the nextPageToken to get the next batch of results
             params['pageToken'] = response.json()['nextPageToken']
 
-        # Create a list of tuples with thumbnail URL and video ID as the caption
+        # Create a list of tuples with thumbnail URL, video ID, and video title
         gallery_items = [
-            (result["snippet"]["thumbnails"]["medium"]["url"], result["id"]["videoId"]) for result in all_results
+            (
+                result["snippet"].get("thumbnails", {}).get("medium", {}).get("url", "https://via.placeholder.com/150"),
+                result["id"]["videoId"],
+                result["snippet"].get("title", "No title available")
+            ) for result in all_results
         ]
 
         return gallery_items
@@ -172,14 +176,14 @@ with gr.Blocks() as demo:
             gallery_items = youtube_api_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
+        # Display videos with thumbnails, video IDs, and titles
+        gallery_items_display = [(item[0], f"{item[2]} ({item[1]})") 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
     def on_video_select(evt: gr.SelectData):
         # Extract the video ID from the event value, which is a dictionary containing details of the selected item
-        selected_video_id = evt.value["caption"]
+        selected_video_id = evt.value["caption"].split('(')[-1][:-1]  # Extract video ID from caption
         video_url = f"https://www.youtube.com/watch?v={selected_video_id}"
         logging.debug(f"Video selected: {video_url}")
         return video_url