Update app.py

app.py

@@ -225,8 +225,10 @@ async def process_video(video_data, temp_dir, progress=gr.Progress()):
             torch.cuda.empty_cache()
 
 async def gradio_interface(video_file=None, stream_url=None):
+    # Input validation: yield error message instead of returning
     if not video_file and not stream_url:
-        return "Please upload a video or provide a stream URL.", "", "", ""
+        yield "Please upload a video or provide a stream URL.", "", "", ""
+        return  # Use bare return to exit the generator
     
     temp_dir = tempfile.mkdtemp(prefix="Ultralytics_")
     try:
@@ -256,6 +258,9 @@ async def gradio_interface(video_file=None, stream_url=None):
                 video_data = f.read()
             async for result in process_video(video_data, temp_dir):
                 yield result
+    except Exception as e:
+        logger.error(f"Error in gradio_interface: {str(e)}")
+        yield f"Error: {str(e)}", "", "", ""
     finally:
         shutil.rmtree(temp_dir, ignore_errors=True)