Update hf_demo.py

hf_demo.py

@@ -1,4 +1,4 @@
-# hf_demo.py – ARF v4 API with Memory
+# hf_demo.py – ARF v4 API with Memory & Health Check
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 import gradio as gr
@@ -31,7 +31,6 @@ memory = RAGGraphMemory(faiss_index)
 # ---------------------------------------------------------------------------
 # API Endpoints
 # ---------------------------------------------------------------------------
-
 @app.get("/")
 async def root():
     return {
@@ -62,48 +61,4 @@ async def store_incident(event_data: dict, analysis: dict):
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/api/v1/memory/similar")
-async def find_similar_incidents(action: str, k: int = 5):
-    class DummyEvent:
-        def __init__(self, action):
-            self.component = "user_action"
-            self.latency_p99 = 0.0
-            self.error_rate = 0.0
-            self.throughput = 0
-            self.cpu_util = 0.0
-            self.memory_util = 0.0
-            self.timestamp = datetime.now()
-            self.severity = "low"
-    event = DummyEvent(action)
-    analysis = {"action": action}
-    similar = memory.find_similar(event, analysis, k=k)
-    results = []
-    for node in similar:
-        results.append({
-            "incident_id": node.incident_id,
-            "component": node.component,
-            "severity": node.severity,
-            "timestamp": node.timestamp,
-            "metrics": node.metrics,
-            "agent_analysis": node.agent_analysis,
-            "similarity_score": node.metadata.get("similarity_score", 0.0)
-        })
-    return {"similar": results, "count": len(results)}
-
-@app.get("/api/v1/memory/stats")
-async def memory_stats():
-    return memory.get_graph_stats()
-
-# Optional Gradio interface
-iface = gr.Interface(
-    fn=lambda: f"ARF v4 - Current risk: {risk_engine.get_current_risk().mean:.2f}",
-    inputs=[],
-    outputs="text",
-    title="ARF v4 Demo"
-)
-app = gr.mount_gradio_app(app, iface, path="/")
-
-# ============== MAIN ENTRY POINT ==============
-if __name__ == "__main__":
-    # Launch the Gradio interface – this will start the server
-    # Hugging Face automatically provides the correct port
-    iface.launch(server_name="0.0.0.0")
+async def find_similar_incidents(action: str, k