Update app.py

app.py

@@ -54,13 +54,17 @@ def process_video(video_path, output_option):
         if hasattr(result, 'boxes'):
             object_count += len(result.boxes)
 
+    # Set default outputs
+    video_output = matched_mp4_files[0] if matched_mp4_files else None
+    count_output = object_count
+
     # Determine the output based on user's choice
     if output_option == "Count":
-        return object_count
+        return count_output, None
     elif output_option == "Video":
-        return matched_mp4_files[0]
+        return None, video_output
     elif output_option == "Both":
-        return object_count, matched_mp4_files[0]
+        return count_output, video_output
 
 # Define Gradio inputs
 video_input = gr.Video()
@@ -76,7 +80,7 @@ examples = [
 video_interface = gr.Interface(
     fn=process_video,
     inputs=[video_input, output_option_input],
-    outputs=[gr.Textbox(label="Object Count"), "video"],  # Update outputs to support multiple types
+    outputs=[gr.Textbox(label="Object Count"), "video"],  # Ensure two outputs are defined
     title="YOLO Video Tracking Application",
     description="A simple application to track objects in a video using YOLO model. Upload your own video, or click one of the examples to load them.",
     article="""<div>