Replace static constellations with interactive 3D Atlas (drag to rotate, color by food group or cuisine, basket diamonds mint, neighbours amber)

app.py

@@ -999,6 +999,116 @@ def cultural_context(ingredient, sibling="chem", k=4):
 # Constellations: sibling alignment + recipe constellation
 # =====================================================================
 
+def render_3d_atlas(sibling, basket, show_neighbours=True, k=8, color_mode="food group"):
+    """Interactive 3D UMAP. Drag/scroll to navigate. Basket pops in mint, neighbours amber."""
+    m = MODELS[sibling]
+    coords2 = UMAP_DATA[sibling]
+    n = len(NAMES_BY_IDX)
+    # third axis = standardised PC1 of the embedding
+    E = m.E - m.E.mean(axis=0, keepdims=True)
+    _, _, Vt = np.linalg.svd(E, full_matrices=False)
+    pc1 = E @ Vt[0]
+    pc1 = (pc1 - pc1.mean()) / (pc1.std() + 1e-9)
+    scale = (coords2.max() - coords2.min()) * 0.25
+    z = (pc1 * scale).astype(np.float32)
+
+    # Colors: by food group OR by closest cuisine
+    if color_mode == "cuisine":
+        cuisine_pole_keys = [c for c in _CUISINES if f"cuisine:{c}" in m.supervised_poles]
+        cuisine_poles = np.stack([_unit(m.supervised_poles[f"cuisine:{c}"]) for c in cuisine_pole_keys])
+        Xn = m.E / np.linalg.norm(m.E, axis=1, keepdims=True)
+        sims = Xn @ cuisine_poles.T
+        best = sims.argmax(axis=1)
+        palette = ["#288B79","#F4B86E","#E8C0E8","#9BC9E8","#D7E89B","#FFAA8A","#A8D5CA","#E8E0A0"]
+        colors = [palette[best[i] % len(palette)] for i in range(n)]
+        hover_text = [
+            f"<b>{NAMES_BY_IDX[i]}</b><br>closest cuisine: {cuisine_pole_keys[best[i]].replace('_',' ')}<br>group: {FOOD_GROUPS[i]}"
+            for i in range(n)
+        ]
+    else:
+        colors = [FG_COLORS.get(fg, "#cccccc") for fg in FOOD_GROUPS]
+        hover_text = [f"<b>{NAMES_BY_IDX[i]}</b><br>group: {FOOD_GROUPS[i]}" for i in range(n)]
+
+    basket_set = set(basket or [])
+    basket_idxs = [m.vocab[b] for b in (basket or []) if b in m.vocab]
+    neighbour_set = set()
+    if show_neighbours and basket_idxs:
+        c = _basket_centroid(m, basket)
+        if c is not None:
+            for nm, _ in _topk(m, c, int(k), exclude=basket):
+                neighbour_set.add(nm)
+
+    keep = lambda i: NAMES_BY_IDX[i] not in basket_set and NAMES_BY_IDX[i] not in neighbour_set
+    bg_idx = [i for i in range(n) if keep(i)]
+
+    fig = go.Figure()
+    fig.add_trace(go.Scatter3d(
+        x=[float(coords2[i,0]) for i in bg_idx],
+        y=[float(coords2[i,1]) for i in bg_idx],
+        z=[float(z[i]) for i in bg_idx],
+        mode="markers",
+        marker=dict(size=3, color=[colors[i] for i in bg_idx], opacity=0.55, line=dict(width=0)),
+        text=[hover_text[i] for i in bg_idx],
+        hovertemplate="%{text}<extra></extra>",
+        name="ingredients", showlegend=False,
+    ))
+    if neighbour_set:
+        ni = [i for i in range(n) if NAMES_BY_IDX[i] in neighbour_set]
+        fig.add_trace(go.Scatter3d(
+            x=[float(coords2[i,0]) for i in ni],
+            y=[float(coords2[i,1]) for i in ni],
+            z=[float(z[i]) for i in ni],
+            mode="markers+text",
+            marker=dict(size=8, color="#F59E0B", opacity=0.96,
+                        line=dict(color="#ffffff", width=1.5),
+                        symbol="circle"),
+            text=[NAMES_BY_IDX[i].replace("_"," ") for i in ni],
+            textposition="top center",
+            textfont=dict(size=11, color="#1f2937", family="Inter"),
+            hovertemplate="<b>%{text}</b> (neighbour)<extra></extra>",
+            name=f"top-{int(k)} neighbours",
+        ))
+    if basket_idxs:
+        fig.add_trace(go.Scatter3d(
+            x=[float(coords2[i,0]) for i in basket_idxs],
+            y=[float(coords2[i,1]) for i in basket_idxs],
+            z=[float(z[i]) for i in basket_idxs],
+            mode="markers+text",
+            marker=dict(size=15, color=KAIKAKU_ACCENT, symbol="diamond",
+                        line=dict(color="#0f172a", width=2)),
+            text=[NAMES_BY_IDX[i].replace("_"," ") for i in basket_idxs],
+            textposition="top center",
+            textfont=dict(size=13, color="#0f172a", family="Inter"),
+            hovertemplate="<b>%{text}</b> (basket)<extra></extra>",
+            name="basket",
+        ))
+
+    fig.update_layout(
+        title=dict(
+            text=f"<b>3D Atlas</b> · Epicure-{sibling.capitalize()} · 1,790 ingredients · drag to rotate, scroll to zoom",
+            font=dict(size=13, color="#0f172a", family="Inter"), x=0.02, xanchor="left",
+        ),
+        scene=dict(
+            xaxis=dict(title="UMAP 1", color="#475569", gridcolor="#e2e8f0",
+                       backgroundcolor="#ffffff", showspikes=False, zeroline=False),
+            yaxis=dict(title="UMAP 2", color="#475569", gridcolor="#e2e8f0",
+                       backgroundcolor="#ffffff", showspikes=False, zeroline=False),
+            zaxis=dict(title="PC1", color="#475569", gridcolor="#e2e8f0",
+                       backgroundcolor="#ffffff", showspikes=False, zeroline=False),
+            bgcolor="#ffffff",
+            camera=dict(up=dict(x=0, y=0, z=1), eye=dict(x=1.6, y=1.6, z=1.0)),
+            aspectmode="cube",
+        ),
+        height=720,
+        margin=dict(l=10, r=10, t=46, b=10),
+        paper_bgcolor="#ffffff",
+        legend=dict(orientation="h", yanchor="bottom", y=0.0, xanchor="right", x=1.0,
+                    font=dict(color="#0f172a", size=11), bgcolor="rgba(255,255,255,0.85)",
+                    bordercolor="#e2e8f0", borderwidth=1),
+    )
+    return fig
+
+
 def render_sibling_alignment(ingredient):
     if not ingredient or ingredient not in MODELS["cooc"].vocab:
         fig, ax = _gallery_axes(figsize=(16, 5))
