Diagram: ViT label, larger in-box type, AUC/AAC anchored in their regions

app.py

@@ -92,7 +92,7 @@ def _schematic_curve():
     """A reentrant titration curve used only for the diagram's thumbnails."""
     x = pipeline.intensity_ladder(pipeline.NUM_IMAGES).numpy()
     lx = (np.log10(x) - np.log10(x[0])) / (np.log10(x[-1]) - np.log10(x[0]))
-    p = np.clip(1.55 * np.exp(-((lx - 0.52) ** 2) / (2 * 0.17 ** 2)) - 0.16, 0, 1)
+    p = np.clip(1.55 * np.exp(-((lx - 0.42) ** 2) / (2 * 0.17 ** 2)) - 0.16, 0, 1)
     calls = (np.random.default_rng(0).random(p.size) < p).astype(int)
     curve = metrics.moving_average(calls, pipeline.NUM_IMAGES // pipeline.WINDOW_DIVISOR)
     return x, curve
@@ -111,8 +111,8 @@ def plot_explainer():
     stage = fig.add_axes([0, 0, 1, 1]); stage.set_axis_off()
     stage.set_xlim(0, 1); stage.set_ylim(0, 1)
 
-    xs = [0.098, 0.368, 0.632, 0.902]
-    box_y, box_h, box_w = 0.30, 0.50, 0.155
+    xs = [0.112, 0.371, 0.629, 0.888]
+    box_y, box_h, box_w = 0.30, 0.50, 0.196
 
     def node(cx, title, caption):
         stage.add_patch(FancyBboxPatch(
@@ -120,17 +120,17 @@ def plot_explainer():
             boxstyle="round,pad=0.012,rounding_size=0.02",
             linewidth=1.1, edgecolor=MUTED, facecolor="white", zorder=2))
         stage.text(cx, box_y + box_h + 0.06, title, ha="center", va="bottom",
-                   fontsize=11, color=INK, fontweight="bold", zorder=4)
+                   fontsize=12, color=INK, fontweight="bold", zorder=4)
         if caption:
             stage.text(cx, box_y - 0.05, caption, ha="center", va="top",
-                       fontsize=8.2, color=MUTED, linespacing=1.5, zorder=4)
+                       fontsize=9.2, color=MUTED, linespacing=1.5, zorder=4)
 
     def arrow(x0, x1, label):
         stage.annotate("", xy=(x1, box_y + box_h / 2), xytext=(x0, box_y + box_h / 2),
                        arrowprops=dict(arrowstyle="-|>", lw=1.6, color=INK,
                                        shrinkA=0, shrinkB=0), zorder=3)
         stage.text((x0 + x1) / 2, box_y + box_h / 2 + 0.045, label, ha="center",
-                   va="bottom", fontsize=8.5, color=CURVE, fontweight="bold",
+                   va="bottom", fontsize=9.5, color=CURVE, fontweight="bold",
                    linespacing=1.35, zorder=4)
 
     node(xs[0], "IDP sequence\n+ expression level", f"{pipeline.SEQUENCE_LENGTH} aa, scanned over\n{pipeline.NUM_IMAGES} concentrations")
@@ -139,14 +139,14 @@ def plot_explainer():
     node(xs[3], "AUC, AAC", None)
 
     arrow(xs[0] + box_w / 2, xs[1] - box_w / 2, "CELL-FM")
-    arrow(xs[1] + box_w / 2, xs[2] - box_w / 2, "ViT classification\nmodel")
+    arrow(xs[1] + box_w / 2, xs[2] - box_w / 2, "ViT")
     arrow(xs[2] + box_w / 2, xs[3] - box_w / 2, "integrate")
 
     # the last node's caption replaces the bullets that used to sit under the title
     stage.text(xs[3], box_y - 0.05, "AUC — range spent condensed", ha="center", va="top",
-               fontsize=8.2, color=CURVE, fontweight="bold", zorder=4)
+               fontsize=9.2, color=CURVE, fontweight="bold", zorder=4)
     stage.text(xs[3], box_y - 0.16, "AAC — condensation lost to\nreentrant dissolution",
-               ha="center", va="top", fontsize=8.2, color=GAP, fontweight="bold",
+               ha="center", va="top", fontsize=9.2, color=GAP, fontweight="bold",
                linespacing=1.5, zorder=4)
 
     def panel(cx, pad_x=0.020, pad_y=0.055):
@@ -164,11 +164,11 @@ def plot_explainer():
 
     # 1. inputs: the sequence, and the concentration ladder it is scanned over
     a = panel(xs[0]); a.set_axis_off(); a.set_xlim(0, 1); a.set_ylim(0, 1)
-    a.text(0.5, 0.80, "KSFGTPFGGG…", ha="center", va="center", fontsize=8.5,
+    a.text(0.5, 0.80, "KSFGTPFGGG…", ha="center", va="center", fontsize=10.5,
            family="monospace", color=INK)
     a.imshow(np.linspace(0, 1, 128)[None, :], cmap="magma",
              extent=(0.06, 0.94, 0.30, 0.50), aspect="auto")
-    a.text(0.5, 0.13, "low → high", ha="center", va="center", fontsize=7.5, color=MUTED)
+    a.text(0.5, 0.13, "low → high", ha="center", va="center", fontsize=9, color=MUTED)
 
     # 2. a ladder of images, not one -- three rungs stand in for all 512.
     # Size the axes square in inches; imshow would otherwise letterbox each frame
@@ -183,7 +183,7 @@ def plot_explainer():
         ax.imshow(_synthetic_cell(c, seed=k + 1), cmap="magma", vmin=0, vmax=1)
         frame(ax, boxed=False)
     stage.text(xs[1], y0 - 0.035, "low → high concentration", ha="center", va="top",
-               fontsize=7.2, color=MUTED, zorder=4)
+               fontsize=8.6, color=MUTED, zorder=4)
 
     # 3. the curve those calls become
     c_ax = panel(xs[2])
@@ -199,10 +199,19 @@ def plot_explainer():
     d.plot(x, flat, color=GAP, linewidth=1.0, linestyle="--", alpha=0.85)
     d.set_xscale("log"); d.set_xlim(x.min(), x.max()); d.set_ylim(-0.05, 1.08)
     d.set_facecolor("white"); frame(d)
-    d.text(0.33, 0.28, "AUC", transform=d.transAxes, fontsize=9, color=CURVE,
-           fontweight="bold", ha="center")
-    d.text(0.80, 0.62, "AAC", transform=d.transAxes, fontsize=9, color=GAP,
-           fontweight="bold", ha="center")
+    # place each label inside its own shaded region, computed from the curve
+    plateau = np.flatnonzero(curve > 0.9)
+    auc_i = plateau[len(plateau) // 2] if plateau.size else int(np.argmax(curve))
+    d.text(x[auc_i], 0.42, "AUC", fontsize=12, color=CURVE, fontweight="bold",
+           ha="center", va="center")
+    clear = np.flatnonzero((flat - curve > 0.45) & (curve < 0.35))
+    if clear.size:
+        # Sit just inside the open part of the gap, but never so far right that a
+        # centred label overruns the axes -- the midpoint of the region does.
+        lx = (np.log10(x) - np.log10(x[0])) / (np.log10(x[-1]) - np.log10(x[0]))
+        target = min(max(lx[clear[0]] + 0.09, 0.66), 0.78)
+        d.text(x[int(np.argmin(np.abs(lx - target)))], 0.55, "AAC", fontsize=12,
+               color=GAP, fontweight="bold", ha="center", va="center")
 
     return fig