Update app.py

app.py

@@ -335,7 +335,7 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
     
 
     if vis_frame_type == "check":
-        return gr.update(value=jpeg_images), gr.update(value=None)
+        return gr.update(value=jpeg_images), gr.update(value=None), gr.update(choices=jpeg_images)
     elif vis_frame_type == "render":
         # Create a video clip from the image sequence
         original_fps = get_video_fps(video_in)
@@ -351,7 +351,7 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
             codec='libx264'
         )
         
-        return gr.update(value=None), gr.update(value=final_vid_output_path)
+        return gr.update(value=None), gr.update(value=final_vid_output_path), None
 
 def update_ui(vis_frame_type):
     if vis_frame_type == "check":
@@ -403,6 +403,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=None, value=None)
                 output_result = gr.Image()
                 with gr.Row():
                     vis_frame_type = gr.Radio(label="Propagation level", choices=["check", "render"], value="check", scale=2)
@@ -446,7 +447,7 @@ with gr.Blocks() as demo:
     ).then(
         fn = propagate_to_all,
         inputs = [video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type],
-        outputs = [output_propagated, output_video]
+        outputs = [output_propagated, output_video, working_frame]
     )
 
 demo.launch(show_api=False, show_error=True)