Update app.py

app.py

@@ -1,28 +1,43 @@
+import gradio as gr
 import cv2
+import pandas as pd
 import os
-import glob
 
-def extract_frames_from_videos(input_folder, output_folder):
+def extract_frames(video_path, csv_path):
+    output_folder = "extracted_frames"
     os.makedirs(output_folder, exist_ok=True)
-    video_files = glob.glob(os.path.join(input_folder, "*.mp4"))  # Modify if using other formats
     
-    image_count = 0  # To maintain sequential numbering across videos
+    # Read timestamps from CSV
+    df = pd.read_csv(csv_path)
     
-    for video_file in video_files:
-        cap = cv2.VideoCapture(video_file)
+    cap = cv2.VideoCapture(video_path)
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    
+    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)
         success, frame = cap.read()
         
-        while success:
-            image_path = os.path.join(output_folder, f"{image_count}.jpg")
+        if success:
+            image_path = os.path.join(output_folder, f"frame_{index}.jpg")
             cv2.imwrite(image_path, frame)
-            image_count += 1
-            success, frame = cap.read()
-        
-        cap.release()
+            extracted_images.append(image_path)
     
-    print(f"Extraction complete. {image_count} images saved in {output_folder}")
+    cap.release()
+    return extracted_images
+
+gui = gr.Interface(
+    fn=extract_frames,
+    inputs=[
+        gr.Video(label="Upload Video"),
+        gr.File(label="Upload CSV with timestamps")
+    ],
+    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."
+)
 
-# Example usage
-input_folder = "videos"  # Change this to the path where your videos are stored
-output_folder = "output_images"
-extract_frames_from_videos(input_folder, output_folder)
+gui.launch()