Update hf_demo.py

hf_demo.py

@@ -1,6 +1,4 @@
 # 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
@@ -62,8 +60,6 @@ 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.
     """
-    # In a real implementation you would parse a ReliabilityEvent.
-    # For simplicity, we pass a dict – you may need to adapt.
     try:
         incident_id = memory.store_incident(event_data, analysis)
         return {"incident_id": incident_id}
@@ -75,9 +71,7 @@ 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.
-    For production, integrate sentence-transformers.
     """
-    # Create a dummy event from the action text
     class DummyEvent:
         def __init__(self, action):
             self.component = "user_action"
@@ -91,7 +85,6 @@ async def find_similar_incidents(action: str, k: int = 5):
     event = DummyEvent(action)
     analysis = {"action": action}
     similar = memory.find_similar(event, analysis, k=k)
-    # Convert nodes to dict
     results = []
     for node in similar:
         results.append({
@@ -116,15 +109,4 @@ iface = gr.Interface(
     outputs="text",
     title="ARF v4 Demo"
 )
-app = gr.mount_gradio_app(app, iface, path="/")
-
-# ============== MAIN ENTRY POINT ==============
-if __name__ == "__main__":
-    port = int(os.environ.get('PORT', 7860))
-    uvicorn.run(
-        "hf_demo:app",
-        host="0.0.0.0",
-        port=port,
-        log_level="info",
-        reload=False
-    )
+app = gr.mount_gradio_app(app, iface, path="/")