Update services/video_service.py

services/video_service.py

@@ -6,7 +6,7 @@ VIDEO_DIR = "data"
 video_files = [os.path.join(VIDEO_DIR, file) for file in sorted(os.listdir(VIDEO_DIR)) if file.endswith((".mp4", ".avi"))]
 video_index = 0
 cap = None
-FRAME_SKIP = 5  # Increased frame skipping for faster processing
+FRAME_SKIP = 5
 
 def get_next_video_frame():
     global cap, video_index
@@ -16,6 +16,8 @@ def get_next_video_frame():
 
     if cap is None or not cap.isOpened():
         cap = cv2.VideoCapture(video_files[video_index])
+        if not cap.isOpened():
+            raise RuntimeError(f"Failed to open video file: {video_files[video_index]}")
 
     for _ in range(FRAME_SKIP):
         ret, frame = cap.read()
@@ -23,18 +25,15 @@ def get_next_video_frame():
             cap.release()
             video_index = (video_index + 1) % len(video_files)
             cap = cv2.VideoCapture(video_files[video_index])
+            if not cap.isOpened():
+                raise RuntimeError(f"Failed to open video file: {video_files[video_index]}")
             ret, frame = cap.read()
             if not ret:
                 raise RuntimeError(f"Cannot read video {video_files[video_index]}")
             break
 
-    # Resize frame to a smaller resolution for faster processing
-    frame = cv2.resize(frame, (320, 240))  # Reduced resolution
-    return frame
+    if frame is None or frame.size == 0:
+        raise RuntimeError("Empty frame received. Check camera or video file.")
 
-def reset_video_index():
-    global video_index, cap
-    video_index = 0
-    if cap is not None:
-        cap.release()
-    cap = None
+    frame = cv2.resize(frame, (320, 240))
+    return frame