Update app.py

app.py

@@ -27,6 +27,8 @@ import matplotlib.pyplot as plt
 from PIL import Image, ImageFilter
 from sam2.build_sam import build_sam2_video_predictor
 
+from moviepy.editor import ImageSequenceClip
+
 def preprocess_image(image):
     return image, gr.State([]), gr.State([]), image, gr.State()
 
@@ -303,10 +305,18 @@ def propagate_to_all(checkpoint, stored_inference_state, stored_frame_names, vid
         # Close the plot
         plt.close()
 
+    
+
     if vis_frame_type == "check":
         return gr.update(value=jpeg_images, visible=True), gr.update(visible=False, value=None)
     elif vis_frame_type == "render":
-        return gr.update(visible=False, value=None), gr.update(value=final_vid, visible=True)
+        # Create a video clip from the image sequence
+        fps = 24  # Frames per second
+        clip = ImageSequenceClip(jpeg_images, fps=fps)
+        # Write the result to a file
+        final_vid_output_path = "output_video.mp4"
+        video.write_videofile(output_path, codec='libx264')
+        return gr.update(visible=False, value=None), gr.update(value=final_vid_output_path, visible=True)
     
 
 with gr.Blocks() as demo: