Update app.py

app.py

@@ -340,40 +340,40 @@ def process_video(
         callback=callback,
     )
 
-    # Ghi lại video bằng OpenCV để đảm bảo fps/duration hợp lệ (tránh NaN:NaN trên thanh thời gian)
-    cap = cv2.VideoCapture(output_path)
-    if not cap.isOpened():
-        # Nếu không mở được thì trả về file gốc (Gradio vẫn có thể play dạng blob)
-        return output_path
-
-    fps = cap.get(cv2.CAP_PROP_FPS)
-    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-
-    # Nếu fps không hợp lệ thì đặt mặc định
-    if fps is None or fps <= 0 or np.isnan(fps):
-        fps = 25.0
-
-    fixed_path = output_path.replace(".mp4", "_fixed.mp4")
-    fourcc = cv2.VideoWriter_fourcc(*"mp4v")
-    writer = cv2.VideoWriter(fixed_path, fourcc, fps, (width, height))
-
-    while True:
-        ok, frame = cap.read()
-        if not ok:
-            break
-        writer.write(frame)
-
-    cap.release()
-    writer.release()
-
-    # Xoá file tạm gốc, dùng file đã fix
+    # (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)
-    except OSError:
-        pass
+        output_path = final_path
 
-    return fixed_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:
@@ -403,7 +403,7 @@ with gr.Blocks(title="Nhận dạng phương tiện giao thông", theme=gr.theme
                 "### ✅ Kết quả đã xử lý\n"
                 "Hiển thị và thống kê số lượng theo lớp."
             )
-            video_output = gr.Video(label="Video đã xử lý")
+            video_output = gr.Video(label="Video đã xử lý", format="mp4")
 
     # Tùy chọn cấu hình
     with gr.Accordion("⚙️ Tùy chọn nâng cao", open=False):