Update app.py

app.py

@@ -215,6 +215,7 @@ def segment_frame(evt: gr.SelectData, label, video_state):
     video_state["masks"] = np.expand_dims(mask[:, :, 0], axis=0)
 
     for i in range(len(video_state["input_points"])):
+
         point = video_state["input_points"][i]
         if video_state["input_labels"][i] == 0:
             cv2.circle(painted_image, point, radius=3, color=(0, 0, 255), thickness=-1)
@@ -396,7 +397,14 @@ def track_video(n_frames, video_state):
         video_file, codec="libx264", audio=False, verbose=False, logger=None
     )
     print("Tracking done")
-    return video_file, video_state
+    print("Tracking done, file:", video_file)
+    try:
+        exists = os.path.exists(video_file)
+        size = os.path.getsize(video_file) if exists else -1
+        print("File exists?", exists, "size:", size)
+    except Exception as e:
+        print("Error checking video file:", repr(e))
+    return video_file
 
 
 @torch.no_grad()
@@ -761,7 +769,7 @@ with gr.Blocks() as demo:
         track_btn.click(
             track_video,
             inputs=[n_frames_slider, video_state],
-            outputs=[video_output, video_state],
+            outputs=[video_output],
         )
 
         ref_image_input.change(