Update app.py

app.py

@@ -356,7 +356,7 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
     print(f"JPEG_IMAGES: {jpeg_images}")
 
     if vis_frame_type == "check":
-        return gr.update(value=jpeg_images), gr.update(value=None), gr.update(choices=available_frames_to_check, value=working_frame, visible=True), available_frames_to_check
+        return gr.update(value=jpeg_images), gr.update(value=None), gr.update(choices=available_frames_to_check, value=working_frame, visible=True), available_frames_to_check, gr.update(visible=True)
     elif vis_frame_type == "render":
         # Create a video clip from the image sequence
         original_fps = get_video_fps(video_in)
@@ -372,7 +372,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), working_frame, available_frames_to_check
+        return gr.update(value=None), gr.update(value=final_vid_output_path), working_frame, available_frames_to_check, gr.update(visible=True)
 
 def update_ui(vis_frame_type):
     if vis_frame_type == "check":
@@ -399,7 +399,7 @@ def reset_propagation(predictor, stored_inference_state):
     
     predictor.reset_state(stored_inference_state)
     print(f"RESET State: {stored_inference_state} ")
-    return gr.update(value=None, visible=False), stored_inference_state, None, ["frame_0.jpg"], first_frame_path
+    return gr.update(value=None, visible=False), stored_inference_state, None, ["frame_0.jpg"], first_frame_path, gr.update(visible=False)
     
 with gr.Blocks() as demo:
     first_frame_path = gr.State()
@@ -453,7 +453,7 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     vis_frame_type = gr.Radio(label="Propagation level", choices=["check", "render"], value="check", scale=2)
                     propagate_btn = gr.Button("Propagate", scale=1)
-                    reset_prpgt_brn = gr.Button("Reset", scale=0.75)
+                reset_prpgt_brn = gr.Button("Reset", scale=0.75, visible=False)
                 output_propagated = gr.Gallery(label="Propagated Mask samples gallery", visible=False)
                 output_video = gr.Video(visible=False)
                 # output_result_mask = gr.Image()
@@ -545,7 +545,7 @@ with gr.Blocks() as demo:
     reset_prpgt_brn.click(
         fn = reset_propagation,
         inputs = [loaded_predictor, stored_inference_state],
-        outputs = [output_propagated, stored_inference_state, output_result, available_frames_to_check, input_first_frame_image, working_frame],
+        outputs = [output_propagated, stored_inference_state, output_result, available_frames_to_check, input_first_frame_image, working_frame, reset_prpgt_brn],
         queue=False
     )
 
@@ -557,7 +557,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, available_frames_to_check, working_frame],
-        outputs = [output_propagated, output_video, working_frame, available_frames_to_check]
+        outputs = [output_propagated, output_video, working_frame, available_frames_to_check, reset_prpgt_brn]
     )
 
 demo.launch(show_api=False, show_error=True)