Update app.py

app.py

@@ -10,7 +10,7 @@ from inference import (
     SHAP_OUTCOMES,
     predict_with_comparison,
     create_all_shap_plots,
-    create_all_pie_charts,
+    icon_array,
 )
 
 
@@ -179,7 +179,6 @@ def vocfrqpr_from_voc2ypr(voc_status):
 
 
 def apply_grouped_preset(selected_value):
-    # Header row clicked — reset dropdown, leave all fields unchanged
     if not selected_value or selected_value in HEADER_VALUES:
         return [gr.update(value=None)] + [gr.update()] * 6
 
@@ -188,7 +187,7 @@ def apply_grouped_preset(selected_value):
         return [gr.update()] * 7
 
     return [
-        gr.update(),                             # leave dropdown showing selection
+        gr.update(),
         gr.update(value=preset["DONORF"]),
         gr.update(value=preset["CONDGRPF"]),
         gr.update(value=preset["CONDGRP_FINAL"]),
@@ -280,16 +279,19 @@ def predict_gradio(*values):
         df = pd.DataFrame(rows)
 
         shap_plots = create_all_shap_plots(user_vals, max_display=10)
-        pie_charts = create_all_pie_charts(calibrated_probs)
+
+        # Icon arrays for each outcome
+        icon_outcomes = ["DEAD", "GF", "AGVHD", "CGVHD", "VOCPSHI", "STROKEHI"]
+        icon_plots = {o: icon_array(calibrated_probs[o], o) for o in icon_outcomes}
 
         return (
             df,
-            pie_charts["DEAD"],
-            pie_charts["GF"],
-            pie_charts["AGVHD"],
-            pie_charts["CGVHD"],
-            pie_charts["VOCPSHI"],
-            pie_charts["STROKEHI"],
+            icon_plots["DEAD"],
+            icon_plots["GF"],
+            icon_plots["AGVHD"],
+            icon_plots["CGVHD"],
+            icon_plots["VOCPSHI"],
+            icon_plots["STROKEHI"],
             shap_plots["DEAD"],
             shap_plots["GF"],
             shap_plots["AGVHD"],
@@ -331,7 +333,7 @@ custom_css = """
 # Gradio UI
 # ---------------------------------------------------------------------------
 
-with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
+with gr.Blocks(title="HCT Outcome Prediction Model", css=custom_css) as demo:
     gr.Markdown(
         """
         # HCT Outcome Prediction Model
@@ -388,7 +390,6 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
         outputs=inputs_dict["VOCFRQPR"],
     )
 
-    # outputs[0] is the dropdown itself so clicking a header resets it to None
     grouped_regimen_dropdown.change(
         fn=apply_grouped_preset,
         inputs=grouped_regimen_dropdown,
@@ -415,23 +416,23 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
         )
 
     gr.Markdown("---")
-    gr.Markdown("## Pie Charts")
+    gr.Markdown("## Icon Arrays")
 
     with gr.Row():
         with gr.Column():
-            pie_dead    = gr.Plot(label="Death")
+            icon_dead    = gr.Plot(label="Death")
         with gr.Column():
-            pie_gf      = gr.Plot(label="Graft Failure")
+            icon_gf      = gr.Plot(label="Graft Failure")
         with gr.Column():
-            pie_agvhd   = gr.Plot(label="Acute Graft-versus-Host Disease")
+            icon_agvhd   = gr.Plot(label="Acute Graft-versus-Host Disease")
 
     with gr.Row():
         with gr.Column():
-            pie_cgvhd   = gr.Plot(label="Chronic Graft-versus-Host Disease")
+            icon_cgvhd   = gr.Plot(label="Chronic Graft-versus-Host Disease")
         with gr.Column():
-            pie_vocpshi = gr.Plot(label="Vaso-Occlusive Crisis Post-HCT")
+            icon_vocpshi = gr.Plot(label="Vaso-Occlusive Crisis Post-HCT")
         with gr.Column():
-            pie_stroke  = gr.Plot(label="Stroke Post-HCT")
+            icon_stroke  = gr.Plot(label="Stroke Post-HCT")
 
     gr.Markdown("---")
     gr.Markdown("## SHAP - Feature Importance")
@@ -461,7 +462,7 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
         inputs=inputs_list,
         outputs=[
             output_table,
-            pie_dead, pie_gf, pie_agvhd, pie_cgvhd, pie_vocpshi, pie_stroke,
+            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,
         ],
@@ -469,4 +470,4 @@ with gr.Blocks(title="HCT Outcome Prediction Model") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(ssr_mode=False, css=custom_css)
+    demo.launch(ssr_mode=False)