Update app.py

app.py

@@ -29,6 +29,19 @@ from sam2.build_sam import build_sam2_video_predictor
 
 from moviepy.editor import ImageSequenceClip
 
+def get_video_fps(video_path):
+    # Open the video file
+    cap = cv2.VideoCapture(video_path)
+    
+    if not cap.isOpened():
+        print("Error: Could not open video.")
+        return None
+    
+    # Get the FPS of the video
+    fps = cap.get(cv2.CAP_PROP_FPS)
+
+    return fps
+
 def preprocess_image(image):
     return image, gr.State([]), gr.State([]), image, gr.State()
 
@@ -258,7 +271,7 @@ def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_
 
     return "output_first_frame.jpg", frame_names, inference_state
 
-def propagate_to_all(checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type):   
+def propagate_to_all(video_in, video_incheckpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type):   
     #### PROPAGATION ####
     sam2_checkpoint, model_cfg = load_model(checkpoint)
     predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
@@ -311,7 +324,8 @@ def propagate_to_all(checkpoint, stored_inference_state, stored_frame_names, vid
         return gr.update(value=jpeg_images, visible=True), gr.update(visible=False, value=None)
     elif vis_frame_type == "render":
         # Create a video clip from the image sequence
-        fps = 24  # Frames per second
+        original_fps = get_video_fps(video_in)
+        fps = original_fps  # Frames per second
         clip = ImageSequenceClip(jpeg_images, fps=fps)
         # Write the result to a file
         final_vid_output_path = "output_video.mp4"
@@ -353,7 +367,7 @@ with gr.Blocks() as demo:
             with gr.Column():
                 output_result = gr.Image()
                 with gr.Row():
-                    vis_frame_type = gr.Radio(choices=["check", "render"], value="render", scale=2)
+                    vis_frame_type = gr.Radio(choices=["check", "render"], value="check", scale=2)
                     propagate_btn = gr.Button("Propagate", scale=1)
                 output_propagated = gr.Gallery(visible=False)
                 output_video = gr.Video(visible=False)
@@ -388,7 +402,7 @@ with gr.Blocks() as demo:
 
     propagate_btn.click(
         fn = propagate_to_all,
-        inputs = [checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type],
+        inputs = [video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type],
         outputs = [output_propagated, output_video]
     )