Update app.py

app.py

@@ -340,41 +340,40 @@ def process_video(
         callback=callback,
     )
 
-    # Re-encode sang H.264 để trình duyệt đọc đúng metadata (tránh NaN:NaN)
-    final_path = output_path.replace(".mp4", "_h264.mp4")
+    # 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
     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
+    except OSError:
+        pass
 
-    return output_path
+    return fixed_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: