Update app.py

app.py

@@ -3,68 +3,63 @@ import requests
 import time
 import json
 
-def test_antena_remota(ip, usuario, password):
+def test_antena_vpn(ip, usuario, password):
     try:
-        # 1. Primero intentamos login para obtener cookie/token
         session = requests.Session()
         
-        # URL de login (común en equipos Ubiquiti)
-        login_url = f"http://{ip}/api/login"
-        
-        # Datos de login
-        login_data = {
-            "username": usuario,
-            "password": password
-        }
-        
-        # Headers comunes
+        # Headers para evitar bloqueos
         headers = {
-            "Content-Type": "application/json",
-            "User-Agent": "Mozilla/5.0"
+            "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36",
+            "Accept": "application/json",
+            "Content-Type": "application/json"
         }
         
-        # Intentar login
-        response = session.post(login_url, json=login_data, headers=headers, timeout=10)
+        # Intentar endpoints comunes de Ubiquiti
+        endpoints_try = [
+            f"http://{ip}/api/login",
+            f"http://{ip}/api/v1/login", 
+            f"http://{ip}/api/v2/login",
+            f"http://{ip}/login",
+            f"https://{ip}/api/login"
+        ]
         
-        if response.status_code == 200:
-            # Login exitoso, ahora obtener métricas
-            status_url = f"http://{ip}/api/v1/status"
-            status_response = session.get(status_url, timeout=10)
-            
-            if status_response.status_code == 200:
-                metricas = status_response.json()
-                resultado = f"# ✅ CONEXIÓN EXITOSA - {ip}\n\n"
-                resultado += f"**Estado:** Conectado via API Web\n"
-                resultado += f"**Hora:** {time.strftime('%H:%M:%S')}\n\n"
-                resultado += "## MÉTRICAS OBTENIDAS:\n"
-                resultado += f"```json\n{json.dumps(metricas, indent=2)[:1000]}\n```"
-                return resultado
-            else:
-                return f"# ⚠️ CONECTADO PERO NO MÉTRICAS\n\nError al obtener status: {status_response.status_code}"
-        else:
-            return f"# ❌ ERROR LOGIN\n\nNo se pudo autenticar: {response.status_code}"
-            
-    except Exception as e:
-        return f"# ❌ ERROR DE CONEXIÓN\n\n**IP:** {ip}\n**Error:** {str(e)}\n**Hora:** {time.strftime('%H:%M:%S')}"
-
-# INTERFAZ GRADIO
-with gr.Blocks(title="Test Ubiquiti Web API") as demo:
-    gr.Markdown("# 📡 Test Antena Ubiquiti (Web API)")
-    
-    with gr.Row():
-        with gr.Column():
-            ip_input = gr.Textbox(label="IP del Equipo", placeholder="192.168.1.24", value="192.168.1.24")
-            usuario_input = gr.Textbox(label="Usuario", value="admin")
-            password_input = gr.Textbox(label="Contraseña", type="password")
-            test_btn = gr.Button("🚀 Ejecutar Test Web", variant="primary")
+        login_data = {"username": usuario, "password": password}
         
-        with gr.Column():
-            resultado_output = gr.Markdown()
-    
-    test_btn.click(
-        fn=test_antena_remota,
-        inputs=[ip_input, usuario_input, password_input],
-        outputs=resultado_output
-    )
+        for login_url in endpoints_try:
+            try:
+                print(f"Probando: {login_url}")
+                response = session.post(
+                    login_url, 
+                    json=login_data, 
+                    headers=headers, 
+                    timeout=10,
+                    verify=False  # Ignorar SSL si es necesario
+                )
+                
+                if response.status_code in [200, 201]:
+                    print(f"✅ Login exitoso en: {login_url}")
+                    
+                    # Probar endpoints de status
+                    status_endpoints = [
+                        f"http://{ip}/api/v1/status",
+                        f"http://{ip}/api/status",
+                        f"http://{ip}/status",
+                        f"http://{ip}/api/v1/device",
+                        f"http://{ip}/api/v2/devices"
+                    ]
+                    
+                    for status_url in status_endpoints:
+                        try:
+                            status_response = session.get(status_url, timeout=10, verify=False)
+                            if status_response.status_code == 200:
+                                metricas = status_response.json()
+                                return f"""
+# ✅ CONEXIÓN EXITOSA - {ip}
+
+**Método:** API Web Ubiquiti
+**Endpoint:** {status_url}
+**Hora:** {time.strftime('%H:%M:%S')}
 
-demo.launch()
+## DATOS OBTENIDOS:
+```json
+{json.dumps(metricas, indent=2)[:1500]}