video file

app.py

@@ -167,34 +167,17 @@ def print_image_info(img):
         print(f"{key}: {value}")
 
 def extract_frames(video_path):
-    """
-    Extract frames from a video file
-    
-    Args:
-        video_path (str): Path to the video file
-    
-    Returns:
-        list: List of frames as PIL Image objects
-    """
-    frames = []
-    print(f"video path {video_path}")
+    # Open the video file
     cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened():
-        print("Error: Could not open video file.")
-        return []
-    
-    while True:
-        ret, frame = cap.read()
-        if not ret:
-            break
-        
-        # Convert OpenCV BGR to RGB
+    frames = []
+    success, frame = cap.read()
+    while success:
+        # Convert frame from BGR (OpenCV default) to RGB
         frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        
-        # Convert to PIL Image
-        pil_image = Image.fromarray(frame_rgb)
-        frames.append(pil_image)
-    
+        # Convert the numpy array (frame) to a PIL Image
+        pil_frame = Image.fromarray(frame_rgb)
+        frames.append(pil_frame)
+        success, frame = cap.read()
     cap.release()
     return frames