Update app.py

app.py

@@ -73,6 +73,8 @@ GROUPED_REGIMEN_CHOICES = [
     ("── MISMATCHED UNRELATED / CORD BLOOD ──",    "__header_mismatched_cord__"),
     ("Bolanos-Meade et al 2022 (Mismatched/Cord)", "Bolanos-Meade et al 2022 (Mismatched/Cord)"),
     ("Patel et al 2020 (Mismatched/Cord)",         "Patel et al 2020 (Mismatched/Cord)"),
+    ("── CUSTOM ──",                               "__header_custom__"),
+    ("Custom",                                     "Custom"),
 ]
 HEADER_VALUES = {v for _, v in GROUPED_REGIMEN_CHOICES if v.startswith("__header_")}
 
@@ -89,6 +91,19 @@ PUBLISHED_PRESETS = {
     "Patel et al 2020 (Mismatched/Cord)":         {"CONDGRPF": "NMA", "CONDGRP_FINAL": "TBI/Cy/Flu/TT",            "ATGF": "ATG",         "GVHD_FINAL": "Post-CY + siro +- MMF", "HLA_FINAL": "7/8", "DONORF": "Mismatched unrelated donor or cord blood"},
 }
 
+# Donor types that strictly lock HLA to 8/8
+MSD_DONOR = "HLA identical sibling"
+MUD_DONOR = "Matched unrelated donor"
+LOCKED_88_DONORS = {MSD_DONOR, MUD_DONOR}
+
+# Donor types that cannot use 8/8
+NON_88_DONORS = {"HLA mismatch relative", "Mismatched unrelated donor or cord blood"}
+
+# Mapping of which published regimens belong to which donor category
+MSD_REGIMENS  = {"Hsieh et al 2014", "Krishnamurti et al 2019", "King et al 2015", "Walters et al 1996"}
+MMUD_REGIMENS = {"Bolanos-Meade et al 2022 (Mismatched/Cord)", "Patel et al 2020 (Mismatched/Cord)",
+                 "Bolanos-Meade et al 2022 (HLA Mismatch)", "Patel et al 2020 (HLA Mismatch)"}
+
 PATIENT_FEATURES = ["AGE", "AGEGPFF", "SEX", "KPS", "RCMVPR"]
 DONOR_FEATURES   = ["DONORF", "GRAFTYPE", "HLA_FINAL", "CONDGRPF", "CONDGRP_FINAL", "ATGF", "GVHD_FINAL"]
 DISEASE_FEATURES = ["NACS2YR", "EXCHTFPR", "VOC2YPR", "VOCFRQPR", "SCATXRSN"]
@@ -108,6 +123,95 @@ NO_EVENT_COLOR = "#43a047"
 SHAP_ORDER     = ["DEAD", "GF", "AGVHD", "CGVHD", "VOCPSHI", "EFS", "STROKEHI", "OS"]
 
 
+# ─────────────────────────────────────────────────────────────────────────────
+# CONSTRAINT VALIDATION HELPERS
+# ─────────────────────────────────────────────────────────────────────────────
+
+def _get_hla_constraint_for_donor(donor):
+    """
+    Returns (allowed_hla_choices, locked_to_88, locked_from_88)
+    - locked_to_88: donor must have 8/8 (MSD or MUD)
+    - locked_from_88: donor cannot have 8/8 (mismatch/cord)
+    """
+    if donor in LOCKED_88_DONORS:
+        return ["8/8"], True, False
+    elif donor in NON_88_DONORS:
+        return ["7/8", "≤ 6/8"], False, True
+    return HLA_FINAL_CHOICES, False, False
+
+
+def _validate_counterfactual_constraints(base_dict, wi_dict):
+    """
+    Returns a list of constraint violation messages (strings).
+    Empty list means no violations.
+    """
+    violations = []
+
+    # 1. Sex is immutable — cannot be changed
+    if base_dict.get("SEX") and wi_dict.get("SEX"):
+        if base_dict["SEX"] != wi_dict["SEX"]:
+            violations.append(
+                "❌ Immutable feature: Sex cannot be changed in a counterfactual scenario."
+            )
+
+    # 2. Age cannot be decreased
+    base_age = base_dict.get("AGE")
+    wi_age   = wi_dict.get("AGE")
+    if base_age is not None and wi_age is not None:
+        try:
+            if float(wi_age) < float(base_age):
+                violations.append(
+                    f"❌ Unacceptable counterfactual: Age cannot be decreased "
+                    f"(baseline {base_age} → what-if {wi_age})."
+                )
+        except (TypeError, ValueError):
+            pass
+
+    # 3. Donor type ↔ HLA matching constraint
+    wi_donor = wi_dict.get("DONORF")
+    wi_hla   = wi_dict.get("HLA_FINAL")
+    if wi_donor and wi_hla:
+        if wi_donor in LOCKED_88_DONORS and wi_hla != "8/8":
+            violations.append(
+                f"❌ HLA constraint: Donor type '{wi_donor}' requires strictly 8/8 HLA matching."
+            )
+        elif wi_donor in NON_88_DONORS and wi_hla == "8/8":
+            violations.append(
+                f"❌ HLA constraint: Donor type '{wi_donor}' cannot have 8/8 HLA matching."
+            )
+
+    # 4. Cross-donor regimen mismatch (MMUD regimen for MSD, or MSD regimen for MMUD)
+    base_donor = base_dict.get("DONORF")
+    wi_regimen_label = wi_dict.get("__regimen_label__", "")  # injected below if available
+
+    # We detect this via DONORF in the what-if vs the preset's expected donor
+    if wi_donor:
+        # Check if what-if donor is MSD but the preset fields look like an MMUD preset
+        wi_condgrp = wi_dict.get("CONDGRP_FINAL", "")
+        wi_gvhd    = wi_dict.get("GVHD_FINAL", "")
+        # PostCY GVHD prophylaxis is characteristic of haploidentical / MMUD regimens
+        is_mmud_style_gvhd = wi_gvhd in {"Post-CY + siro +- MMF", "Post-CY + MMF + CNI"}
+        if wi_donor == MSD_DONOR and is_mmud_style_gvhd:
+            violations.append(
+                "❌ Unacceptable counterfactual: Post-cyclophosphamide GVHD prophylaxis "
+                "(characteristic of MMUD/haploidentical regimens) is inconsistent with "
+                f"donor type '{MSD_DONOR}'."
+            )
+
+    return violations
+
+
+def _hla_update_for_donor(donor_value):
+    """Return gr.update for HLA dropdown based on donor selection."""
+    if not donor_value:
+        return gr.update(choices=HLA_FINAL_CHOICES, interactive=True)
+    if donor_value in LOCKED_88_DONORS:
+        return gr.update(choices=["8/8"], value="8/8", interactive=False)
+    elif donor_value in NON_88_DONORS:
+        return gr.update(choices=["7/8", "≤ 6/8"], value=None, interactive=True)
+    return gr.update(choices=HLA_FINAL_CHOICES, interactive=True)
+
+
 # ─────────────────────────────────────────────────────────────────────────────
 # SHARED HELPERS
 # ─────────────────────────────────────────────────────────────────────────────
@@ -156,20 +260,57 @@ def vocfrqpr_from_voc2ypr(voc_status):
 
 
 def apply_grouped_preset(selected_value):
+    """
+    Applies a published preset or enables Custom mode.
+    Returns updates for:
+      [dropdown, donorf, condgrpf, condgrp_final, atgf, gvhd_final, hla_final,
+       condgrpf_interactive, condgrp_final_interactive, atgf_interactive,
+       gvhd_final_interactive, donorf_interactive]
+    When a published preset is selected, transplant fields are locked (non-interactive).
+    When Custom is selected, transplant fields become interactive (except HLA which follows donor).
+    """
     if not selected_value or selected_value in HEADER_VALUES:
-        return [gr.update(value=None)] + [gr.update()] * 6
+        return (
+            gr.update(value=None),  # dropdown reset
+            gr.update(),            # donorf
+            gr.update(),            # condgrpf
+            gr.update(),            # condgrp_final
+            gr.update(),            # atgf
+            gr.update(),            # gvhd_final
+            gr.update(),            # hla_final
+        )
+
+    if selected_value == "Custom":
+        # Unlock all transplant fields for custom entry
+        return (
+            gr.update(),
+            gr.update(interactive=True),   # donorf
+            gr.update(interactive=True),   # condgrpf
+            gr.update(interactive=True),   # condgrp_final
+            gr.update(interactive=True),   # atgf
+            gr.update(interactive=True),   # gvhd_final
+            gr.update(interactive=True),   # hla_final — will be further constrained by donor change
+        )
+
     preset = PUBLISHED_PRESETS.get(selected_value)
     if not preset:
-        return [gr.update()] * 7
-    return [
+        return (gr.update(),) * 7
+
+    donor = preset["DONORF"]
+    hla_upd = _hla_update_for_donor(donor)
+    # Override the value with the preset's HLA
+    hla_upd_dict = hla_upd if isinstance(hla_upd, dict) else {}
+    hla_final_update = gr.update(value=preset["HLA_FINAL"], interactive=False, choices=["8/8"] if donor in LOCKED_88_DONORS else (["7/8", "≤ 6/8"] if donor in NON_88_DONORS else HLA_FINAL_CHOICES))
+
+    return (
         gr.update(),
-        gr.update(value=preset["DONORF"]),
-        gr.update(value=preset["CONDGRPF"]),
-        gr.update(value=preset["CONDGRP_FINAL"]),
-        gr.update(value=preset["ATGF"]),
-        gr.update(value=preset["GVHD_FINAL"]),
-        gr.update(value=preset["HLA_FINAL"]),
-    ]
+        gr.update(value=preset["DONORF"],        interactive=False),
+        gr.update(value=preset["CONDGRPF"],      interactive=False),
+        gr.update(value=preset["CONDGRP_FINAL"], interactive=False),
+        gr.update(value=preset["ATGF"],          interactive=False),
+        gr.update(value=preset["GVHD_FINAL"],    interactive=False),
+        hla_final_update,
+    )
 
 
 def copy_fields(*vals):
