Update app.py

app.py

@@ -85,7 +85,7 @@ async def root():
 
 @fastapi_app.get("/health")
 async def health():
-    return {"status": "ok", "version": "4.0.0"}
+    return {"status": "ok", "version": "4.2.0"}
 
 @fastapi_app.get("/v1/get_risk")
 async def get_risk():
@@ -173,7 +173,24 @@ async def evaluate_incident(request: EvaluateRequest):
         else:
             action = RecommendedAction.ESCALATE
 
-        # 6. Build HealingIntent manually
+        # 6. Build risk_factors from component contributions
+        risk_factors = {}
+        weights = contributions.get("weights", {})
+        if weights.get("conjugate", 0.0) > 0:
+            conj_risk = contributions.get("conjugate_mean", risk_score)
+            risk_factors["conjugate"] = weights["conjugate"] * conj_risk
+        if weights.get("hyper", 0.0) > 0:
+            hyper_risk = contributions.get("hyper_mean", risk_score)
+            risk_factors["hyperprior"] = weights["hyper"] * hyper_risk
+        if weights.get("hmc", 0.0) > 0:
+            hmc_risk = contributions.get("hmc_prediction", risk_score)
+            risk_factors["hmc"] = weights["hmc"] * hmc_risk
+
+        # Fallback if no factors added
+        if not risk_factors:
+            risk_factors["conjugate"] = risk_score
+
+        # 7. Build HealingIntent manually
         healing_intent = HealingIntent(
             action=action.value,
             component=intent.service_name,
@@ -183,7 +200,7 @@ async def evaluate_incident(request: EvaluateRequest):
             incident_id="",  # Not used in this context
             detected_at=datetime.now(timezone.utc).timestamp(),
             risk_score=risk_score,
-            risk_factors=contributions.get("weights", {}),  # You may need to extract properly
+            risk_factors=risk_factors,
             cost_projection=cost_projection,
             recommended_action=action,
             similar_incidents=similar_incidents,
@@ -208,7 +225,7 @@ async def evaluate_incident(request: EvaluateRequest):
                 min(1.0, risk_score + 0.05),
             ]
         if "risk_contributions" not in response_dict:
-            # Convert contributions to list format
+            # Convert contributions to list format (keeping only factors)
             response_dict["risk_contributions"] = [
                 {"factor": k, "contribution": v}
                 for k, v in contributions.items() if k not in ["weights", "conjugate_mean", "hmc_prediction"]
@@ -264,11 +281,27 @@ def get_risk_snapshot():
             "risk_score": float(risk_value),
             "outcome": None,
         })
+
+        # Build risk_factors for UI
+        risk_factors = {}
+        weights = contributions.get("weights", {})
+        if weights.get("conjugate", 0.0) > 0:
+            conj_risk = contributions.get("conjugate_mean", risk_value)
+            risk_factors["conjugate"] = weights["conjugate"] * conj_risk
+        if weights.get("hyper", 0.0) > 0:
+            hyper_risk = contributions.get("hyper_mean", risk_value)
+            risk_factors["hyperprior"] = weights["hyper"] * hyper_risk
+        if weights.get("hmc", 0.0) > 0:
+            hmc_risk = contributions.get("hmc_prediction", risk_value)
+            risk_factors["hmc"] = weights["hmc"] * hmc_risk
+        if not risk_factors:
+            risk_factors["conjugate"] = risk_value
+
         return {
             "risk": float(risk_value),
             "status": "critical" if risk_value > 0.8 else "normal",
             "explanation": explanation,
-            "contributions": contributions,
+            "risk_factors": risk_factors,
             "decision_id": decision_id,
             "decision": decision,
             "timestamp": datetime.now(timezone.utc).isoformat()
@@ -278,7 +311,7 @@ def get_risk_snapshot():
         return {"error": str(e)}
 
 def get_health_snapshot():
-    return {"status": "ok", "version": "4.0.0", "service": "ARF OSS API", "timestamp": datetime.now(timezone.utc).isoformat()}
+    return {"status": "ok", "version": "4.2.0", "service": "ARF OSS API", "timestamp": datetime.now(timezone.utc).isoformat()}
 
 def get_memory_snapshot():
     if memory.has_historical_data():
@@ -297,8 +330,10 @@ def record_outcome_ui(success: bool):
         logger.exception("Outcome update failed")
     return {"decision_id": last["decision_id"], "outcome": last["outcome"], "timestamp": datetime.now(timezone.utc).isoformat()}
 
-with gr.Blocks(title="ARF v4 Demo") as demo:
-    gr.Markdown("# Agentic Reliability Framework v4\n### Probabilistic Infrastructure Governance")
+with gr.Blocks(title="ARF v4.2.0 Demo", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# Agentic Reliability Framework v4.2.0")
+    gr.Markdown("### Probabilistic Infrastructure Governance – [📚 API Docs](/api/docs) | [📦 GitHub](https://github.com/arf-foundation/agentic-reliability-framework) | [📅 Book a Call](https://calendly.com/petter2025us/30min)")
+    gr.Markdown("---")
     with gr.Row():
         health_output = gr.JSON(label="Health")
         risk_output = gr.JSON(label="Current Risk")