Update app.py

app.py

@@ -395,11 +395,11 @@ def switch_working_frame(working_frame, scanned_frames, video_frames_dir):
             new_working_frame = os.path.join(video_frames_dir, scanned_frames[ann_frame_idx])
     return gr.State([]), gr.State([]), new_working_frame, new_working_frame
 
-def reset_propagation(predictor, stored_inference_state):
+def reset_propagation(first_frame_path, 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, "frame_0.jpg", gr.update(visible=False)
+    return first_frame_path, gr.update(value=None, visible=False), stored_inference_state, None, ["frame_0.jpg"], first_frame_path, "frame_0.jpg", gr.update(visible=False)
     
 with gr.Blocks() as demo:
     first_frame_path = gr.State()
@@ -544,8 +544,8 @@ 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, reset_prpgt_brn],
+        inputs = [first_frame_path, loaded_predictor, stored_inference_state],
+        outputs = [points_map, output_propagated, stored_inference_state, output_result, available_frames_to_check, input_first_frame_image, working_frame, reset_prpgt_brn],
         queue=False
     )