Update app.py

app.py

@@ -24,8 +24,8 @@ line_zone_annotator = sv.LineZoneAnnotator(
 )
 
 
-def process_video(video_path, orientation="Horizontal"):
-    """Process video: detect objects and return annotated video."""
+def process_video(video_path, orientation="Ngang", frame_skip=1):
+    """Xử lý video: nhận dạng phương tiện và trả về video đã annotate."""
     if video_path is None:
         return None
 
@@ -59,9 +59,17 @@ def process_video(video_path, orientation="Horizontal"):
     previous_positions = {}
     class_counts = {name: 0 for name in SELECTED_CLASS_NAMES}
     crossed_ids = set()
+    last_annotated_frame = None
 
     def callback(frame: np.ndarray, index: int) -> np.ndarray:
-        nonlocal previous_positions, class_counts, crossed_ids
+        nonlocal previous_positions, class_counts, crossed_ids, last_annotated_frame
+
+        try:
+            fs = max(1, int(frame_skip))
+        except Exception:
+            fs = 1
+        if fs > 1 and index % fs != 0:
+            return last_annotated_frame if last_annotated_frame is not None else frame
 
         results = model(frame, verbose=False)[0]
         detections = sv.Detections.from_ultralytics(results)
@@ -120,6 +128,8 @@ def process_video(video_path, orientation="Horizontal"):
                         (x0 + 10, y0 + 60 + i * 28),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 255, 255), 2)
 
+        last_annotated_frame = annotator_frame
+
         return annotator_frame
 
     output_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
@@ -133,23 +143,35 @@ def process_video(video_path, orientation="Horizontal"):
 
 with gr.Blocks(title="Object Detection", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🚗 Nhận dạng phương tiện (YOLOv8 + ByteTrack)")
-    gr.Markdown("Upload video, hệ thống sẽ nhận dạng phương tiện trong video.")
+    gr.Markdown("Tải lên video, hệ thống sẽ nhận dạng phương tiện trong video.")
 
     with gr.Row():
-        video_input = gr.Video(label="Video input")
-        video_output = gr.Video(label="Video output")
+        video_input = gr.Video(label="Video đầu vào")
+        video_output = gr.Video(label="Video đã xử lý")
 
     orientation_input = gr.Radio(
-        choices=["Horizontal", "Vertical"],
-        value="Horizontal",
-        label="Line orientation"
+        choices=["Ngang", "Dọc"],
+        value="Ngang",
+        label="Vị trí line"
     )
 
-    btn = gr.Button("▶️ Process Video")
-    btn.click(fn=process_video, inputs=[video_input, orientation_input], outputs=video_output)
+    frame_skip_input = gr.Slider(
+        minimum=1,
+        maximum=10,
+        step=1,
+        value=1,
+        label="Frame skip (1 = xử lý mọi frame)"
+    )
+
+    btn = gr.Button("▶️ Xử lý video")
+    btn.click(
+        fn=process_video,
+        inputs=[video_input, orientation_input, frame_skip_input],
+        outputs=video_output
+    )
 
     gr.Markdown("""
-    ### Author: Trần Hải Nam - 223332840
+    ### Sinh viên: Trần Hải Nam - 223332840
     """)
 
 if __name__ == "__main__":