Update app.py

app.py

@@ -336,7 +336,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=jpeg_images, value=jpeg_image[0])
+        return gr.update(value=jpeg_images), gr.update(value=None), gr.update(choices=jpeg_images, value=jpeg_images[0], visible=True)
     elif vis_frame_type == "render":
         # Create a video clip from the image sequence
         original_fps = get_video_fps(video_in)
@@ -404,8 +404,8 @@ 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()
+                working_frame = gr.Dropdown(label="working frame ID", choices=None, value=None, visible=False)
+                output_result = gr.Image(label="current working mask ref")
                 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)