Update app.py

app.py

@@ -1,41 +1,24 @@
-pip install opencv-python
-import cv2
-import os
+import subprocess
+import sys
 import gradio as gr
 
-def extract_clip(video_path, clip_start, clip_duration):
-    video = cv2.VideoCapture(video_path)
-    
-    success, frame = video.read()
-    count = 0
-    frames = []
-    while success:
-        current_time = video.get(cv2.CAP_PROP_POS_MSEC) / 1000
-        
-        # Seek to the desired starting point
-        if current_time < clip_start:
-            video.set(cv2.CAP_PROP_POS_MSEC, clip_start * 1000)
-            success, frame = video.read()
-            continue
-        
-        # Check duration after reaching the clip_start timestamp
-        elif current_time - clip_start <= clip_duration:
-            frames.append(frame)
-            
-            success, frame = video.read()
-            count += 1
-
-        else:
-            break
+def install_opencv():
+    try:
+        import cv2
+        print("OpenCV is already installed.")
+    except ImportError:
+        print("Installing OpenCV...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python"])
+        print("OpenCV installation completed.")
 
-    video.release()
-    return frames
+def extract_clip(video_path, clip_start, clip_duration):
+    # ... (rest of your extract_clip function)
 
 def display_frames(frames):
-    num_cols = 4  # Number of columns in the image grid
-    num_rows = (len(frames) + num_cols - 1) // num_cols
-    grid_image = cv2.hconcat([cv2.hconcat(frames[i:i+num_cols]) for i in range(0, len(frames), num_cols)])
-    return grid_image
+    # ... (rest of your display_frames function)
+
+# Install OpenCV if not already installed
+install_opencv()
 
 # Create Gradio interface
 video_input = gr.inputs.Video(type="file")