Update app.py

app.py

@@ -111,7 +111,27 @@ def video_vision(video_input_path, prompt):
             seg_frame = visualize(pred_mask, image_paths[frame_idx], temp_dir)
             seg_frames.append(seg_frame)
 
-        return result['prediction'], seg_frames
+        output_video = "output_video.mp4"
+
+        # Read the first image to get the size (resolution)
+        frame = cv2.imread(seg_fraes[0])
+        height, width, layers = frame.shape
+
+        # Define the video codec and create VideoWriter object
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Codec for MP4
+        video = cv2.VideoWriter(output_video, fourcc, 24.0, (width, height))
+
+        # Iterate over the image paths and write to the video
+        for img_path in images:
+            frame = cv2.imread(img_path)
+            video.write(frame)
+
+        # Release the video writer
+        video.release()
+
+        print(f"Video created successfully at {output_video}")
+
+        return result['prediction'], output_video
             
     else:
         return result['prediction'], None
@@ -164,7 +184,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
                         submit_video_btn = gr.Button("Submit", scale=1)
                 with gr.Column():
                     vid_output_res = gr.Textbox(label="Response")
-                    output_video = gr.Gallery(label="Segmentation")
+                    output_video = gr.Video(label="Segmentation")
             
             submit_video_btn.click(
                 fn = video_vision,