Update app.py

app.py

@@ -44,10 +44,11 @@ def youtube_search(query, max_results=50):
                         logging.debug(f"Missing video ID or thumbnail for entry: {entry}")
             else:
                 logging.warning("No entries found in search result.")
-            return gallery_items
+            return gallery_items, ""
     except Exception as e:
-        logging.error(f"Error during YouTube yt-dlp request: {e}")
-        return []
+        error_message = f"Error during YouTube yt-dlp request: {e}"
+        logging.error(error_message)
+        return [], error_message
 
 # Function to display the video using the video URL
 def show_video(video_url):
@@ -93,6 +94,7 @@ with gr.Blocks() as demo:
             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")
+            error_output = gr.Textbox(label="Error Message", interactive=False, visible=False)
         
         with gr.Column(scale=2):
             selected_video_link = gr.Textbox(label="Selected Video Link", interactive=False)
@@ -101,9 +103,10 @@ with gr.Blocks() as demo:
 
     # Define search button behavior
     def update_search_results(query):
-        gallery_items = youtube_search(query)
-        logging.debug(f"Number of gallery items returned: {len(gallery_items)}")
-        return gallery_items
+        gallery_items, error_message = youtube_search(query)
+        if error_message:
+            return [], error_message, gr.update(visible=True)
+        return gallery_items, "", 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):
@@ -118,9 +121,9 @@ with gr.Blocks() as demo:
         logging.debug(f"Playing video with URL: {video_url}")
         return show_video(video_url)
 
-    search_button.click(update_search_results, inputs=search_query_input, outputs=search_output)
+    search_button.click(update_search_results, inputs=search_query_input, outputs=[search_output, error_output, error_output])
     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)
 
 # Launch the Gradio interface
-demo.launch()
+demo.launch()