Update SuperAgent/main.py

SuperAgent/main.py

@@ -155,19 +155,22 @@ async def chat_message(payload: ChatRequest) -> ChatResponse:
 
 @app.get("/api/v1/schedules")
 async def list_schedules():
-    port = os.getenv("PORT", "8000")
+    # Hugging Face injects SPACE_ID. If present, force port to 7860. Otherwise default to 8000.
+    port = "7860" if os.getenv("SPACE_ID") else os.getenv("PORT", "8000")
     analysis_agent_url = os.getenv("ANALYSIS_AGENT_URL", f"http://127.0.0.1:{port}/analysis")
     if "localhost" in analysis_agent_url or "127.0.0.1" in analysis_agent_url:
+        analysis_agent_url = analysis_agent_url.replace("localhost", "127.0.0.1")
         analysis_agent_url = re.sub(r":\d+", f":{port}", analysis_agent_url)
     async with httpx.AsyncClient(timeout=10.0) as client:
         response = await client.get(f"{analysis_agent_url}/api/v1/schedules")
         return response.json()
-
 @app.post("/api/v1/schedules/run/{schedule_id}")
 async def run_schedule(schedule_id: str, offset: int = 0, span_days: int = 1):
-    port = os.getenv("PORT", "8000")
+    # Hugging Face injects SPACE_ID. If present, force port to 7860. Otherwise default to 8000.
+    port = "7860" if os.getenv("SPACE_ID") else os.getenv("PORT", "8000")
     analysis_agent_url = os.getenv("ANALYSIS_AGENT_URL", f"http://127.0.0.1:{port}/analysis")
     if "localhost" in analysis_agent_url or "127.0.0.1" in analysis_agent_url:
+        analysis_agent_url = analysis_agent_url.replace("localhost", "127.0.0.1")
         analysis_agent_url = re.sub(r":\d+", f":{port}", analysis_agent_url)
     async with httpx.AsyncClient(timeout=60.0) as client:
         response = await client.post(
@@ -175,34 +178,20 @@ async def run_schedule(schedule_id: str, offset: int = 0, span_days: int = 1):
             params={"offset": offset, "span_days": span_days}
         )
         return response.json()
-
 @app.get("/api/v1/anomalies")
 async def get_active_anomalies(user_id: str = None, schema_name: str = None):
-    # Centralized schema verification for anomalies endpoint
-    resolved_schema = "openremote"
-    if schema_name and schema_name.lower() != "openremote":
-        try:
-            from app.services.asset_registry import asset_registry
-            from sqlalchemy import text
-            with asset_registry.SessionLocal() as session:
-                res = session.execute(text(
-                    "SELECT 1 FROM information_schema.tables WHERE table_schema = :schema AND table_name = 'asset' LIMIT 1"
-                ), {"schema": schema_name})
-                if res.scalar() is not None:
-                    resolved_schema = schema_name
-        except Exception:
-            resolved_schema = "openremote"
-
-    port = os.getenv("PORT", "8000")
+    # Hugging Face injects SPACE_ID. If present, force port to 7860. Otherwise default to 8000.
+    port = "7860" if os.getenv("SPACE_ID") else os.getenv("PORT", "8000")
     anomaly_agent_url = os.getenv("ANOMALY_AGENT_URL", f"http://127.0.0.1:{port}/anomaly")
     if "localhost" in anomaly_agent_url or "127.0.0.1" in anomaly_agent_url:
+        anomaly_agent_url = anomaly_agent_url.replace("localhost", "127.0.0.1")
         anomaly_agent_url = re.sub(r":\d+", f":{port}", anomaly_agent_url)
     async with httpx.AsyncClient(timeout=300.0) as client:
         response = await client.get(
             f"{anomaly_agent_url}/api/v1/anomalies",
             params={
                 "user_id": user_id or DEFAULT_USER_ID,
-                "schema_name": resolved_schema
+                "schema_name": schema_name
             }
         )
         return response.json()