Update services/video_service.py

services/video_service.py

@@ -1,79 +1,87 @@
 import cv2
 import os
+import logging
 
-# Global variables
-cap = None
-frame_index = 0
-current_video_path = None
+# Setup logging
+logging.basicConfig(
+    filename="app.log",
+    level=logging.INFO,
+    format="%(asctime)s - %(levelname)s - %(message)s"
+)
 
-def preload_video(video_path="sample.mp4"):
+# Global video capture object
+video_cap = None
+current_frame_index = 0
+
+def preload_video(video_path):
     """
     Preload the video file.
     Args:
-        video_path (str): Path to the video file.
+        video_path: Path to the video file
     Returns:
-        str: Status message indicating success or failure.
+        str: Status message
     """
-    global cap, frame_index, current_video_path
+    global video_cap, current_frame_index
     try:
-        # Reset previous capture if any
-        if cap is not None:
-            cap.release()
-        
-        # Check if video file exists
         if not os.path.exists(video_path):
-            raise FileNotFoundError(f"Video file {video_path} not found. Please upload a video file named 'sample.mp4' or provide a different path.")
+            raise FileNotFoundError(f"Video file not found at {video_path}")
 
-        # Initialize video capture
-        cap = cv2.VideoCapture(video_path)
-        if not cap.isOpened():
-            raise RuntimeError(f"Failed to open video file {video_path}. Ensure the file is a valid video format (e.g., .mp4).")
+        video_cap = cv2.VideoCapture(video_path)
+        if not video_cap.isOpened():
+            raise RuntimeError("Failed to open video file.")
 
-        frame_index = 0
-        current_video_path = video_path
-        return f"Successfully loaded video: {video_path}"
+        current_frame_index = 0
+        total_frames = int(video_cap.get(cv2.CAP_PROP_FRAME_COUNT))
+        logging.info(f"Preloaded video: {video_path} with {total_frames} frames.")
+        return f"Successfully loaded video: {video_path} ({total_frames} frames)"
     except Exception as e:
-        cap = None
+        logging.error(f"Error preloading video: {str(e)}")
         return f"Error loading video: {str(e)}"
 
 def get_next_video_frame():
     """
     Get the next frame from the video.
     Returns:
-        numpy array: Video frame
-    Raises:
-        RuntimeError: If video is not preloaded or frame cannot be read.
+        numpy array: Frame if successful, None otherwise
     """
-    global cap, frame_index
-    if cap is None:
-        raise RuntimeError("Video not preloaded. Call preload_video() first.")
-    
-    ret, frame = cap.read()
-    if not ret:
-        # Loop the video
-        cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
-        frame_index = 0
-        ret, frame = cap.read()
+    global video_cap, current_frame_index
+    if video_cap is None or not video_cap.isOpened():
+        logging.error("Video capture not initialized.")
+        return None
+
+    try:
+        # Set frame position
+        video_cap.set(cv2.CAP_PROP_POS_FRAMES, current_frame_index)
+        ret, frame = video_cap.read()
         if not ret:
-            raise RuntimeError(f"Failed to read frame from video {current_video_path}.")
-    
-    frame_index += 1
-    return frame
+            # Loop back to the start
+            current_frame_index = 0
+            video_cap.set(cv2.CAP_PROP_POS_FRAMES, current_frame_index)
+            ret, frame = video_cap.read()
+            if not ret:
+                raise RuntimeError("Failed to read frame from video.")
+
+        current_frame_index += 1
+        logging.debug(f"Retrieved frame {current_frame_index} from video.")
+        return frame
+    except Exception as e:
+        logging.error(f"Error retrieving frame: {str(e)}")
+        return None
 
 def reset_video_index():
     """
-    Reset the video frame index.
+    Reset the frame index to the beginning.
     """
-    global frame_index
-    frame_index = 0
-    if cap is not None:
-        cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
+    global current_frame_index
+    current_frame_index = 0
+    logging.info("Reset video frame index to 0.")
 
 def release_video():
     """
     Release the video capture object.
     """
-    global cap
-    if cap is not None:
-        cap.release()
-        cap = None
+    global video_cap
+    if video_cap is not None:
+        video_cap.release()
+        video_cap = None
+        logging.info("Released video capture object.")