Update app.py

app.py

@@ -52,7 +52,7 @@ def preprocess_video_in(video_path):
     cap.release()
 
     # 'image' is the first frame extracted from video_in
-    return first_frame, gr.State([]), gr.State([]), first_frame, first_frame
+    return first_frame, gr.State([]), gr.State([]), first_frame, first_frame, output_dir
 
 def get_point(point_type, tracking_points, trackings_input_label, first_frame_path, evt: gr.SelectData):
     print(f"You selected {evt.value} at {evt.index} from {evt.target}")
@@ -167,7 +167,7 @@ def show_masks(image, masks, scores, point_coords=None, box_coords=None, input_l
     return combined_images, mask_images
 
 
-def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_input_label):
+def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_input_label, frames_output_dir):
     # 1. We need to preprocess the video and store frames in the right directory
     # — Penser à utiliser un ID unique pour le dossier
 
@@ -190,7 +190,8 @@ def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_
 
     
     # `video_dir` a directory of JPEG frames with filenames like `<frame_index>.jpg`
-    video_dir = "./videos/bedroom"
+    print(f"STATE FRAME OUTPUT DIRECTORY: {frames_output_dir.value}")
+    video_dir = frames_output_dir.value
     
     # scan all the JPEG frame names in this directory
     frame_names = [
@@ -279,6 +280,7 @@ with gr.Blocks() as demo:
     first_frame_path = gr.State()
     tracking_points = gr.State([])
     trackings_input_label = gr.State([])
+    frames_output_dir = gr.State()
     with gr.Column():
         gr.Markdown("# SAM2 Video Predictor")
         gr.Markdown("This is a simple demo for video segmentation with SAM2.")
@@ -317,7 +319,7 @@ with gr.Blocks() as demo:
     video_in.upload(
         fn = preprocess_video_in, 
         inputs = [video_in], 
-        outputs = [first_frame_path, tracking_points, trackings_input_label, input_first_frame_image, points_map],
+        outputs = [first_frame_path, tracking_points, trackings_input_label, input_first_frame_image, points_map, frames_output_dir],
         queue = False
     )
 
@@ -330,7 +332,7 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         fn = sam_process,
-        inputs = [input_first_frame_image, checkpoint, tracking_points, trackings_input_label],
+        inputs = [input_first_frame_image, checkpoint, tracking_points, trackings_input_label, frames_output_dir],
         outputs = [output_result]
     )