Update services/video_service.py

services/video_service.py

@@ -4,26 +4,25 @@ import cv2
 import random
 import os
 
-VIDEO_FOLDER = "data/"
-VIDEO_FILES = [
-    "drone_day.mp4",
-    "night_intrusion.mp4",
-    "thermal_hotspot.mp4",
-    "shadow_dust_issue.mp4",
-    "alert_response.mp4",
-]
+VIDEO_FOLDER = "data"
 
-def get_video_frame():
-    """Randomly pick a video and yield frames one by one."""
-    video_path = os.path.join(VIDEO_FOLDER, random.choice(VIDEO_FILES))
-    cap = cv2.VideoCapture(video_path)
+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")]
 
+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()
         if not ret:
+            # Restart when video ends
             cap.release()
-            # Once finished, pick another random video and continue
-            video_path = os.path.join(VIDEO_FOLDER, random.choice(VIDEO_FILES))
             cap = cv2.VideoCapture(video_path)
             continue
-        yield frame
+        yield frame, os.path.basename(video_path)