Update app.py

app.py

@@ -1,25 +1,24 @@
-cimport gradio as gr
+import gradio as gr
 from video2images import Video2Images
-from PIL import Image
 import os
 
-def extract_frames(video_filepath, out_dir):
-    # Use a temporary filename in the current directory
-    out_filepath = os.path.join(os.getcwd(), "extracted_frames")
+def extract_frames(video_file):
+    # Temporary directory to store frames
+    temp_dir = "temp_frames"
+    os.makedirs(temp_dir, exist_ok=True)
+    # Save uploaded video to temporary directory
+    video_filepath = os.path.join(temp_dir, "uploaded_video.mp4")
+    video_file.save(video_filepath)
+    # Use the path of the uploaded video for extracting frames
+    out_filepath = os.path.join(temp_dir, "extracted_frames")
     # Create Video2Images object
     v2i = Video2Images(video_filepath, out_dir=out_filepath)
     # Extract frames
     v2i.start()
-    # Load extracted frames as PIL images
-    frames = []
-    for frame_filename in os.listdir(out_filepath):
-        frame_path = os.path.join(out_filepath, frame_filename)
-        frame_image = Image.open(frame_path)
-        frames.append(frame_image)
-    return frames  # Return list of PIL images
+    return out_filepath  # Return the directory where frames are saved
 
 inputs = gr.Video(label="Upload Video")
-outputs = gr.ImageGallery(label="Extracted Frames")
+outputs = gr.Textbox(label="Output Directory")
 
 title = "Extract Frames from Video"
 description = "Upload a video file and extract frames from it."