Update hf_demo.py

hf_demo.py

@@ -61,4 +61,68 @@ 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
+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
+# ---------------------------------------------------------------------------
+def current_risk_text():
+    return f"ARF v4 - Current risk: {risk_engine.get_current_risk().mean:.2f}"
+
+def health_check():
+    return {
+        "status": "running",
+        "version": "ARF v4 Demo",
+        "timestamp": datetime.utcnow().isoformat()
+    }
+
+with gr.Blocks() as demo:
+    gr.Markdown("## ARF v4 Demo & Health Check")
+
+    with gr.Row():
+        # Risk output
+        gr.Markdown("### Current Risk")
+        gr.Textbox(value=current_risk_text(), label="Risk Output", interactive=False)
+
+    with gr.Row():
+        # Health check
+        gr.Markdown("### Health Check")
+        health_output = gr.JSON()
+        gr.Button("Check Health").click(fn=health_check, outputs=health_output)
+
+# Mount Gradio on FastAPI
+app = gr.mount_gradio_app(app, demo, path="/")
+
+# ============== MAIN ENTRY POINT ==============
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0")