Update hf_demo.py

hf_demo.py

@@ -1,7 +1,6 @@
 # hf_demo.py – ARF v4 API with Gradio frontend (optional)
 import logging
 import uuid
-import os
 from datetime import datetime, timezone
 from typing import Dict, Any, Optional
 
@@ -89,7 +88,7 @@ async def get_risk():
         "decision_id": decision_id,
         "timestamp": datetime.now(timezone.utc).isoformat(),
         "risk_score": float(risk_value),
-        "outcome": None,  # will be filled when feedback is given
+        "outcome": None,
     })
 
     return {
@@ -113,7 +112,6 @@ 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,
@@ -129,7 +127,6 @@ async def record_outcome(decision_id: str, success: bool):
     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)
@@ -139,7 +136,7 @@ async def record_outcome(decision_id: str, success: bool):
     return {"error": "decision not found"}
 
 # ========================= GRADIO UI (Optional) =========================
-# This mounts the Gradio interface at the root path ("/").
+# Mount the Gradio interface at the root path.
 
 def get_risk_snapshot():
     try:
@@ -218,12 +215,5 @@ 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
-app = gr.mount_gradio_app(app, demo, path="/")
-
-# ========================= ENTRY POINT =========================
-# This ensures the server starts and stays alive when the script is run directly.
-if __name__ == "__main__":
-    import uvicorn
-    port = int(os.environ.get("PORT", 7860))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+# Mount Gradio app at the root path – the Hugging Face platform will serve this FastAPI app.
+app = gr.mount_gradio_app(app, demo, path="/")