Update app.py

app.py

@@ -1,12 +1,15 @@
 import gradio as gr
 import cv2
 import os
+import shutil
+import zipfile
 
 def extract_frames(video_path):
     output_folder = "extracted_frames"
     os.makedirs(output_folder, exist_ok=True)
     
     cap = cv2.VideoCapture(video_path)
+    fps = cap.get(cv2.CAP_PROP_FPS)
     frame_count = 0
     extracted_images = []
     
@@ -15,20 +18,26 @@ def extract_frames(video_path):
         if not success:
             break
         
-        image_path = os.path.join(output_folder, f"frame_{frame_count}.jpg")
+        image_path = os.path.join(output_folder, f"frame_{int(frame_count/fps)}.jpg")
         cv2.imwrite(image_path, frame)
         extracted_images.append(image_path)
         frame_count += 1
     
     cap.release()
-    return extracted_images
+    
+    zip_filename = "extracted_frames.zip"
+    with zipfile.ZipFile(zip_filename, 'w') as zipf:
+        for img in extracted_images:
+            zipf.write(img, os.path.basename(img))
+    
+    return zip_filename
 
 gui = gr.Interface(
     fn=extract_frames,
     inputs=gr.Video(label="Upload Video"),
-    outputs=gr.Gallery(label="Extracted Frames"),
+    outputs=gr.File(label="Download Extracted Frames as ZIP"),
     title="Video Frame Extractor",
-    description="Upload a video to extract all frames sequentially."
+    description="Upload a video to extract all frames sequentially and download as a ZIP file."
 )
 
 gui.launch()