Update app.py

app.py

@@ -246,11 +246,11 @@ def get_mask_sam_process(
     # predictor.reset_state(inference_state) # if any previous tracking, reset
 
     ### HANDLING WORKING FRAME
-    new_working_frame = None
+    # new_working_frame = None
     # Add new point
     if working_frame is None:
         ann_frame_idx = 0  # the frame index we interact with, 0 if it is the first frame
-        new_working_frame = "frames_output_images/frame_0.jpg"
+        working_frame = "frame_0.jpg"
     else:
         # Use a regular expression to find the integer
         match = re.search(r'frame_(\d+)', working_frame)
@@ -258,8 +258,8 @@ def get_mask_sam_process(
             # Extract the integer from the match
             frame_number = int(match.group(1))
             ann_frame_idx = frame_number
-            new_working_frame = f"frames_output_images/frame_{ann_frame_idx}.jpg"
-    print(f"NEW_WORKING_FRAME PATH: {new_working_frame}")
+            
+    print(f"NEW_WORKING_FRAME PATH: {working_frame}")
     
     ann_obj_id = 1  # give a unique id to each object we interact with (it can be any integers)
     
@@ -288,7 +288,7 @@ def get_mask_sam_process(
     plt.close()
     torch.cuda.empty_cache()
     
-    return "output_first_frame.jpg", frame_names, inference_state
+    return "output_first_frame.jpg", frame_names, inference_state, gr.update(value=working_frame, visible=True)
 
 def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type, progress=gr.Progress(track_tqdm=True)):   
     #### PROPAGATION ####
@@ -510,6 +510,7 @@ with gr.Blocks() as demo:
             output_result, 
             stored_frame_names, 
             stored_inference_state,
+            working_frame,
         ]
     )