Update app.py

app.py

@@ -2,7 +2,9 @@ import os
 import shutil
 import subprocess
 import tempfile
+
 import cv2
+import imageio
 import numpy as np
 import gradio as gr
 import supervision as sv
@@ -38,13 +40,19 @@ def process_video(
     if isinstance(video_path, dict):
         video_path = video_path.get("path", video_path)
 
-    video_info = sv.VideoInfo.from_video_path(video_path)
+    cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        return None
+
+    fps = cap.get(cv2.CAP_PROP_FPS)
+    if fps is None or fps <= 0 or np.isnan(fps):
+        fps = 30
 
     byte_tracker = sv.ByteTrack(
         track_activation_threshold=0.25,
         lost_track_buffer=30,
         minimum_matching_threshold=0.8,
-        frame_rate=video_info.fps or 30,
+        frame_rate=fps,
         minimum_consecutive_frames=3
     )
     byte_tracker.reset()
@@ -334,44 +342,23 @@ def process_video(
         return annotator_frame
 
     output_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
-    sv.process_video(
-        source_path=video_path,
-        target_path=output_path,
-        callback=callback,
-    )
+    writer = imageio.get_writer(output_path, fps=fps)
 
-    # (Tuỳ chọn) Re-encode sang H.264 để trình duyệt/Gradio đọc tốt hơn
-    final_path = output_path.replace(".mp4", "_h264.mp4")
-    try:
-        import imageio_ffmpeg
-        ffmpeg_exe = imageio_ffmpeg.get_ffmpeg_exe()
-    except ImportError:
-        ffmpeg_exe = shutil.which("ffmpeg")
-
-    if ffmpeg_exe:
-        subprocess.run(
-            [
-                ffmpeg_exe,
-                "-y",
-                "-i",
-                output_path,
-                "-c:v",
-                "libx264",
-                "-preset",
-                "ultrafast",
-                "-crf",
-                "23",
-                "-movflags",
-                "+faststart",
-                "-pix_fmt",
-                "yuv420p",
-                final_path,
-            ],
-            stdout=subprocess.DEVNULL,
-            stderr=subprocess.DEVNULL,
-        )
-        os.remove(output_path)
-        output_path = final_path
+    index = 0
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+
+        annotated_frame = callback(frame, index)
+        index += 1
+
+        # imageio expects RGB frames
+        annotated_frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
+        writer.append_data(annotated_frame_rgb)
+
+    writer.close()
+    cap.release()
 
     return output_path