Update services/video_service.py

services/video_service.py

@@ -1,30 +1,38 @@
-import os
-import random
 import cv2
+import os
 
+# Directory containing sample videos
 VIDEO_DIR = "data"
+video_files = sorted([
+    os.path.join(VIDEO_DIR, f)
+    for f in os.listdir(VIDEO_DIR)
+    if f.endswith(".mp4") or f.endswith(".avi") or f.endswith(".mov")
+])
 
-def get_random_video_frame():
-    video_files = [f for f in os.listdir(VIDEO_DIR) if f.endswith('.mp4')]
-    if not video_files:
-        raise Exception("No videos found in the data folder!")
-
-    video_path = os.path.join(VIDEO_DIR, random.choice(video_files))
-    cap = cv2.VideoCapture(video_path)
+# Globals
+current_video_index = 0
+current_cap = None
 
-    if not cap.isOpened():
-        raise Exception(f"Failed to open video: {video_path}")
+def get_next_video_frame():
+    global current_video_index, current_cap
 
-    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    if frame_count == 0:
-        raise Exception(f"No frames found in video: {video_path}")
+    # Initialize capture if none
+    if current_cap is None:
+        if not video_files:
+            raise RuntimeError("No videos found in the 'data' folder.")
+        current_cap = cv2.VideoCapture(video_files[current_video_index])
 
-    random_frame = random.randint(0, frame_count - 1)
-    cap.set(cv2.CAP_PROP_POS_FRAMES, random_frame)
-    ret, frame = cap.read()
-    cap.release()
+    # Read frame
+    ret, frame = current_cap.read()
 
     if not ret:
-        raise Exception(f"Failed to read frame at {random_frame} from {video_path}")
+        # Video ended, move to next
+        current_cap.release()
+        current_video_index = (current_video_index + 1) % len(video_files)
+        current_cap = cv2.VideoCapture(video_files[current_video_index])
+        ret, frame = current_cap.read()
+
+        if not ret:
+            raise RuntimeError("Failed to read frame from video.")
 
     return frame