Update main.py

main.py

@@ -1,7 +1,9 @@
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import RedirectResponse
 from pydantic import BaseModel
 import uvicorn
 import logging
+import os
 
 # Import our custom modules
 from anomaly_detector import AnomalyDetector
@@ -10,16 +12,20 @@ from rag_agent import SentinelAgent
 # 1. Initialize the App
 app = FastAPI(
     title="Sentinel MLOps Agent",
-    description="Autonomous Anomaly Detection & RAG Investigation API",
+    description="Autonomous Anomaly Detection & RAG Investigation API. Powered by Gemini 2.5.",
     version="1.0"
 )
 
 # 2. Load the Engines
 print("🔋 Starting Sentinel Engines...")
-detector = AnomalyDetector()
-agent = SentinelAgent()
+try:
+    detector = AnomalyDetector()
+    agent = SentinelAgent()
+    print("✅ Engines Active.")
+except Exception as e:
+    print(f"❌ Error loading engines: {e}")
 
-# 3. Define the Data Format (Validation)
+# 3. Define the Data Format
 class MetricData(BaseModel):
     timestamp: str
     service_name: str
@@ -30,8 +36,8 @@ class MetricData(BaseModel):
 async def monitor_system(data: MetricData):
     """
     Receives live system data.
-    - If Normal: Returns OK.
-    - If Anomaly: Triggers Gemini 2.5 to investigate.
+    - If Normal (< 2.5 sigma): Returns OK.
+    - If Anomaly (> 2.5 sigma): Triggers Gemini 2.5 to investigate logs.
     """
     # Step A: Check for Physics/Math Anomaly
     is_anomaly, msg, z_score = detector.update(data.cpu_usage)
@@ -44,7 +50,7 @@ async def monitor_system(data: MetricData):
         }
     
     # Step B: If Anomaly, Wake up the Agent
-    print(f"🚨 ALERT: Anomaly on {data.service_name} detected!")
+    print(f"🚨 ALERT: Anomaly on {data.service_name} detected! (Z={z_score:.2f})")
     
     report = agent.investigate(
         anomaly_value=data.cpu_usage, 
@@ -59,10 +65,14 @@ async def monitor_system(data: MetricData):
         "investigation_report": report
     }
 
-@app.get("/")
-def home():
-    return {"message": "Sentinel AI Agent is Active. Send POST to /monitor"}
+# 5. Root Redirect (The Magic Trick)
+@app.get("/", include_in_schema=False)
+def root():
+    # Automatically redirects users to the Swagger UI
+    return RedirectResponse(url="/docs")
 
-# 5. Run the Server (Mobile Friendly)
+# 6. Run the Server
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    # Use the port required by Hugging Face (7860) or default to 8000
+    port = int(os.getenv("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)