Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 import cv2
 from ultralytics import YOLO
 import os
-import moviepy.editor as mp
+from moviepy.editor import VideoFileClip
 
-# 1. Khởi tạo mô hình (Hugging Face sẽ tự tải bản yolov8n.pt chuẩn)
+# 1. Khởi tạo mô hình
 model = YOLO('yolov8n.pt')
 
 def process_video(video_path):
@@ -28,7 +28,7 @@ def process_video(video_path):
     # Vạch đếm ở 60% chiều cao màn hình
     line_y = int(height * 0.6) 
 
-    # File tạm trung gian (OpenCV ghi định dạng mp4v)
+    # File tạm trung gian (định dạng mp4v)
     temp_output = 'temp_raw.mp4'
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(temp_output, fourcc, fps, (width, height))
@@ -38,7 +38,7 @@ def process_video(video_path):
         if not ret:
             break
 
-        # Thực hiện Tracking (persist=True như bản gốc của bạn)
+        # Thực hiện Tracking
         results = model.track(frame, classes=vehicle_class_ids, persist=True, verbose=False)
 
         # Vẽ vạch đếm màu đỏ
@@ -57,11 +57,11 @@ def process_video(video_path):
                 cx = int((x1 + x2) / 2)
                 cy = int((y1 + y2) / 2)
 
-                # Vẽ khung và tâm xe (Logic của bạn)
+                # Vẽ khung và tâm xe
                 cv2.rectangle(frame, (int(x1), int(y1)), (int(x2), int(y2)), (255, 255, 0), 2)
                 cv2.circle(frame, (cx, cy), 5, (0, 255, 0), -1) 
 
-                # Logic đếm qua vạch cải tiến
+                # Logic đếm qua vạch
                 if obj_id in track_history and obj_id not in counted_ids:
                     prev_cy = track_history[obj_id]
                     if (prev_cy < line_y and cy >= line_y) or (prev_cy > line_y and cy <= line_y):
@@ -71,7 +71,7 @@ def process_video(video_path):
                 
                 track_history[obj_id] = cy
 
-        # --- BẢNG THỐNG KÊ GÓC PHẢI (Thiết kế gốc của bạn) ---
+        # --- BẢNG THỐNG KÊ (Thiết kế gốc của bạn) ---
         total = sum(vehicle_count.values())
         overlay = frame.copy()
         box_w, box_h = 250, 160
@@ -92,19 +92,18 @@ def process_video(video_path):
     cap.release()
     out.release()
 
-    # --- CHUYỂN ĐỔI SANG CHUẨN H.264 ĐỂ XEM TRỰC TIẾP TRÊN WEB ---
+    # --- CHUYỂN ĐỔI SANG CHUẨN H.264 ĐỂ XEM TRỰC TIẾP ---
     final_output = 'result_display.mp4'
     try:
-        # Sử dụng moviepy để encode lại video
-        clip = mp.VideoFileClip(temp_output)
+        # Chuyển đổi định dạng để trình duyệt web có thể Play trực tiếp
+        clip = VideoFileClip(temp_output)
         clip.write_videofile(final_output, codec="libx264", audio=False)
         clip.close()
-        # Xóa file tạm cho nhẹ bộ nhớ server
         if os.path.exists(temp_output):
             os.remove(temp_output)
     except Exception as e:
         print(f"Lỗi encode video: {e}")
-        return temp_output # Trả về file gốc nếu lỗi
+        return temp_output
     
     return final_output
 
@@ -114,7 +113,7 @@ interface = gr.Interface(
     inputs=gr.Video(label="Tải video của bạn lên"),
     outputs=gr.Video(label="Kết quả đếm xe (Xem trực tiếp)"),
     title="🚗 Hệ thống Đếm Xe Thông Minh (YOLOv8)",
-    description="Ứng dụng tự động nhận diện và đếm số lượng xe qua vạch. Kết quả được tối ưu hóa để xem trực tiếp trên trình duyệt.",
+    description="Nhận diện và đếm số lượng xe qua vạch. Kết quả xem được trực tiếp trên web.",
 )
 
 if __name__ == "__main__":