Update services/video_service.py

services/video_service.py

@@ -1,28 +1,61 @@
-# services/video_service.py
-
-import cv2
-import random
 import os
+import random
+import cv2
 
-VIDEO_FOLDER = "data"
+# Path to the directory containing anomaly videos
+VIDEO_DIR = "sample_videos"
 
-def list_video_files():
-    """List all video files inside the data/ folder."""
-    return [os.path.join(VIDEO_FOLDER, f) for f in os.listdir(VIDEO_FOLDER) if f.endswith(".mp4")]
+# Supported video extensions
+VIDEO_EXTENSIONS = ['.mp4', '.avi', '.mov']
 
-def get_random_video_frame():
-    """Randomly pick a video from the folder and stream frames from it."""
-    video_files = list_video_files()
-    if not video_files:
-        raise FileNotFoundError("No video files found in the 'data/' folder.")
-    
-    video_path = random.choice(video_files)
-    cap = cv2.VideoCapture(video_path)
-    while cap.isOpened():
-        ret, frame = cap.read()
+# Load all video file paths once at startup
+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 not video_files:
+    raise ValueError(f"No valid videos found in {VIDEO_DIR}. Please upload videos first.")
+
+# Random state for reproducibility if needed
+random_state = random.Random()
+
+# 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 iter_video_frames(video_cap):
+    while True:
+        ret, frame = video_cap.read()
         if not ret:
-            # Restart when video ends
-            cap.release()
-            cap = cv2.VideoCapture(video_path)
-            continue
-        yield frame, os.path.basename(video_path)
+            break
+        yield frame
+
+    video_cap.release()
+
+# Main function to get next frame
+
+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