Update services/video_service.py

services/video_service.py

@@ -2,60 +2,41 @@ import os
 import random
 import cv2
 
-# Path to the directory containing anomaly videos
-VIDEO_DIR = "sample_videos"
+VIDEO_DIR = "data"  # Corrected to 'data' folder
 
-# Supported video extensions
-VIDEO_EXTENSIONS = ['.mp4', '.avi', '.mov']
+# Ensure the folder exists
+os.makedirs(VIDEO_DIR, exist_ok=True)
 
-# Load all video file paths once at startup
+# List available video files
 video_files = [
     os.path.join(VIDEO_DIR, file)
     for file in os.listdir(VIDEO_DIR)
-    if any(file.lower().endswith(ext) for ext in VIDEO_EXTENSIONS)
+    if file.lower().endswith((".mp4", ".avi", ".mov"))
 ]
 
 if not video_files:
-    raise ValueError(f"No valid videos found in {VIDEO_DIR}. Please upload videos first.")
+    raise FileNotFoundError(
+        f"No videos found in '{VIDEO_DIR}'. Please upload .mp4, .avi, or .mov videos."
+    )
 
-# Random state for reproducibility if needed
-random_state = random.Random()
+current_video_idx = 0
+video_capture = None
 
-# Global frame iterator and capture
-current_video = None
-frame_iterator = None
-
-def open_new_video():
-    global current_video, frame_iterator
-    video_path = random_state.choice(video_files)
-    current_video = cv2.VideoCapture(video_path)
-
-    if not current_video.isOpened():
-        raise IOError(f"Cannot open video: {video_path}")
-
-    frame_iterator = iter_video_frames(current_video)
+def get_random_video_frame():
+    global video_capture, current_video_idx
 
-def iter_video_frames(video_cap):
-    while True:
-        ret, frame = video_cap.read()
-        if not ret:
-            break
-        yield frame
+    if video_capture is None or not video_capture.isOpened():
+        selected_video = video_files[current_video_idx]
+        video_capture = cv2.VideoCapture(selected_video)
 
-    video_cap.release()
+    ret, frame = video_capture.read()
 
-# Main function to get next frame
+    if not ret:
+        # Move to next video
+        current_video_idx = (current_video_idx + 1) % len(video_files)
+        selected_video = video_files[current_video_idx]
+        video_capture.release()
+        video_capture = cv2.VideoCapture(selected_video)
+        ret, frame = video_capture.read()
 
-def get_random_video_frame():
-    global current_video, frame_iterator
-
-    if frame_iterator is None:
-        open_new_video()
-
-    try:
-        frame = next(frame_iterator)
-        return frame
-    except StopIteration:
-        open_new_video()
-        frame = next(frame_iterator)
-        return frame
+    return frame