Update video_processor.py

video_processor.py

@@ -1,52 +1,59 @@
 import cv2
 import os
-import tempfile
+import uuid
+import numpy as np
+from utils import extract_frames, save_video
 
-def process_video(video_path, clip_duration=10, target_fps=25):
-    """
-    Processes the input video:
-    - Trims the last `clip_duration` seconds
-    - Samples frames at `target_fps`
+TEMP_DIR = "temp_videos"
+os.makedirs(TEMP_DIR, exist_ok=True)
+
+def process_live_video():
+    # Simulated live stream (replace with real buffer system if needed)
+    live_source = "sample_videos/lbw_sample.mp4"
     
-    Returns:
-        frames: List of BGR frames
-        metadata: Dict with original FPS, trimmed duration, frame count
-    """
-    cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened():
-        raise IOError("Cannot open video file.")
-
-    orig_fps = cap.get(cv2.CAP_PROP_FPS)
+    cap = cv2.VideoCapture(live_source)
+    fps = cap.get(cv2.CAP_PROP_FPS)
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    duration = total_frames / orig_fps
+    duration = total_frames / fps
 
-    # Calculate starting point for last `clip_duration` seconds
-    start_time = max(0, duration - clip_duration)
-    start_frame = int(start_time * orig_fps)
+    # Trim last 10 seconds
+    trim_start = max(0, duration - 10)
+    start_frame = int(trim_start * fps)
 
-    # Set video to start point
     cap.set(cv2.CAP_PROP_POS_FRAMES, start_frame)
-
-    # Read and sample frames
     frames = []
-    frame_interval = int(orig_fps // target_fps) if target_fps < orig_fps else 1
-    frame_idx = 0
-
     while True:
         ret, frame = cap.read()
         if not ret:
             break
-        if frame_idx % frame_interval == 0:
-            frames.append(frame)
-        frame_idx += 1
+        frames.append(frame)
+    cap.release()
+
+    video_id = str(uuid.uuid4())
+    trimmed_path = os.path.join(TEMP_DIR, f"{video_id}_trimmed.mp4")
+    save_video(frames, trimmed_path, fps)
+
+    # Extract & Analyze
+    decision = "Analyzing..."  # Placeholder, will use lbw_detector later
+    return trimmed_path, decision
+
 
+def process_uploaded_video(video_file):
+    cap = cv2.VideoCapture(video_file)
+    fps = cap.get(cv2.CAP_PROP_FPS)
+
+    frames = []
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+        frames.append(frame)
     cap.release()
 
-    metadata = {
-        "original_fps": orig_fps,
-        "sampled_fps": target_fps,
-        "clip_duration": clip_duration,
-        "sampled_frames": len(frames)
-    }
+    video_id = str(uuid.uuid4())
+    saved_path = os.path.join(TEMP_DIR, f"{video_id}_upload.mp4")
+    save_video(frames, saved_path, fps)
 
-    return frames, metadata
+    # Extract & Analyze
+    decision = "Analyzing..."  # Placeholder for lbw_detector output
+    return saved_path, decision