Update app.py

app.py

@@ -371,16 +371,17 @@ def generate_action_timeline():
     return fig
 
 def refresh_dashboard():
-    total = len(decision_history)
-    approved = sum(1 for _, d, _ in decision_history if d.get("approved", False))
-    blocked = total - approved
-    avg_risk = np.mean([r for _, r in risk_history]) if risk_history else 0.5
-    control_stats = {
-        "total_decisions": total,
-        "approved_actions": approved,
-        "blocked_actions": blocked,
-        "average_risk": float(avg_risk)
-    }
+    with history_lock:
+        total = len(decision_history)
+        approved = sum(1 for _, d, _ in decision_history if d.get("approved", False))
+        blocked = total - approved
+        avg_risk = np.mean([r for _, r in risk_history]) if risk_history else 0.5
+        control_stats = {
+            "total_decisions": total,
+            "approved_actions": approved,
+            "blocked_actions": blocked,
+            "average_risk": float(avg_risk)
+        }
     return (
         control_stats,
         generate_risk_gauge(),
@@ -401,7 +402,7 @@ oss_caps = {
 }
 
 # ----------------------------------------------------------------------
-# Health endpoint (custom route)
+# Health endpoint (async)
 # ----------------------------------------------------------------------
 async def health_endpoint():
     return {"status": "healthy", "timestamp": datetime.utcnow().isoformat()}
@@ -540,12 +541,8 @@ with gr.Blocks(title="ARF v4 – Bayesian Risk Scoring Demo") as demo:
         outputs=[hmc_summary, hmc_trace_plot, hmc_pair_plot]
     )
 
-# Add health endpoint
-demo.add_http_route(
-    path="/health",
-    method="GET",
-    endpoint=health_endpoint
-)
+# Add health endpoint using the underlying FastAPI app
+demo.fastapi_app.add_api_route("/health", health_endpoint, methods=["GET"])
 
 if __name__ == "__main__":
     demo.queue()