Update services/video_service.py

services/video_service.py

@@ -1,32 +1,35 @@
-import cv2
+# services/video_service.py
+
 import os
+import cv2
 
-VIDEO_DIR = "data"  # All your videos are stored under 'data/' folder
-video_files = sorted([
-    os.path.join(VIDEO_DIR, f)
-    for f in os.listdir(VIDEO_DIR)
-    if f.lower().endswith((".mp4", ".avi", ".mov"))
-])
+VIDEO_DIR = "data"  # Your videos are in /data
+video_files = sorted(
+    [os.path.join(VIDEO_DIR, f) for f in os.listdir(VIDEO_DIR) if f.lower().endswith(('.mp4', '.avi', '.mov'))]
+)
 
-video_index = 0
-cap = None
+current_video_idx = 0
+video_capture = None
 
 def get_next_video_frame():
-    global cap
-    global video_index
+    global current_video_idx
+    global video_capture
 
     if not video_files:
-        raise RuntimeError("No videos found in the data directory.")
+        raise Exception("No video files found in 'data' directory!")
 
-    if cap is None or not cap.isOpened():
-        cap = cv2.VideoCapture(video_files[video_index])
+    if video_capture is None:
+        if current_video_idx >= len(video_files):
+            current_video_idx = 0  # restart from beginning
+        video_capture = cv2.VideoCapture(video_files[current_video_idx])
 
-    ret, frame = cap.read()
+    ret, frame = video_capture.read()
 
     if not ret:
-        cap.release()
-        video_index = (video_index + 1) % len(video_files)
-        cap = cv2.VideoCapture(video_files[video_index])
-        ret, frame = cap.read()
+        # End of current video, move to next
+        video_capture.release()
+        video_capture = None
+        current_video_idx += 1
+        return get_next_video_frame()  # recursively call
 
     return frame