Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py – ARF v4 API with Gradio frontend (optional)
+# app.py – ARF v4 API with Gradio frontend (FastAPI mounted under /api)
 import logging
 import uuid
 from datetime import datetime, timezone
@@ -18,10 +18,10 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # ========================= FASTAPI APP =========================
-app = FastAPI(title="ARF v4 API")
+fastapi_app = FastAPI(title="ARF v4 API")
 
 # Enable CORS for your frontend
-app.add_middleware(
+fastapi_app.add_middleware(
     CORSMiddleware,
     allow_origins=["https://arf-frontend-sandy.vercel.app"],
     allow_methods=["*"],
@@ -58,17 +58,16 @@ class _DemoIntent:
     service_name = "demo"
 
 # ========================= API ENDPOINTS =========================
-
-@app.get("/")
+@fastapi_app.get("/")
 async def root():
     """Root endpoint – returns a welcome message."""
     return {"message": "ARF v4 API. See /docs for documentation."}
 
-@app.get("/health")
+@fastapi_app.get("/health")
 async def health():
     return {"status": "ok", "version": "4.0.0"}
 
-@app.get("/api/v1/get_risk")
+@fastapi_app.get("/v1/get_risk")
 async def get_risk():
     """Return the current demo risk."""
     intent = _DemoIntent()
@@ -88,7 +87,7 @@ async def get_risk():
         "decision_id": decision_id,
         "timestamp": datetime.now(timezone.utc).isoformat(),
         "risk_score": float(risk_value),
-        "outcome": None,
+        "outcome": None,  # will be filled when feedback is given
     })
 
     return {
@@ -101,17 +100,18 @@ async def get_risk():
         "timestamp": datetime.now(timezone.utc).isoformat()
     }
 
-@app.get("/api/v1/history")
+@fastapi_app.get("/v1/history")
 async def get_history():
     """Return the last 10 decisions."""
     return decision_history[-10:]
 
-@app.post("/api/v1/incidents/evaluate", response_model=EvaluateResponse)
+@fastapi_app.post("/v1/incidents/evaluate", response_model=EvaluateResponse)
 async def evaluate_incident(request: EvaluateRequest):
     """
     Evaluate an incident and return a risk score with explainability.
     This is a placeholder – replace with actual call to your risk engine.
     """
+    # For now, return a dummy response
     return EvaluateResponse(
         risk_score=0.23,
         base_risk=0.15,
@@ -121,12 +121,13 @@ async def evaluate_incident(request: EvaluateRequest):
         confidence=0.9
     )
 
-@app.post("/api/v1/feedback")
+@fastapi_app.post("/v1/feedback")
 async def record_outcome(decision_id: str, success: bool):
     """Record the outcome of a decision (success/failure)."""
     for dec in decision_history:
         if dec["decision_id"] == decision_id:
             dec["outcome"] = "success" if success else "failure"
+            # Update the risk engine (optional)
             intent = _DemoIntent()
             try:
                 risk_engine.update_outcome(intent, success)
@@ -135,8 +136,9 @@ async def record_outcome(decision_id: str, success: bool):
             return {"status": "ok", "decision_id": decision_id, "outcome": dec["outcome"]}
     return {"error": "decision not found"}
 
-# ========================= GRADIO UI (Optional) =========================
-# Mount the Gradio interface at the root path.
+# ========================= GRADIO UI (Main App) =========================
+# Gradio interface will be served at the root path ("/").
+# The FastAPI app is mounted under the "/api" subpath.
 
 def get_risk_snapshot():
     try:
@@ -215,5 +217,8 @@ with gr.Blocks(title="ARF v4 Demo") as demo:
     memory_btn.click(fn=get_memory_snapshot, outputs=memory_output)
     history_btn.click(fn=lambda: decision_history[-10:], outputs=decision_output)
 
-# Mount Gradio app at the root path – the Hugging Face platform will serve this FastAPI app.
-app = gr.mount_gradio_app(app, demo, path="/")
+# ========================= Mount FastAPI under /api =========================
+app = gr.mount_gradio_app(demo, fastapi_app, path="/api")
+
+# Note: No manual server start. Hugging Face will serve this `app` (which is a Gradio app)
+# and the FastAPI endpoints will be available under /api.