Align axis labels with theory and scenario hints

app.py

@@ -424,6 +424,33 @@ DEFAULT_PROMPTS = {
     "Free form": "Brainstorm creative uses of brain-steered language models in five bullet points."
 }
 
+SCENARIO_AXIS_TEXT = {
+    "Technical writing": {
+        "left_label": "Semantic / Content (meaning-heavy, concrete) [empathetic/actionable tone]",
+        "baseline_label": "Baseline",
+        "right_label": "Syntactic / Function (structure-heavy, abstract) [formal/policy tone]",
+        "left_caption": "*Steered toward meaning (brain semantic side)*",
+        "baseline_caption": "*No brain steering*",
+        "right_caption": "*Steered toward structure (brain syntactic side)*",
+    },
+    "Educational": {
+        "left_label": "Semantic / Content (meaning-heavy, concrete) [analogy/concrete style]",
+        "baseline_label": "Baseline",
+        "right_label": "Syntactic / Function (structure-heavy, abstract) [definition/logical style]",
+        "left_caption": "*Steered toward meaning (brain semantic side)*",
+        "baseline_caption": "*No brain steering*",
+        "right_caption": "*Steered toward structure (brain syntactic side)*",
+    },
+    "Free form": {
+        "left_label": "Semantic / Content (meaning-heavy, concrete)",
+        "baseline_label": "Baseline",
+        "right_label": "Syntactic / Function (structure-heavy, abstract)",
+        "left_caption": "*Steered toward meaning (brain semantic side)*",
+        "baseline_caption": "*No brain steering*",
+        "right_caption": "*Steered toward structure (brain syntactic side)*",
+    },
+}
+
 with gr.Blocks(
     title="Cognitive Proxy",
     theme=gr.themes.Base(
@@ -508,34 +535,52 @@ with gr.Blocks(
 
             with gr.Row():
                 with gr.Column():
-                    gr.HTML('<div class="value-label">Concrete / Analogies</div>')
+                    axis_text = SCENARIO_AXIS_TEXT["Technical writing"]
+                    left_label = gr.HTML(f'<div class="value-label">{axis_text["left_label"]}</div>')
                     output_semantic = gr.Textbox(
                         label="",
                         lines=10,
                         interactive=False,
                         container=False
                     )
-                    gr.Markdown("*Steered toward meaning patterns*", elem_classes=["caption"])
+                    left_caption = gr.Markdown(axis_text["left_caption"], elem_classes=["caption"])
 
                 with gr.Column():
-                    gr.HTML('<div class="value-label">Baseline</div>')
+                    baseline_label = gr.HTML(f'<div class="value-label">{axis_text["baseline_label"]}</div>')
                     output_baseline = gr.Textbox(
                         label="",
                         lines=10,
                         interactive=False,
                         container=False
                     )
-                    gr.Markdown("*No brain steering*", elem_classes=["caption"])
+                    baseline_caption = gr.Markdown(axis_text["baseline_caption"], elem_classes=["caption"])
 
                 with gr.Column():
-                    gr.HTML('<div class="value-label">Abstract / Logical</div>')
+                    right_label = gr.HTML(f'<div class="value-label">{axis_text["right_label"]}</div>')
                     output_syntactic = gr.Textbox(
                         label="",
                         lines=10,
                         interactive=False,
                         container=False
                     )
-                    gr.Markdown("*Steered toward structure patterns*", elem_classes=["caption"])
+                    right_caption = gr.Markdown(axis_text["right_caption"], elem_classes=["caption"])
+
+            def update_axis_labels(selected):
+                data = SCENARIO_AXIS_TEXT.get(selected, SCENARIO_AXIS_TEXT["Free form"])
+                return (
+                    data["left_label"],
+                    data["baseline_label"],
+                    data["right_label"],
+                    data["left_caption"],
+                    data["baseline_caption"],
+                    data["right_caption"],
+                )
+
+            scenario.change(
+                update_axis_labels,
+                inputs=[scenario],
+                outputs=[left_label, baseline_label, right_label, left_caption, baseline_caption, right_caption],
+            )
 
             generate_btn.click(
                 generate_variants,