Update app.py

app.py

@@ -332,4 +332,133 @@ with gr.Blocks(title="HCT Outcome Prediction Model", css=custom_css) as demo:
         """
     )
 
-    inputs_dict =
+    inputs_dict = {}
+
+    with gr.Row():
+        # Patient
+        with gr.Column(scale=1):
+            gr.Markdown("### Patient Characteristics")
+            for f in PATIENT_FEATURES:
+                inputs_dict[f] = make_component(f)
+
+        # Transplant
+        with gr.Column(scale=1):
+            gr.Markdown("### Transplant Characteristics")
+
+            grouped_regimen_dropdown = gr.Dropdown(
+                choices=GROUPED_REGIMEN_CHOICES,
+                value=None,
+                label="Published conditioning regimen",
+                info="Auto-fills Donor Type, Conditioning Intensity, Conditioning Regimen, "
+                     "Serotherapy and GVHD Prophylaxis",
+            )
+
+            donorf_comp   = inputs_dict["DONORF"]        = make_component("DONORF")
+            inputs_dict["GRAFTYPE"]                      = make_component("GRAFTYPE")
+            condgrpf      = inputs_dict["CONDGRPF"]      = make_component("CONDGRPF")
+            condgrp_final = inputs_dict["CONDGRP_FINAL"] = make_component("CONDGRP_FINAL")
+            atgf          = inputs_dict["ATGF"]          = make_component("ATGF")
+            gvhd_final    = inputs_dict["GVHD_FINAL"]    = make_component("GVHD_FINAL")
+            hla_final     = inputs_dict["HLA_FINAL"]     = make_component("HLA_FINAL")
+
+        # Disease
+        with gr.Column(scale=1):
+            gr.Markdown("### Disease Characteristics")
+            for f in DISEASE_FEATURES:
+                inputs_dict[f] = make_component(f)
+
+    # reactive callbacks
+    inputs_dict["AGE"].change(
+        fn=get_age_group,
+        inputs=inputs_dict["AGE"],
+        outputs=inputs_dict["AGEGPFF"],
+    )
+
+    inputs_dict["VOC2YPR"].change(
+        fn=vocfrqpr_from_voc2ypr,
+        inputs=inputs_dict["VOC2YPR"],
+        outputs=inputs_dict["VOCFRQPR"],
+    )
+
+    grouped_regimen_dropdown.change(
+        fn=apply_grouped_preset,
+        inputs=grouped_regimen_dropdown,
+        outputs=[
+            grouped_regimen_dropdown,
+            donorf_comp, condgrpf, condgrp_final, atgf, gvhd_final, hla_final,
+        ],
+    )
+
+    inputs_list = [inputs_dict[f] for f in ALL_FEATURES]
+
+    btn = gr.Button("Predict", elem_classes="predict-button", size="lg")
+
+    gr.Markdown("---")
+    gr.Markdown("## Prediction Results")
+    gr.Markdown("### Predicted Outcomes")
+
+    with gr.Column():
+        output_table = gr.Dataframe(
+            headers=["Outcome", "Probability", "95% CI"],
+            label="",
+            row_count=(len(REPORTING_OUTCOMES), "dynamic"),
+            column_count=(3, "fixed"),
+        )
+
+    gr.Markdown("---")
+    gr.Markdown("## Icon Arrays")
+
+    with gr.Row():
+        with gr.Column():
+            icon_dead    = gr.Plot(label="Death")
+        with gr.Column():
+            icon_gf      = gr.Plot(label="Graft Failure")
+        with gr.Column():
+            icon_agvhd   = gr.Plot(label="Acute Graft-versus-Host Disease")
+
+    with gr.Row():
+        with gr.Column():
+            icon_cgvhd   = gr.Plot(label="Chronic Graft-versus-Host Disease")
+        with gr.Column():
+            icon_vocpshi = gr.Plot(label="Vaso-Occlusive Crisis Post-HCT")
+        with gr.Column():
+            icon_stroke  = gr.Plot(label="Stroke Post-HCT")
+
+    gr.Markdown("---")
+    gr.Markdown("## SHAP - Feature Importance")
+
+    with gr.Row():
+        with gr.Column():
+            shap_dead   = gr.Plot(label="Death")
+        with gr.Column():
+            shap_gf     = gr.Plot(label="Graft Failure")
+        with gr.Column():
+            shap_agvhd  = gr.Plot(label="Acute Graft-versus-Host Disease")
+        with gr.Column():
+            shap_cgvhd  = gr.Plot(label="Chronic Graft-versus-Host Disease")
+
+    with gr.Row():
+        with gr.Column():
+            shap_vocpshi = gr.Plot(label="Vaso-Occlusive Crisis Post-HCT")
+        with gr.Column():
+            shap_efs     = gr.Plot(label="Event-Free Survival")
+        with gr.Column():
+            shap_stroke  = gr.Plot(label="Stroke Post-HCT")
+        with gr.Column():
+            shap_os      = gr.Plot(label="Overall Survival")
+
+    btn.click(
+        fn=predict_gradio,
+        inputs=inputs_list,
+        outputs=[
+            output_table,
+            icon_dead, icon_gf, icon_agvhd, icon_cgvhd, icon_vocpshi, icon_stroke,
+            shap_dead, shap_gf, shap_agvhd, shap_cgvhd,
+            shap_vocpshi, shap_efs, shap_stroke, shap_os,
+        ],
+    )
+
+if __name__ == "__main__":
+    demo.launch(
+        ssr_mode=False,
+    )