Update services/video_service.py

services/video_service.py

@@ -1,9 +1,31 @@
 import cv2
+import random
+import os
 
-def get_video_frame(video_path):
-    cap = cv2.VideoCapture(video_path)
-    while cap.isOpened():
-        ret, frame = cap.read()
-        if not ret:
-            break
-        yield frame
+# List of all videos available for cycling
+VIDEO_FOLDER = "data"
+VIDEO_FILES = [
+    "drone_day.mp4",
+    "thermal_hotspot.mp4",
+    "shadow_dust_issue.mp4",
+    "night_intrusion.mp4",
+    "alert_response.mp4"
+]
+
+video_paths = [os.path.join(VIDEO_FOLDER, video) for video in VIDEO_FILES]
+
+def get_random_video_frame():
+    while True:
+        video_path = random.choice(video_paths)
+        cap = cv2.VideoCapture(video_path)
+
+        if not cap.isOpened():
+            continue  # skip invalid file
+
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+            yield frame, os.path.basename(video_path)
+
+        cap.release()