Update app.py

app.py

@@ -194,11 +194,10 @@ class LiveYouTubeObjectDetector:
     @staticmethod
     def get_live_streams(query=""):
         return SearchService.search(query if query else "world live cams", SearchFilter.LIVE)
-
+    
     def render(self):
         with gr.Blocks(title="Object Detection in Live YouTube Streams",
-                       css="footer {visibility: hidden}",
-                       analytics_enabled=False) as app:
+                       css="footer {visibility: hidden}", analytics_enabled=False) as app:
             self.page_title.render()
             with gr.Column():
                 with gr.Group():
@@ -212,28 +211,28 @@ class LiveYouTubeObjectDetector:
                     self.search_button.render()
             with gr.Row():
                 self.gallery.render()
-
+    
             @self.gallery.select(inputs=None, outputs=[self.annotated_image, self.stream_input], scroll_to_output=True)
             def detect_objects_from_gallery_item(evt: gr.SelectData):
                 if evt.index is not None and evt.index < len(self.streams):
                     selected_stream = self.streams[evt.index]
                     stream_url = SearchService.get_youtube_url(selected_stream["video_id"])
                     self.stream_input.value = stream_url
-                    return self.detect_objects(stream_url)
-                return self.create_black_image()
-
+                    annotated_image_result = self.detect_objects(stream_url)
+                    return annotated_image_result, stream_url
+                return self.create_black_image(), ""
+    
             @self.search_button.click(inputs=[self.search_input], outputs=[self.gallery])
             def search_live_streams(query):
                 self.streams = self.get_live_streams(query)
                 gallery_items = [(stream["thumbnail_url"], stream["title"]) for stream in self.streams]
                 return gallery_items
-
+    
             @self.submit_button.click(inputs=[self.stream_input], outputs=[self.annotated_image])
             def detect_objects_from_url(url):
                 return self.detect_objects(url)
-
+    
         app.queue().launch(show_api=False, debug=True)
 
-
 if __name__ == "__main__":
     LiveYouTubeObjectDetector().render()