@@ -2004,38 +2114,53 @@ Three sibling ingredient embeddings from [arXiv:2605.22391](https://arxiv.org/ab
                                   outputs=[cx_df, cx_md])
 
         # ---------- Tab: CONSTELLATIONS ----------
-        with gr.Tab("Constellations"):
-            with gr.Tabs():
-                with gr.Tab("Sibling alignment"):
-                    gr.Markdown("Same ingredient on all three UMAPs. The spectrum-of-models thesis as a single image.")
-                    with gr.Row():
-                        sa_ing = gr.Dropdown(choices=ALL_INGREDIENTS, value="basil", label="Ingredient")
-                        sa_btn = gr.Button("Render", variant="primary")
-                    sa_plot = gr.Plot(label="")
-                    sa_btn.click(render_sibling_alignment, inputs=[sa_ing], outputs=[sa_plot], show_progress="full")
-                    sa_ing.change(render_sibling_alignment, inputs=[sa_ing], outputs=[sa_plot])
-                    demo.load(render_sibling_alignment, inputs=[sa_ing], outputs=[sa_plot])
-
-                with gr.Tab("Recipe constellation"):
-                    gr.Markdown("A recipe drawn as a constellation on the UMAP, edges to nearest basket-mates.")
-                    with gr.Row():
-                        rc_sib = gr.Radio(choices=["cooc","core","chem"], value="chem", label="Sibling")
-                        rc_ings = gr.Dropdown(choices=ALL_INGREDIENTS, multiselect=True,
-                                              value=["tomato","basil","garlic","olive_oil","mozzarella_cheese"],
-                                              label="Recipe ingredients", max_choices=12)
-                        rc_btn = gr.Button("Render", variant="primary")
-                    rc_plot = gr.Plot(label="")
-                    rc_btn.click(render_recipe_constellation, inputs=[rc_sib, rc_ings], outputs=[rc_plot], show_progress="full")
-                    demo.load(render_recipe_constellation, inputs=[rc_sib, rc_ings], outputs=[rc_plot])
-                    gr.Examples(
-                        examples=[
-                            ["chem", ["tomato","basil","garlic","olive_oil","mozzarella_cheese"]],
-                            ["chem", ["chicken","lemongrass","coconut_milk","fish_sauce","lime","ginger","chili_pepper"]],
-                            ["chem", ["beef","cumin","coriander","onion","garlic","tomato","cinnamon"]],
-                            ["core", ["chocolate","strawberry","cream","sugar","vanilla"]],
-                        ],
-                        inputs=[rc_sib, rc_ings], label="Try a recipe",
-                    )
+        with gr.Tab("3D Atlas"):
+            gr.Markdown(
+                "**Interactive 3D map of all 1,790 ingredients.** "
+                "Drag with mouse to rotate. Scroll to zoom. Hover for ingredient names. "
+                "Basket members appear as mint diamonds; their nearest neighbours pop in amber."
+            )
+            with gr.Row():
+                atl_sib = gr.Radio(choices=["cooc","core","chem"], value="chem", label="Sibling")
+                atl_color = gr.Radio(choices=["food group", "cuisine"], value="food group",
+                                     label="Color points by")
+                atl_basket = gr.Dropdown(
+                    choices=ALL_INGREDIENTS,
+                    value=["miso","basil","chocolate","tomato"],
+                    label="Highlight these ingredients",
+                    multiselect=True, max_choices=15,
+                )
+            with gr.Row():
+                atl_show_nb = gr.Checkbox(value=True, label="Show top-K neighbours")
+                atl_k = gr.Slider(3, 20, value=8, step=1, label="K")
+                atl_btn = gr.Button("Update", variant="primary")
+            atl_plot = gr.Plot(
+                value=render_3d_atlas("chem", ["miso","basil","chocolate","tomato"], True, 8, "food group"),
+                label="",
+            )
+            atl_btn.click(render_3d_atlas,
+                          inputs=[atl_sib, atl_basket, atl_show_nb, atl_k, atl_color],
+                          outputs=atl_plot, show_progress="minimal")
+            atl_sib.change(render_3d_atlas,
+                           inputs=[atl_sib, atl_basket, atl_show_nb, atl_k, atl_color],
+                           outputs=atl_plot, show_progress="minimal")
+            atl_color.change(render_3d_atlas,
+                             inputs=[atl_sib, atl_basket, atl_show_nb, atl_k, atl_color],
+                             outputs=atl_plot, show_progress="minimal")
+            atl_basket.change(render_3d_atlas,
+                              inputs=[atl_sib, atl_basket, atl_show_nb, atl_k, atl_color],
+                              outputs=atl_plot, show_progress="minimal")
+            gr.Examples(
+                examples=[
+                    ["chem", "food group", ["miso","basil","chocolate","tomato"], True, 8],
+                    ["chem", "cuisine", ["chicken","lemongrass","coconut_milk","fish_sauce"], True, 10],
+                    ["core", "cuisine", ["tomato","mozzarella_cheese","basil","olive_oil"], True, 8],
+                    ["chem", "food group", ["cumin","coriander","turmeric","cardamom","fenugreek_seed"], True, 10],
+                    ["cooc", "food group", ["red_wine","brandy","whiskey","bourbon","cognac"], True, 8],
+                ],
+                inputs=[atl_sib, atl_color, atl_basket, atl_show_nb, atl_k],
+                label="Try one of these baskets",
+            )
 
         # ---------- Tab: PAPER STATS ----------
         with gr.Tab("Paper stats"):