Update app.py

app.py

@@ -1,43 +1,34 @@
 import gradio as gr
 import cv2
-import pandas as pd
 import os
 
-def extract_frames(video_path, csv_path):
+def extract_frames(video_path):
     output_folder = "extracted_frames"
     os.makedirs(output_folder, exist_ok=True)
     
-    # Read timestamps from CSV
-    df = pd.read_csv(csv_path)
-    
     cap = cv2.VideoCapture(video_path)
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    
+    frame_count = 0
     extracted_images = []
     
-    for index, row in df.iterrows():
-        timestamp = row['start']  # Using 'start' column for frame extraction
-        frame_number = int(timestamp * fps)
-        cap.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
+    while True:
         success, frame = cap.read()
+        if not success:
+            break
         
-        if success:
-            image_path = os.path.join(output_folder, f"frame_{index}.jpg")
-            cv2.imwrite(image_path, frame)
-            extracted_images.append(image_path)
+        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()
     return extracted_images
 
 gui = gr.Interface(
     fn=extract_frames,
-    inputs=[
-        gr.Video(label="Upload Video"),
-        gr.File(label="Upload CSV with timestamps")
-    ],
+    inputs=gr.Video(label="Upload Video"),
     outputs=gr.Gallery(label="Extracted Frames"),
-    title="Frame Extractor from Video",
-    description="Upload a video and a CSV file with timestamps to extract specific frames."
+    title="Video Frame Extractor",
+    description="Upload a video to extract all frames sequentially."
 )
 
-gui.launch()
+gui.launch()