Update app.py

app.py

@@ -325,7 +325,7 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
     
 
     if vis_frame_type == "check":
-        return gr.update(value=jpeg_images, visible=True), gr.update(visible=False, value=None)
+        return gr.update(value=jpeg_images), gr.update(value=None)
     elif vis_frame_type == "render":
         # Create a video clip from the image sequence
         original_fps = get_video_fps(video_in)
@@ -333,9 +333,14 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
         clip = ImageSequenceClip(jpeg_images, fps=fps)
         # Write the result to a file
         final_vid_output_path = "output_video.mp4"
-        clip.write_videofile(output_path, codec='libx264')
-        return gr.update(visible=False, value=None), gr.update(value=final_vid_output_path, visible=True)
-    
+        clip.write_videofile(final_vid_output_path, codec='libx264')
+        return gr.update(value=None), gr.update(value=final_vid_output_path)
+
+def update_ui(vis_frame_type):
+    if vis_frame_type == "check":
+        return gr.update(visible=True), gr.update(visible=False)
+    elif vis_frame_type == "render":
+        return gr.update(visible=False), gr.update(visible=True)
 
 with gr.Blocks() as demo:
     first_frame_path = gr.State()
@@ -405,6 +410,11 @@ with gr.Blocks() as demo:
     )
 
     propagate_btn.click(
+        fn = update_ui,
+        inputs = [vis_frame_type],
+        outputs = [output_propagated, output_video],
+        queue=False
+    ).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]