Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import cv2
 import os
-import shutil
 import zipfile
 
 def extract_frames(video_path):
@@ -10,17 +9,19 @@ def extract_frames(video_path):
     
     cap = cv2.VideoCapture(video_path)
     fps = cap.get(cv2.CAP_PROP_FPS)
-    frame_count = 0
     extracted_images = []
+    frame_count = 0
     
     while True:
         success, frame = cap.read()
         if not success:
             break
         
-        image_path = os.path.join(output_folder, f"frame_{int(frame_count/fps)}.jpg")
-        cv2.imwrite(image_path, frame)
-        extracted_images.append(image_path)
+        if frame_count % int(fps / 60) == 0:  # Extract 60 frames per second
+            image_path = os.path.join(output_folder, f"frame_{frame_count}.jpg")
+            cv2.imwrite(image_path, frame)
+            extracted_images.append(image_path)
+        
         frame_count += 1
     
     cap.release()
@@ -37,7 +38,7 @@ gui = gr.Interface(
     inputs=gr.Video(label="Upload Video"),
     outputs=gr.File(label="Download Extracted Frames as ZIP"),
     title="Video Frame Extractor",
-    description="Upload a video to extract all frames sequentially and download as a ZIP file."
+    description="Upload a video to extract frames at 60 FPS and download as a ZIP file."
 )
 
 gui.launch()