Update hf_demo.py

hf_demo.py

@@ -1,4 +1,6 @@
 # hf_demo.py – ARF v4 API with Memory
+import os
+import uvicorn
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 import gradio as gr
@@ -47,7 +49,6 @@ async def health():
 
 @app.get("/api/v1/get_risk")
 async def get_risk():
-    # Get current system risk (this method exists in v4)
     risk_score = risk_engine.get_current_risk()
     return {
         "system_risk": risk_score.mean,
@@ -56,10 +57,6 @@ async def get_risk():
 
 @app.post("/api/v1/incident")
 async def store_incident(event_data: dict, analysis: dict):
-    """
-    Store an incident in memory.
-    event_data should contain at least 'component', 'latency_p99', 'error_rate', etc.
-    """
     try:
         incident_id = memory.store_incident(event_data, analysis)
         return {"incident_id": incident_id}
@@ -68,10 +65,6 @@ async def store_incident(event_data: dict, analysis: dict):
 
 @app.get("/api/v1/memory/similar")
 async def find_similar_incidents(action: str, k: int = 5):
-    """
-    Find incidents similar to the given action text.
-    This uses a simple embedding fallback (random) for OSS.
-    """
     class DummyEvent:
         def __init__(self, action):
             self.component = "user_action"
@@ -109,4 +102,9 @@ iface = gr.Interface(
     outputs="text",
     title="ARF v4 Demo"
 )
-app = gr.mount_gradio_app(app, iface, path="/")
+app = gr.mount_gradio_app(app, iface, path="/")
+
+# ============== MAIN ENTRY POINT ==============
+if __name__ == "__main__":
+    port = int(os.environ.get('PORT', 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)