02-ago: [proyecto] vigilancia - presencia por persona (YOLO) + zonas autorizadas/no autorizadas por camara en el aviso de Telegram

vigilancia/main.py

@@ -36,18 +36,31 @@ def cargar_identificador():
 def componer_alerta(resultados, cam_nombre, ts):
     conocidas = sorted({r["nombre"] for r in resultados if "nombre" in r})
     desconocidas = sum(1 for r in resultados if r.get("desconocido"))
+    zona = "autorizada" if cam_nombre in config.ZONAS_AUTORIZADAS else "NO autorizada"
+    sufijo = f" (zona {zona})"
 
     if conocidas and desconocidas:
         return (f"👧 {', '.join(conocidas)} | "
                 f"⚠️ {desconocidas} persona(s) desconocida(s) "
-                f"en {cam_nombre} - {ts}")
+                f"en {cam_nombre}{sufijo} - {ts}")
     if conocidas:
         return (f"👧 {', '.join(conocidas)} detectada(s) "
-                f"en {cam_nombre} - {ts}")
+                f"en {cam_nombre}{sufijo} - {ts}")
     if desconocidas:
-        return (f"⚠️ {desconocidas} persona(s) desconocida(s) "
-                f"en {cam_nombre} - {ts}")
-    return f"🚨 Movimiento en {cam_nombre} - {ts}"
+        return (f"🚨 {desconocidas} persona(s) en {cam_nombre}"
+                f"{sufijo} - {ts}")
+    return f"🚨 Presencia en {cam_nombre}{sufijo} - {ts}"
+
+
+def detectar_presencia(modelo, frame):
+    """True si hay al menos una persona en el frame (YOLO)."""
+    if modelo is None:
+        return True
+    try:
+        r = modelo.predict(frame, verbose=False, conf=0.4)
+        return any(int(b.cls) == 0 for b in r[0].boxes)
+    except Exception:
+        return True
 
 
 def dentro_de_franja_horaria(hora_actual=None) -> bool:
@@ -77,6 +90,14 @@ def main():
 
     ident = cargar_identificador()
 
+    modelo_presencia = None
+    try:
+        from ultralytics import YOLO
+        ruta_yolo = os.path.join(os.path.dirname(os.path.abspath(__file__)), "yolov8n.pt")
+        modelo_presencia = YOLO(ruta_yolo)
+    except Exception as e:
+        print(f"No se pudo cargar YOLO para presencia: {e}")
+
     print("Sistema de vigilancia iniciado. Franja activa: "
           f"{config.HORA_INICIO}:00 - {config.HORA_FIN}:00")
 
@@ -91,36 +112,46 @@ def main():
                 continue
 
             detector = detectores[i]
-            detectado = detector.hay_movimiento(frame)
+            movido = detector.hay_movimiento(frame)
 
             ahora = time.time()
-            if detectado and (ahora - ultima_alerta[i] > config.COOLDOWN):
-                ts = datetime.now().strftime("%Y%m%d_%H%M%S")
-                ruta = os.path.join(
-                    config.SNAPSHOT_DIR, f"snapshot_{cam.nombre}_{ts}.jpg"
-                )
+            if not movido:
+                continue
+            if (ahora - ultima_alerta[i]) <= config.COOLDOWN:
+                continue
 
-                texto = f"🚨 Movimiento en {cam.nombre} - {ts}"
-                if ident is not None:
-                    try:
-                        resultados = ident.analizar_frame(frame)
-                        if resultados:
-                            texto = componer_alerta(resultados, cam.nombre, ts)
-                    except Exception as e:
-                        print(f"[{cam.nombre}] Error identificando: {e}")
+            if not detectar_presencia(modelo_presencia, frame):
+                continue
 
-                cv2_ok = True
+            ts = datetime.now().strftime("%Y%m%d_%H%M%S")
+            ruta = os.path.join(
+                config.SNAPSHOT_DIR, f"snapshot_{cam.nombre}_{ts}.jpg"
+            )
+
+            texto = f"🚨 Presencia en {cam.nombre} - {ts}"
+            if ident is not None:
                 try:
-                    cv2.imwrite(ruta, frame)
+                    resultados = ident.analizar_frame(frame)
+                    if resultados:
+                        texto = componer_alerta(resultados, cam.nombre, ts)
                 except Exception as e:
-                    cv2_ok = False
-                    print(f"[{cam.nombre}] Error guardando snapshot: {e}")
+                    print(f"[{cam.nombre}] Error identificando: {e}")
+            else:
+                zona = "autorizada" if cam.nombre in config.ZONAS_AUTORIZADAS else "NO autorizada"
+                texto = f"🚨 Presencia en {cam.nombre} (zona {zona}) - {ts}"
 
-                if cv2_ok:
-                    enviar_telegram(ruta, texto)
-                disparar_webhook_alexa(texto)
-                ultima_alerta[i] = ahora
-                print(texto)
+            cv2_ok = True
+            try:
+                cv2.imwrite(ruta, frame)
+            except Exception as e:
+                cv2_ok = False
+                print(f"[{cam.nombre}] Error guardando snapshot: {e}")
+
+            if cv2_ok:
+                enviar_telegram(ruta, texto)
+            disparar_webhook_alexa(texto)
+            ultima_alerta[i] = ahora
+            print(texto)
 
         time.sleep(0.5)