Attention Explorer: admin panel config, examples from config.json

- Add Attention Explorer section to admin panel (model + examples JSON editor)
- Examples read from config.json at startup (not hardcoded)
- Model description text reads from config
- Admin save handler persists attention_model and attention_examples

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

app.py

@@ -943,6 +943,34 @@ def admin_save_presets(presets_json):
     )
 
 
+def admin_save_attention(attn_model, examples_json):
+    """Save attention explorer settings from admin panel.
+
+    Returns (status_msg, config_json).
+    """
+    try:
+        examples = json.loads(examples_json)
+    except (json.JSONDecodeError, TypeError) as e:
+        cfg = manager.get_config()
+        return f"Invalid JSON: {e}", json.dumps(cfg, indent=2)
+
+    if not isinstance(examples, list):
+        cfg = manager.get_config()
+        return "Examples must be a JSON array", json.dumps(cfg, indent=2)
+
+    manager.update_config(
+        attention_model=attn_model.strip(),
+        attention_examples=examples,
+    )
+    cfg = manager.get_config()
+    return (
+        f"Attention settings saved. Model: {attn_model.strip()}. "
+        f"Note: model change takes effect on next Explore click. "
+        f"Example changes require app restart.",
+        json.dumps(cfg, indent=2),
+    )
+
+
 # ---------------------------------------------------------------------------
 # Build the Gradio app
 # ---------------------------------------------------------------------------
@@ -1234,18 +1262,23 @@ def create_app():
           # ==================================================================
           with gr.Tab("Attention Explorer", id="attn"):
             gr.Markdown("### Attention Explorer")
+            _attn_model_name = cfg.get("attention_model", "gpt2-medium")
             gr.Markdown(
                 "See which words the model pays attention to when processing a sentence. "
-                "Uses GPT-2 Medium (345M parameters, 24 layers, 16 attention heads). "
+                f"Uses `{_attn_model_name}`. "
                 "Click a word to see curved lines connecting it to the words it attended to — "
                 "thicker lines mean stronger attention."
             )
 
-            # Example sentence pairs — labeled by polysemy word
+            # Example sentence pairs — read from config, fall back to defaults
+            _default_attn_examples = [
+                ["bass", "He tuned his bass and plugged into the", "On the lake she caught a bass and pulled it onto the"],
+                ["spring", "She wound the metal spring and the clock began to", "After the long winter the warm spring rain made the flowers"],
+                ["light", "She flipped the switch and the light began to", "The bag was so light she carried it with"],
+            ]
             attn_example_pairs = [
-                ("bass", "He tuned his bass and plugged into the", "On the lake she caught a bass and pulled it onto the"),
-                ("spring", "She wound the metal spring and the clock began to", "After the long winter the warm spring rain made the flowers"),
-                ("light", "She flipped the switch and the light began to", "The bag was so light she carried it with"),
+                tuple(ex) for ex in cfg.get("attention_examples", _default_attn_examples)
+                if isinstance(ex, (list, tuple)) and len(ex) == 3
             ]
 
             attn_example_btns = []
@@ -1359,6 +1392,22 @@ def create_app():
                 admin_save_btn = gr.Button("Save Defaults")
                 admin_save_msg = gr.Markdown("")
 
+                gr.Markdown("---")
+                gr.Markdown("#### Attention Explorer")
+                admin_attn_model = gr.Textbox(
+                    label="Attention model",
+                    value=cfg.get("attention_model", "gpt2-medium"),
+                    info="GPT-2 family: gpt2, gpt2-medium, gpt2-large. Changes take effect on next Explore click (reloads model).",
+                )
+                admin_attn_examples = gr.Code(
+                    value=json.dumps(cfg.get("attention_examples", []), indent=2),
+                    language="json",
+                    interactive=True,
+                    label="Example sentences (JSON: [[\"word\", \"sent_a\", \"sent_b\"], ...])",
+                )
+                admin_attn_save_btn = gr.Button("Save Attention Settings")
+                admin_attn_save_msg = gr.Markdown("")
+
                 gr.Markdown("---")
                 gr.Markdown("#### System Prompt Presets")
                 gr.Markdown(
@@ -1464,6 +1513,13 @@ def create_app():
                 outputs=[admin_presets_msg, admin_config_display, sp_preset, admin_presets],
             )
 
+            # Save attention settings
+            admin_attn_save_btn.click(
+                fn=admin_save_attention,
+                inputs=[admin_attn_model, admin_attn_examples],
+                outputs=[admin_attn_save_msg, admin_config_display],
+            )
+
             # Export slides — uses current Probability Explorer settings
             admin_export_btn.click(
                 fn=generate_slideshow,

config.json

@@ -24,8 +24,20 @@
   "default_attention_sentence": "",
   "default_neighbor_count": 3,
   "attention_examples": [
-    ["bass", "He tuned his bass and plugged into the", "On the lake she caught a bass and pulled it onto the"],
-    ["spring", "She wound the metal spring and the clock began to", "After the long winter the warm spring rain made the flowers"],
-    ["light", "She flipped the switch and the light began to", "The bag was so light she carried it with"]
+    [
+      "bass",
+      "He tuned his bass and plugged into the",
+      "On the lake she caught a bass and pulled it onto the"
+    ],
+    [
+      "spring",
+      "She wound the metal spring and the clock began to",
+      "After the long winter the warm spring rain made the flowers"
+    ],
+    [
+      "light",
+      "She flipped the switch and the light began to",
+      "The bag was so light she carried it"
+    ]
   ]
 }