Upload folder using huggingface_hub

app.py

@@ -679,13 +679,21 @@ def _refresh_futures(s: Session):
     if s.probe is None or s.probe.history_ids is None: return
     s.options = s.probe.generate_previews(s.width, s.preview_len)
 
+def _refresh_headmap(s: Session):
+    """Always run full h * scan + xray after any step that mutates context."""
+    if s.probe is None or s.probe.history_ids is None: return
+    s.scan_data = s.probe.scan_heads(None, None)
+    s.xray = s.probe.perform_xray()
+
 def _refresh_after_step(s: Session):
     _refresh_futures(s)
+    _refresh_headmap(s)
 
 
 def initial_load(model_id: str, system_prompt: str, user_prompt: str, s: Session):
     s.ensure(model_id=model_id, system_prompt=system_prompt or None, user_prompt=user_prompt or "")
     _refresh_futures(s)
+    _refresh_headmap(s)
     s.transcript.append(("(load)", f"loaded {s.probe.model_id} "
                                   f"({s.probe.num_layers}L×{s.probe.num_heads}H)"))
     return panes_html(s) + (status_html(s),)
@@ -694,12 +702,15 @@ def initial_load(model_id: str, system_prompt: str, user_prompt: str, s: Session
 # ── Gradio UI ─────────────────────────────────────────────────────────────
 CSS = """
 .gradio-container { max-width: 100% !important; }
-#cartogemma-grid { display: grid; grid-template-columns: 1fr 2fr; grid-template-rows: auto auto auto; gap: 8px; }
+/* Compact top control row */
+#topbar .gr-textbox textarea, #topbar .gr-textbox input { font-size: 12px !important; padding: 4px 6px !important; }
+#topbar label { font-size: 11px !important; }
+#topbar .gr-form { padding: 4px !important; }
 .pane { border: 1px solid #444; border-radius: 4px; background: #0d1117; color: #c9d1d9;
         font-family: 'JetBrains Mono', 'Fira Code', Consolas, monospace; font-size: 12px; }
 .pane h3 { margin: 0; padding: 4px 8px; background: #161b22; border-bottom: 1px solid #30363d;
            font-size: 11px; letter-spacing: 0.5px; text-transform: uppercase; color: #8b949e; }
-.pane-body { padding: 6px 8px; max-height: 360px; overflow: auto; }
+.pane-body { padding: 6px 8px; max-height: 420px; overflow: auto; }
 .pane-body.scroll { overflow: auto; }
 .dim { color: #6e7681; }
 .context-text { white-space: pre-wrap; word-break: break-word; margin-top: 4px; }
@@ -736,41 +747,46 @@ HELP = """\
 def build_ui():
     with gr.Blocks(title="Cartogemma", css=CSS, theme=gr.themes.Base()) as demo:
         gr.Markdown("# Cartogemma  \n*Mechanistic probe (logit lens · head map · futures · rank sparkline) "
-                    "wrapping `google/gemma-3-1b-it`. Layout faithful to `cartographer3.py`.*")
+                    "on the Gemma-3 family. Layout faithful to `cartographer3.py`.*")
 
         session = gr.State(Session())
 
-        with gr.Row():
-            with gr.Column(scale=2):
-                model_id = gr.Textbox(value=DEFAULT_MODEL_ID, label="Model ID",
-                                      info="Gated model — set HF_TOKEN as a Space secret.")
-            with gr.Column(scale=3):
-                system_prompt = gr.Textbox(label="System prompt", value="", lines=1)
-            with gr.Column(scale=3):
-                user_prompt = gr.Textbox(label="Initial user prompt", value="", lines=1)
-            with gr.Column(scale=1, min_width=120):
-                load_btn = gr.Button("Load model", variant="primary")
+        # ── Compact top bar: model | system | user | load ──
+        with gr.Row(elem_id="topbar"):
+            model_id = gr.Textbox(value=DEFAULT_MODEL_ID, label="Model",
+                                  lines=1, max_lines=1, scale=3, container=True,
+                                  show_label=True)
+            system_prompt = gr.Textbox(label="System", value="",
+                                       lines=1, max_lines=3, scale=4, container=True)
+            user_prompt = gr.Textbox(label="User", value="",
+                                     lines=1, max_lines=3, scale=5, container=True)
+            load_btn = gr.Button("Load", variant="primary", scale=1, min_width=80)
 
         muted_html = gr.HTML()
 
-        with gr.Row():
-            ctx_pane = gr.HTML(label="Context", value="<div class='pane'><h3>Context</h3>"
-                               "<div class='pane-body'><i>Click <b>Load model</b> to begin.</i></div></div>")
-            hm_pane = gr.HTML(label="Head Map", value="<div class='pane'><h3>Head Map "
-                               "(pre-stream | xray | delta-v | L<sub>full</sub>)</h3>"
-                               "<div class='pane-body'><i>Run <code>h *</code> after loading.</i></div></div>")
+        # ── Context: full width ──
+        ctx_pane = gr.HTML(value="<div class='pane'><h3>Context</h3>"
+                           "<div class='pane-body'><i>Click <b>Load</b> to begin.</i></div></div>")
 
+        # ── Futures + Head Map: side by side ──
         with gr.Row():
-            fut_pane = gr.HTML(label="Futures", value="<div class='pane'><h3>Futures</h3>"
-                               "<div class='pane-body'><i>Top-k branches will appear here.</i></div></div>")
-            spark_pane = gr.HTML(label="Sparkline", value="<div class='pane'><h3>Sparkline</h3>"
-                                 "<div class='pane-body'><i>Run <code>spark &lt;token&gt;</code>.</i></div></div>")
+            with gr.Column(scale=3, min_width=280):
+                fut_pane = gr.HTML(value="<div class='pane'><h3>Futures (top-k)</h3>"
+                                   "<div class='pane-body'><i>Top-k branches will appear here.</i></div></div>")
+            with gr.Column(scale=7):
+                hm_pane = gr.HTML(value="<div class='pane'><h3>Head Map "
+                                  "(pre-stream | xray | delta-v | L<sub>full</sub>)</h3>"
+                                  "<div class='pane-body'><i>Auto-runs after every step.</i></div></div>")
+
+        # ── Sparkline: full width below ──
+        spark_pane = gr.HTML(value="<div class='pane'><h3>Sparkline</h3>"
+                             "<div class='pane-body'><i>Run <code>spark &lt;token&gt;</code>.</i></div></div>")
 
         gr.Markdown(HELP)
 
         with gr.Row():
             cmd = gr.Textbox(label="CMD", placeholder="1-N | h * | top 1000,3,1 | spark the | mute 5 2 | …",
-                             scale=8, autofocus=True)
+                             scale=8, autofocus=True, lines=1, max_lines=1)
             run_btn = gr.Button("Run", scale=1, variant="primary")
 
         status = gr.HTML(value="<div class='pane'><h3>Transcript</h3>"