Update app.py

app.py

@@ -1,4 +1,6 @@
 import os
+import shutil
+import subprocess
 import tempfile
 import cv2
 import numpy as np
@@ -331,18 +333,51 @@ def process_video(
 
         return annotator_frame
 
-    # Tạo file tạm để lưu video output (đảm bảo file không bị khóa trên Windows)
-    fd, output_path = tempfile.mkstemp(suffix=".mp4")
-    os.close(fd)
-
+    output_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
     sv.process_video(
         source_path=video_path,
         target_path=output_path,
-        callback=callback,
+        callback=callback
     )
 
-    # Chuẩn hóa path về dạng forward slash cho Gradio/Trình duyệt
-    return output_path.replace("\\", "/")
+    # Re-encode sang H.264 để video dễ xem / tải về trên web
+    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,
+        )
+        try:
+            os.remove(output_path)
+        except OSError:
+            pass
+        output_path = final_path
+
+    return output_path
 
 
 with gr.Blocks(title="Nhận dạng phương tiện giao thông", theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray")) as demo: