Update app.py

app.py

@@ -384,7 +384,7 @@ def switch_working_frame(working_frame, scanned_frames, video_frames_dir):
     new_working_frame = None
     if working_frame == None:
         new_working_frame = os.path.join(video_frames_dir, scanned_frames[0])
-        return new_working_frame, gr.State([]), gr.State([]), new_working_frame, new_working_frame, new_working_frame
+        
     else:
         # Use a regular expression to find the integer
         match = re.search(r'frame_(\d+)', working_frame)
@@ -393,7 +393,7 @@ def switch_working_frame(working_frame, scanned_frames, video_frames_dir):
             frame_number = int(match.group(1))
             ann_frame_idx = frame_number
             new_working_frame = os.path.join(video_frames_dir, scanned_frames[ann_frame_idx])
-            return new_working_frame, gr.State([]), gr.State([]), new_working_frame, new_working_frame, new_working_frame
+    return gr.State([]), gr.State([]), new_working_frame, new_working_frame
 
 def reset_propagation(predictor, stored_inference_state):
     
@@ -446,8 +446,9 @@ with gr.Blocks() as demo:
                     video_in = gr.Video(label="Video IN")
             
             with gr.Column():
-                
-                working_frame = gr.Dropdown(label="working frame ID", choices=[""], value=None, visible=False, allow_custom_value=False, interactive=True)
+                with gf.Row():
+                    working_frame = gr.Dropdown(label="working frame ID", choices=[""], value=None, visible=False, allow_custom_value=False, interactive=True)
+                    change_current = gr.Button("change current ")
                 output_result = gr.Image(label="current working mask ref")
                 with gr.Row():
                     vis_frame_type = gr.Radio(label="Propagation level", choices=["check", "render"], value="check", scale=2)
@@ -511,10 +512,10 @@ with gr.Blocks() as demo:
     )
 
     
-    working_frame.change(
+    change_current.click(
         fn = switch_working_frame,
         inputs = [working_frame, scanned_frames, video_frames_dir],
-        outputs = [first_frame_path, tracking_points, trackings_input_label, input_first_frame_image, points_map, working_frame],
+        outputs = [tracking_points, trackings_input_label, input_first_frame_image, points_map],
         queue=False
     )