Update app.py

app.py

@@ -299,7 +299,7 @@ def get_mask_sam_process(
         available_frames_to_check.append(working_frame)
         print(available_frames_to_check)
     
-    return "output_first_frame.jpg", frame_names, predictor, inference_state, gr.update(choices=available_frames_to_check, value=working_frame, visible=True)
+    return gr.update(visible=True), "output_first_frame.jpg", frame_names, predictor, inference_state, gr.update(choices=available_frames_to_check, value=working_frame, visible=True)
 
 def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type, available_frames_to_check, working_frame, progress=gr.Progress(track_tqdm=True)):   
     #### PROPAGATION ####
@@ -453,7 +453,7 @@ with gr.Blocks() as demo:
             with gr.Column():
                 with gr.Row():
                     working_frame = gr.Dropdown(label="working frame ID", choices=[""], value=None, visible=False, allow_custom_value=False, interactive=True)
-                    change_current = gr.Button("change current ")
+                    change_current = gr.Button("change current", 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)
@@ -539,6 +539,7 @@ with gr.Blocks() as demo:
             available_frames_to_check,
         ],
         outputs = [
+            change_current,
             output_result, 
             stored_frame_names, 
             loaded_predictor,