Update app.py

app.py

@@ -4,31 +4,38 @@ from ultralytics import YOLO
 
 # ── Config ─────────────────────────────────────────────
 MODEL_PATH = "yolov8n.pt"   # peso pre-entrenado (clase “person”)
-CONF_THRES = 0.3
-LINE_RATIO = 0.5            # línea virtual al 50 % de la altura
+CONF_THRES = 0.30           # confianza mínima
+LINE_RATIO = 0.50           # posición de la línea virtual (50 % de la altura)
 # ───────────────────────────────────────────────────────
 
-# carga una sola vez
+# modelo — se carga una sola vez
 model = YOLO(MODEL_PATH)
 
-# estado global
+# estado global del contador
 memory, in_count, out_count = {}, 0, 0
 
 
 def count_people(frame_rgb):
-    """Procesa un frame RGB -> devuelve frame RGB anotado + texto contador."""
+    """
+    ▸ Recibe un frame RGB (numpy) desde la webcam.
+    ▸ Devuelve el frame anotado (RGB) y el texto del contador.
+    """
     global memory, in_count, out_count
 
     if frame_rgb is None:
         return None, ""
 
+    # ── convertir RGB ➜ BGR (OpenCV/YOLO) ──────────────
     frame_bgr = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2BGR)
     h, w = frame_bgr.shape[:2]
     line_y = int(h * LINE_RATIO)
 
-    # detección + tracking
-    results = model.track(frame_bgr, classes=[0], conf=CONF_THRES,
-                          persist=True, verbose=False)
+    # ── detección + tracking (ByteTrack interno) ───────
+    results = model.track(frame_bgr,
+                          classes=[0],        # solo “person”
+                          conf=CONF_THRES,
+                          persist=True,
+                          verbose=False)
 
     annotated = frame_bgr.copy()
     cv2.line(annotated, (0, line_y), (w, line_y), (0, 255, 255), 2)
@@ -40,9 +47,9 @@ def count_people(frame_rgb):
             tid = int(box.id[0]) if box.id is not None else -1
 
             prev_cy = memory.get(tid, cy)
-            if prev_cy < line_y <= cy:        # entra
+            if prev_cy < line_y <= cy:        # cruza de arriba → abajo (ENTRA)
                 in_count += 1
-            elif prev_cy > line_y >= cy:      # sale
+            elif prev_cy > line_y >= cy:      # cruza de abajo → arriba (SALE)
                 out_count += 1
             memory[tid] = cy
 
@@ -54,16 +61,19 @@ def count_people(frame_rgb):
     total = in_count - out_count
     label = f"In: {in_count} | Out: {out_count} | Ocupación: {total}"
 
+    # ── devolver en RGB para que Gradio lo muestre bien ─
     annotated_rgb = cv2.cvtColor(annotated, cv2.COLOR_BGR2RGB)
     return annotated_rgb, label
 
 
 def reset_counts():
+    """Resetea contador y memoria."""
     global memory, in_count, out_count
     memory, in_count, out_count = {}, 0, 0
     return None, ""
 
 
+# ── Interfaz Gradio ────────────────────────────────────
 with gr.Blocks(title="Contador de personas (entrada única)") as demo:
     gr.Markdown("# Contador de personas (entrada única)")
 
@@ -74,9 +84,11 @@ with gr.Blocks(title="Contador de personas (entrada única)") as demo:
 
     btn_clear = gr.Button("Limpiar")
 
-    # 💡 la clave: inputs=[cam]
-    cam.stream(count_people, inputs=[cam], outputs=[out_img, out_lbl])
-    btn_clear.click(reset_counts, outputs=[out_img, out_lbl])
+    # conectar streaming y botón
+    cam.stream(fn=count_people, inputs=[cam], outputs=[out_img, out_lbl])
+    btn_clear.click(fn=reset_counts, outputs=[out_img, out_lbl])
+
+# ── Cola y lanzamiento ─────────────────────────────────
+demo.queue(concurrency_count=2, max_size=60)  # 2 hilos, cola máx 60 frames
+demo.launch()
 
-if __name__ == "__main__":
-    demo.launch(max_threads=2, max_queue=60)  # cola chica para que no se tape