Update app.py

app.py

@@ -1,13 +1,37 @@
+from video2images import Video2Images
+import os
 import gradio as gr
+from PIL import Image
+import shutil
 
-def video_identity(video):
-    return video
+def capture_frames(video_filepath, start_time=None, end_time=None, capture_rate=None, save_format=".jpg", out_dir=None):
+    # Create a Video2Images object
+    v2i = Video2Images(video_filepath, start_time=start_time, end_time=end_time, capture_rate=capture_rate, save_format=save_format, out_dir=out_dir)
+    # Extract frames
+    v2i.extract()
+    # Get paths of captured frames
+    frame_paths = v2i.get_frames_path()
+    # Load frames as PIL images
+    frames = [Image.open(frame_path) for frame_path in frame_paths]
+    return frames
 
+def video_to_gallery(video_path):
+    # Temporary directory to store frames
+    temp_dir = "temp_frames"
+    os.makedirs(temp_dir, exist_ok=True)
+    # Copy video file to temporary directory
+    video_file_path = os.path.join(temp_dir, "uploaded_video.mp4")
+    shutil.copyfile(video_path, video_file_path)
+    # Capture frames from video
+    frames = capture_frames(video_file_path, out_dir=temp_dir)
+    # Remove temporary directory
+    shutil.rmtree(temp_dir)
+    return video_file_path
 
-demo = gr.Interface(video_identity, 
-                    gr.Video(), 
-                    "playable_video", 
-                    )
+inputs = gr.Video(label="Upload Video")
+outputs = gr.Gallery(label="Frames")
 
-if __name__ == "__main__":
-    demo.launch()
+title = "Video to Image Gallery"
+description = "Upload a video and get its frames displayed as a gallery of images."
+
+gr.Interface(fn=video_to_gallery, inputs=inputs, outputs=outputs, title=title, description=description).launch()