Update app.py

app.py

@@ -219,6 +219,7 @@ def get_mask_sam_process(
     video_frames_dir, # extracted_frames_output_dir defined in 'preprocess_video_in' function
     scanned_frames, 
     working_frame: str = None, # current frame being added points
+    available_frames_to_check,
     progress=gr.Progress(track_tqdm=True)
 ):
     
@@ -287,8 +288,12 @@ def get_mask_sam_process(
     plt.savefig(first_frame_output_filename, format='jpg')
     plt.close()
     torch.cuda.empty_cache()
+
+    # Assuming available_frames_to_check.value is a list
+    if working_frame not in available_frames_to_check.value:
+        available_frames_to_check.value.append(working_frame)
     
-    return "output_first_frame.jpg", frame_names, inference_state, gr.update(value=working_frame, visible=True)
+    return "output_first_frame.jpg", frame_names, inference_state, gr.update(choices=available_frames_to_check.value, 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 ####
@@ -389,6 +394,7 @@ with gr.Blocks() as demo:
     scanned_frames = gr.State()
     stored_inference_state = gr.State()
     stored_frame_names = gr.State()
+    available_frames_to_check = gr.State([])
     with gr.Column():
         gr.Markdown("# SAM2 Video Predictor")
         gr.Markdown("This is a simple demo for video segmentation with SAM2.")
@@ -424,6 +430,7 @@ 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)
                 output_result = gr.Image(label="current working mask ref")
                 with gr.Row():
@@ -505,6 +512,7 @@ with gr.Blocks() as demo:
             video_frames_dir, 
             scanned_frames, 
             working_frame,
+            available_frames_to_check,
         ],
         outputs = [
             output_result,