@@ -297,10 +438,6 @@ def _icon_card_html(probability: float, outcome: str,
 
 
 def _build_comparison_icon_grid(base_probs: dict, wi_probs: dict) -> str:
-    """
-    Side-by-side icon array grid.
-    Layout: 3 outcome columns per row, each column has Baseline card + What-If card side by side.
-    """
     rows_html = ""
     for row_start in range(0, len(ICON_OUTCOMES), 3):
         chunk = ICON_OUTCOMES[row_start: row_start + 3]
@@ -398,6 +535,22 @@ def _build_comparison_table_html(base_probs, base_ci, wi_probs, wi_ci) -> str:
     return header + rows + footer
 
 
+def _build_violation_html(violations: list) -> str:
+    if not violations:
+        return ""
+    items = "".join(f"<li style='margin-bottom:6px;'>{v}</li>" for v in violations)
+    return (
+        f'<div style="background:#fff3e0;border:2px solid #e65100;border-radius:8px;'
+        f'padding:14px 18px;font-family:\'Segoe UI\',Arial,sans-serif;margin-bottom:12px;">'
+        f'<div style="font-weight:700;font-size:14px;color:#bf360c;margin-bottom:8px;">'
+        f'⚠️ Constraint Violations — Analysis blocked</div>'
+        f'<ul style="margin:0;padding-left:20px;color:#6d1f00;font-size:13px;">{items}</ul>'
+        f'<div style="margin-top:10px;font-size:11px;color:#888;">'
+        f'Please correct the above before running the counterfactual comparison.</div>'
+        f'</div>'
+    )
+
+
 # ─────────────────────────────────────────────────────────────────────────────
 # MAIN PREDICT CALLBACK
 # ─────────────────────────────────────────────────────────────────────────────
@@ -452,19 +605,28 @@ def predict_gradio(*values):
 def run_whatif_predict(*all_values):
     """
     Receives 2 × len(ALL_FEATURES) values: first block = baseline, second = what-if.
-    Returns: table_html, icon_grid_html,
+    Returns: violation_html, table_html, icon_grid_html,
              base_shap×8, wi_shap×8   (order = SHAP_ORDER)
     """
     n             = len(ALL_FEATURES)
     baseline_vals = all_values[:n]
     whatif_vals   = all_values[n:]
 
+    # Number of outputs: 1 violation + 1 table + 1 icon + 8 base shap + 8 wi shap = 19
+    empty_shaps = [None] * 8
+
     try:
         base_dict   = _values_to_dict(baseline_vals)
         whatif_dict = _values_to_dict(whatif_vals)
         _check_missing(base_dict,   "Baseline")
         _check_missing(whatif_dict, "What-If")
 
+        # ── Constraint validation ────────────────────────────────────────────
+        violations = _validate_counterfactual_constraints(base_dict, whatif_dict)
+        if violations:
+            violation_html = _build_violation_html(violations)
+            return (violation_html, "", "", *empty_shaps, *empty_shaps)
+
         base_probs, base_ci = predict_all_outcomes(
             base_dict,   use_calibration=True, use_signed_voting=True, n_boot_ci=500
         )
@@ -479,6 +641,7 @@ def run_whatif_predict(*all_values):
         wi_shap    = create_all_shap_plots(whatif_dict, max_display=10)
 
         return (
+            "",   # no violations
             table_html,
             icon_html,
             *[base_shap[o] for o in SHAP_ORDER],
@@ -489,6 +652,17 @@ def run_whatif_predict(*all_values):
         raise gr.Error(f"{type(e).__name__}: {str(e)}")
 
 
+# ─────────────────────────────────────────────────────────────────────────────
+# WHAT-IF SEX LOCK CALLBACK
+# ─────────────────────────────────────────────────────────────────────────────
+
+def lock_whatif_sex(baseline_sex):
+    """Mirror baseline sex into what-if and lock it."""
+    if baseline_sex:
+        return gr.update(value=baseline_sex, interactive=False)
+    return gr.update(interactive=False)
+
+
 # ─────────────────────────────────────────────────────────────────────────────
 # CSS
 # ─────────────────────────────────────────────────────────────────────────────
@@ -516,6 +690,12 @@ custom_css = """
     border: none !important; color: white !important; font-weight: 600 !important;
 }
 .copy-from-predict-button:hover { background: linear-gradient(to right, #4a148c, #8e24aa) !important; }
+.counterfactual-button {
+    background: linear-gradient(to right, #1976d2, #42a5f5) !important;
+    border: none !important; color: white !important;
+    font-weight: bold !important; font-size: 15px !important; padding: 12px !important;
+}
+.counterfactual-button:hover { background: linear-gradient(to right, #1565c0, #1e88e5) !important; }
 /* Ensure Outcome column text is never truncated in the results table */
 .output-dataframe table td:first-child,
 .output-dataframe table th:first-child {
@@ -523,6 +703,16 @@ custom_css = """
     word-break: break-word !important;
     min-width: 240px !important;
 }
+/* Constraint info box */
+.constraint-info {
+    background: #e8f5e9;
+    border-left: 4px solid #388e3c;
+    padding: 8px 14px;
+    font-size: 12px;
+    color: #1b5e20;
+    border-radius: 4px;
+    margin-bottom: 8px;
+}
 """
 
 
@@ -530,7 +720,6 @@ custom_css = """
 # BUILD UI
 # ─────────────────────────────────────────────────────────────────────────────
 
-# FIX 1: css moved from gr.Blocks() to demo.launch() for Gradio 6.0
 with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
     gr.Markdown("# HCT Outcome Prediction Model")
 
@@ -572,6 +761,14 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
 
             inputs_dict["AGE"].change(get_age_group, inputs_dict["AGE"], inputs_dict["AGEGPFF"])
             inputs_dict["VOC2YPR"].change(vocfrqpr_from_voc2ypr, inputs_dict["VOC2YPR"], inputs_dict["VOCFRQPR"])
+
+            # HLA locking on Tab 1 based on donor selection
+            p_donorf.change(
+                fn=_hla_update_for_donor,
+                inputs=p_donorf,
+                outputs=p_hla_final,
+            )
+
             grouped_dd.change(
                 apply_grouped_preset, grouped_dd,
                 [grouped_dd, p_donorf, p_condgrpf, p_condgrp_final, p_atgf, p_gvhd_final, p_hla_final],
@@ -584,7 +781,6 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
             gr.Markdown("---")
             gr.Markdown("## Prediction Results")
 
-            # FIX 2: col_count renamed to column_count in Gradio 6.0
             output_table = gr.Dataframe(
                 headers=["Outcome", "Probability", "95% CI"],
                 label="Predicted Outcomes",
@@ -622,7 +818,7 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
             )
 
         # ══════════════════════════════════════════════════════════════════════
-        # TAB 2 — WHAT-IF ANALYSIS
+        # TAB 2 — WHAT-IF / COUNTERFACTUAL ANALYSIS
         # ══════════════════════════════════════════════════════════════════════
         with gr.Tab("Counterfactual Scenarios"):
             gr.Markdown(
@@ -634,6 +830,19 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
                 "- *Copy Baseline → What-If* — mirrors the baseline into the what-if panel so you only change what differs"
             )
 
+            # Constraint legend
+            gr.HTML(
+                '<div class="constraint-info">'
+                '<strong>Active constraints:</strong> '
+                '(1) <em>Sex</em> is immutable and locked in the counterfactual. &nbsp;|&nbsp; '
+                '(2) Transplant characteristics must be changed as <em>grouped clinical scenarios</em> via the published regimen dropdown (or Custom). &nbsp;|&nbsp; '
+                '(3) <em>MSD / MUD donors</em> are locked to 8/8 HLA; mismatch/cord donors cannot use 8/8. &nbsp;|&nbsp; '
+                '(4) VoC = No → VOC frequency defaults to &lt;3/yr. &nbsp;|&nbsp; '
+                '(5) Age cannot be <em>decreased</em>. &nbsp;|&nbsp; '
+                '(6) MMUD-style regimens cannot be assigned to MSD donors and vice-versa.'
+                '</div>'
+            )
+
             wi_baseline_dict = {}
             wi_whatif_dict   = {}
 
@@ -673,6 +882,14 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
 
             wi_baseline_dict["AGE"].change(get_age_group, wi_baseline_dict["AGE"], wi_baseline_dict["AGEGPFF"])
             wi_baseline_dict["VOC2YPR"].change(vocfrqpr_from_voc2ypr, wi_baseline_dict["VOC2YPR"], wi_baseline_dict["VOCFRQPR"])
+
+            # Baseline donor → HLA locking
+            wb_donorf.change(
+                fn=_hla_update_for_donor,
+                inputs=wb_donorf,
+                outputs=wb_hla_final,
+            )
+
             wi_grouped_base.change(
                 apply_grouped_preset, wi_grouped_base,
                 [wi_grouped_base, wb_donorf, wb_condgrpf, wb_condgrp_final, wb_atgf, wb_gvhd_final, wb_hla_final],
@@ -683,7 +900,7 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
             # Wire "Copy from Predict tab" → baseline fields
             copy_from_predict_btn.click(
                 fn=copy_fields,
-                inputs=inputs_list,        # Tab 1 components
+                inputs=inputs_list,
                 outputs=wi_baseline_list,
             )
 
@@ -707,7 +924,7 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
                     wi_grouped_wi         = gr.Dropdown(
                         choices=GROUPED_REGIMEN_CHOICES, value=None,
                         label="Published conditioning regimen (Counterfactual)",
-                        info="Auto-fills what-if transplant fields",
+                        info="Auto-fills what-if transplant fields. Use 'Custom' to enter manually.",
                     )
                     ww_donorf        = wi_whatif_dict["DONORF"]        = make_component("DONORF",        "(What-If)")
                     wi_whatif_dict["GRAFTYPE"]                         = make_component("GRAFTYPE",      "(What-If)")
@@ -718,12 +935,30 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
                     ww_hla_final     = wi_whatif_dict["HLA_FINAL"]     = make_component("HLA_FINAL",     "(What-If)")
 
                 with gr.Column(scale=1):
-                    gr.Markdown("###  Counterfactual — Disease Characteristics")
+                    gr.Markdown("### Counterfactual — Disease Characteristics")
                     for f in DISEASE_FEATURES:
                         wi_whatif_dict[f] = make_component(f, "(Counterfactual)")
 
             wi_whatif_dict["AGE"].change(get_age_group, wi_whatif_dict["AGE"], wi_whatif_dict["AGEGPFF"])
             wi_whatif_dict["VOC2YPR"].change(vocfrqpr_from_voc2ypr, wi_whatif_dict["VOC2YPR"], wi_whatif_dict["VOCFRQPR"])
+
+            # ── CONSTRAINT: Sex is immutable — mirror baseline sex and lock what-if sex ──
+            wi_baseline_dict["SEX"].change(
+                fn=lock_whatif_sex,
+                inputs=wi_baseline_dict["SEX"],
+                outputs=wi_whatif_dict["SEX"],
+            )
+            # Also lock on copy
+            wi_whatif_dict["SEX"].interactive = False
+
+            # ── CONSTRAINT: Counterfactual donor → HLA locking ──────────────
+            ww_donorf.change(
+                fn=_hla_update_for_donor,
+                inputs=ww_donorf,
+                outputs=ww_hla_final,
+            )
+
+            # Preset selection locks transplant fields; Custom unlocks them
             wi_grouped_wi.change(
                 apply_grouped_preset, wi_grouped_wi,
                 [wi_grouped_wi, ww_donorf, ww_condgrpf, ww_condgrp_final, ww_atgf, ww_gvhd_final, ww_hla_final],
@@ -733,20 +968,38 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
 
             copy_to_whatif_btn.click(fn=copy_fields, inputs=wi_baseline_list, outputs=wi_whatif_list)
 
+            # After copying baseline → whatif, re-apply sex lock and HLA lock
+            copy_to_whatif_btn.click(
+                fn=lock_whatif_sex,
+                inputs=wi_baseline_dict["SEX"],
+                outputs=wi_whatif_dict["SEX"],
+            )
+            copy_to_whatif_btn.click(
+                fn=_hla_update_for_donor,
+                inputs=ww_donorf,
+                outputs=ww_hla_final,
+            )
+
             # ── RUN ──────────────────────────────────────────────────────────
             gr.Markdown("---")
-            wi_run_btn = gr.Button("Run Counterfactual Comparison", elem_classes="Counterfactual-button", size="lg")
+            wi_run_btn = gr.Button(
+                "Run Counterfactual Comparison",
+                elem_classes="counterfactual-button",
+                size="lg",
+            )
 
             # ── RESULTS ──────────────────────────────────────────────────────
             gr.Markdown("## Comparison Results")
 
+            # Constraint violation banner (shown if validation fails)
+            wi_violation_html = gr.HTML()
+
             gr.Markdown("### Outcome Probability Table")
             wi_table_html = gr.HTML()
 
             gr.Markdown("---")
             gr.Markdown("### Outcome Icon Arrays — Baseline vs Counterfactual")
-            # FIX 3: sanitize=False removed (parameter no longer exists in Gradio 6.0)
-            gr.Markdown("<small>Each outcome shows Baseline and What-If  side by side.</small>")
+            gr.Markdown("<small>Each outcome shows Baseline and What-If side by side.</small>")
             wi_icon_html = gr.HTML()
 
             gr.Markdown("---")
@@ -768,7 +1021,7 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
                 wi_shap_wi_dead      = gr.Plot(label="Death — Counterfactual")
                 wi_shap_wi_gf        = gr.Plot(label="Graft Failure — Counterfactual")
                 wi_shap_wi_agvhd     = gr.Plot(label="Acute GvHD — Counterfactual")
-                wi_shap_wi_cgvhd     = gr.Plot(label="Chronic GvHD —Counterfactual")
+                wi_shap_wi_cgvhd     = gr.Plot(label="Chronic GvHD — Counterfactual")
             with gr.Row():
                 wi_shap_wi_vocpshi   = gr.Plot(label="VOC Post-HCT — Counterfactual")
                 wi_shap_wi_efs       = gr.Plot(label="Event-Free Survival — Counterfactual")
@@ -779,6 +1032,7 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
                 fn=run_whatif_predict,
                 inputs=wi_baseline_list + wi_whatif_list,
                 outputs=[
+                    wi_violation_html,
                     wi_table_html,
                     wi_icon_html,
                     # baseline SHAP (SHAP_ORDER: DEAD GF AGVHD CGVHD VOCPSHI EFS STROKEHI OS)
@@ -793,5 +1047,4 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
 
 # ─────────────────────────────────────────────────────────────────────────────
 if __name__ == "__main__":
-    # FIX 1 (continued): css passed to launch() instead of gr.Blocks()
     demo.launch(ssr_mode=False, css=custom_css)