Update services/video_service.py

services/video_service.py

@@ -29,9 +29,21 @@ def preload_video(video_path: str) -> None:
         cap = cv2.VideoCapture(video_path)
         if not cap.isOpened():
             raise RuntimeError(f"Failed to open video: {video_path}. Ensure the file exists and is a supported format (.mp4, .avi).")
+
+        # Validate video properties
         total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
         fps = cap.get(cv2.CAP_PROP_FPS)
-        logging.info(f"Loaded video: {video_path}, Total Frames: {total_frames}, FPS: {fps}")
+        if total_frames <= 0:
+            raise RuntimeError(f"Invalid video: {video_path}. No frames detected.")
+        if fps <= 0:
+            raise RuntimeError(f"Invalid video: {video_path}. Invalid FPS.")
+
+        # Check video codec
+        fourcc = int(cap.get(cv2.CAP_PROP_FOURCC))
+        codec = "".join([chr((fourcc >> 8 * i) & 0xFF) for i in range(4)])
+        log_message = (f"Loaded video: {video_path}, Total Frames: {total_frames}, "
+                       f"FPS: {fps}, Codec: {codec}")
+        logging.info(log_message)
         current_frame_idx = 0
     except Exception as e:
         logging.error(f"Error loading video {video_path}: {str(e)}")
@@ -51,10 +63,12 @@ def reset_video_index() -> None:
     Reset the video frame index to the beginning.
     """
     global current_frame_idx
-    if cap is not None:
+    if cap is not None and cap.isOpened():
         cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
         current_frame_idx = 0
         logging.info("Reset video to start.")
+    else:
+        logging.warning("Cannot reset video: Video capture not initialized.")
 
 def get_next_video_frame() -> Optional[np.ndarray]:
     """
@@ -74,10 +88,11 @@ def get_next_video_frame() -> Optional[np.ndarray]:
         success, frame = cap.read()
         if not success or frame is None:
             # Video ended, reset to start
+            logging.info("Reached end of video, resetting to start.")
             reset_video_index()
             success, frame = cap.read()
             if not success or frame is None:
-                logging.warning("Reached end of video and failed to reset.")
+                logging.warning("Failed to retrieve frame after reset.")
                 return None
         current_frame_idx = int(cap.get(cv2.CAP_PROP_POS_FRAMES))
         logging.info(f"Retrieved frame {current_frame_idx}")