fixing camera n1235

requirements.txt

@@ -14,4 +14,6 @@ ultralytics
 av
 aiortc
 requests
-tqdm
+tqdm
+imageio
+imageio-ffmpeg

src/streamlit_app.py

@@ -16,6 +16,7 @@ import warnings
 import requests
 import threading
 import av
+import imageio.v2 as imageio
 from streamlit_webrtc import webrtc_streamer, VideoProcessorBase
 
 # Suppress WebRTC/asyncio warnings and errors
@@ -327,6 +328,27 @@ def process_video(video_path, model, conf_threshold, frame_skip=1):
 
         return rendered_frame
 
+    def _reset_stats():
+        nonlocal processed_frames, detection_counts
+        processed_frames = 0
+        detection_counts = {}
+        progress.progress(0)
+
+    def _run_loop(write_frame):
+        cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
+        frame_index = 0
+        while True:
+            ret, frame = cap.read()
+            if not ret:
+                break
+
+            frame_index += 1
+            if frame_skip > 1 and frame_index % frame_skip != 0:
+                continue
+
+            rendered_frame = _process_frame(frame)
+            write_frame(rendered_frame)
+
     try:
         container = av.open(output_path, mode="w")
         try:
@@ -344,22 +366,7 @@ def process_video(video_path, model, conf_threshold, frame_skip=1):
         stream.height = height
         stream.pix_fmt = "yuv420p"
 
-        frame_index = 0
-        while True:
-            ret, frame = cap.read()
-            if not ret:
-                break
-
-            frame_index += 1
-            if frame_skip > 1 and frame_index % frame_skip != 0:
-                continue
-
-            rendered_frame = _process_frame(frame)
-
-            video_frame = av.VideoFrame.from_ndarray(rendered_frame, format="bgr24")
-            for packet in stream.encode(video_frame):
-                container.mux(packet)
-
+        _run_loop(lambda rendered_frame: [container.mux(p) for p in stream.encode(av.VideoFrame.from_ndarray(rendered_frame, format="bgr24"))])
         for packet in stream.encode():
             container.mux(packet)
     except Exception as e:
@@ -369,24 +376,51 @@ def process_video(video_path, model, conf_threshold, frame_skip=1):
             container.close()
 
     if error_message:
+        _reset_stats()
         try:
-            fourcc = cv2.VideoWriter_fourcc(*"mp4v")
-            writer = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
-
-            cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
-            frame_index = 0
-            while True:
-                ret, frame = cap.read()
-                if not ret:
-                    break
-
-                frame_index += 1
-                if frame_skip > 1 and frame_index % frame_skip != 0:
-                    continue
+            imageio_path = output_path.replace(".webm", ".mp4")
+            writer = imageio.get_writer(
+                imageio_path,
+                fps=fps,
+                codec="libx264",
+                ffmpeg_params=["-pix_fmt", "yuv420p"],
+                format="FFMPEG"
+            )
+            _run_loop(lambda rendered_frame: writer.append_data(cv2.cvtColor(rendered_frame, cv2.COLOR_BGR2RGB)))
+            writer.close()
+            output_path = imageio_path
+            codec_name = "libx264"
+            output_mime = "video/mp4"
+            error_message = None
+        except Exception as e:
+            error_message = f"ImageIO H.264 encode failed: {e}"
 
-                rendered_frame = _process_frame(frame)
-                writer.write(rendered_frame)
+    if error_message:
+        _reset_stats()
+        try:
+            imageio_path = output_path.replace(".mp4", ".webm")
+            writer = imageio.get_writer(
+                imageio_path,
+                fps=fps,
+                codec="libvpx",
+                ffmpeg_params=["-pix_fmt", "yuv420p"],
+                format="FFMPEG"
+            )
+            _run_loop(lambda rendered_frame: writer.append_data(cv2.cvtColor(rendered_frame, cv2.COLOR_BGR2RGB)))
+            writer.close()
+            output_path = imageio_path
+            codec_name = "libvpx"
+            output_mime = "video/webm"
+            error_message = None
+        except Exception as e:
+            error_message = f"ImageIO WebM encode failed: {e}"
 
+    if error_message:
+        _reset_stats()
+        try:
+            fourcc = cv2.VideoWriter_fourcc(*"mp4v")
+            writer = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+            _run_loop(lambda rendered_frame: writer.write(rendered_frame))
             writer.release()
             error_message = None
             codec_name = "mp4v"