Update inference.py

inference.py

@@ -373,7 +373,7 @@ def predict_with_comparison(user_inputs, n_boot_ci: int = DEFAULT_N_BOOT_CI):
 # ---------------------------------------------------------------------------
 
 def _get_shap_values_for_model_outcome(user_inputs, model_outcome, invert, X_proc):
-    """Return per-model SHAP values (shape: n_models × n_processed_features)."""
+    """Return per-model SHAP values (shape: n_models x n_processed_features)."""
     all_model_shap_vals = []
     for rf_model in classification_models[model_outcome]:
         explainer = shap.TreeExplainer(rf_model, model_output="probability", data=shap_background)
@@ -515,6 +515,9 @@ def create_all_shap_plots(user_inputs, max_display=10):
     return {o: create_shap_plot(user_inputs, o, max_display) for o in SHAP_OUTCOMES}
 
 
+# ---------------------------------------------------------------------------
+# Icon array
+# ---------------------------------------------------------------------------
 
 def icon_array(probability, outcome):
     outcome_labels = {
@@ -532,36 +535,48 @@ def icon_array(probability, outcome):
     n_no_event = n_total - n_event
     cols, rows = 10, 10
 
-    shapes = []
+    shapes   = []
     icon_idx = 0
+    sx, sy   = 0.38, 0.38   # icon half-size within each unit cell
 
-    for row in range(rows - 1, -1, -1):   # top → bottom
-        for col in range(cols):            # left → right
+    for row in range(rows):          # row 0 = top
+        for col in range(cols):      # col 0 = left
             color = "#ff6b6b" if icon_idx < n_event else "#4ecdc4"
-            x0 = col * 1.2
-            y0 = row * 1.6
+            cx = col
+            cy = rows - 1 - row      # invert so row 0 renders at the top
 
-            # --- head (circle) ---
-            cx, cy_head, hr = x0 + 0.5, y0 + 1.35, 0.22
+            # ── head (circle) ──────────────────────────────────────────
+            hr = sy * 0.22
+            hx = cx
+            hy = cy + sy * 0.65
             shapes.append(dict(
                 type="circle", xref="x", yref="y",
-                x0=cx - hr, y0=cy_head - hr,
-                x1=cx + hr, y1=cy_head + hr,
+                x0=hx - hr, y0=hy - hr,
+                x1=hx + hr, y1=hy + hr,
                 fillcolor=color, line=dict(color=color, width=0),
             ))
 
-            # --- body (pentagon-ish path) ---
+            # ── body (symmetric trapezoid: shoulders → waist → feet) ───
+            tx0 = cx - sx * 0.32
+            tx1 = cx + sx * 0.32
+            wx0 = cx - sx * 0.20
+            wx1 = cx + sx * 0.20
+            bx0 = cx - sx * 0.32
+            bx1 = cx + sx * 0.32
+
+            ty_top    = cy + sy * 0.38   # shoulder line
+            ty_waist  = cy + sy * 0.00   # waist line
+            ty_bottom = cy - sy * 0.42   # feet line
+
             shapes.append(dict(
                 type="path", xref="x", yref="y",
                 path=(
-                    f"M {x0+0.18},{y0+1.10} "
-                    f"L {x0+0.82},{y0+1.10} "
-                    f"L {x0+0.90},{y0+0.55} "
-                    f"L {x0+0.60},{y0+0.55} "
-                    f"L {x0+0.60},{y0+0.0} "
-                    f"L {x0+0.40},{y0+0.0} "
-                    f"L {x0+0.40},{y0+0.55} "
-                    f"L {x0+0.10},{y0+0.55} Z"
+                    f"M {tx0},{ty_top} "
+                    f"L {tx1},{ty_top} "
+                    f"L {wx1},{ty_waist} "
+                    f"L {bx1},{ty_bottom} "
+                    f"L {bx0},{ty_bottom} "
+                    f"L {wx0},{ty_waist} Z"
                 ),
                 fillcolor=color, line=dict(color=color, width=0),
             ))
@@ -581,24 +596,19 @@ def icon_array(probability, outcome):
         ),
         shapes=shapes,
         xaxis=dict(
-            range=[-0.3, cols * 1.2 + 0.1],
+            range=[-0.65, cols - 0.35],
             showgrid=False, zeroline=False, showticklabels=False,
+            fixedrange=True,
         ),
         yaxis=dict(
-            range=[-0.3, rows * 1.6 + 0.3],
+            range=[-0.65, rows - 0.35],
             showgrid=False, zeroline=False, showticklabels=False,
-            scaleanchor="x", scaleratio=1,
+            fixedrange=True,
         ),
-        height=460,
-        width=430,
-        margin=dict(l=10, r=10, t=90, b=10),
+        height=420,
+        width=400,
+        margin=dict(l=10, r=10, t=80, b=10),
         plot_bgcolor="white",
         paper_bgcolor="white",
     )
-    return fig
-
-
-
-
-
-
+    return fig