Enhance interactive plots, registry, and smart recommendations

Improve interactive plot panel layout and rendering, update registry
with expanded module support, refine smart compare/recommend logic,
and update next_steps guidance and script editor.

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

Author: brycewang-stanford

src/statspai/core/next_steps.py

@@ -584,43 +584,56 @@ def causal_next_steps(result) -> List[Step]:
 # ====================================================================== #
 
 def _detect_family(method: str) -> str:
-    """Classify a method string into a family."""
+    """Classify a method string into a family.
+
+    Order matters: more specific patterns are checked before broader ones
+    (e.g. ``sdid`` → synth before ``did`` → did).
+    """
     m = method.lower()
 
+    # --- Synthetic control (check before DID: "sdid" contains "did") ---
+    if any(k in m for k in ("synth", "scm", "sdid", "augsynth",
+                             "causal impact", "structural time")):
+        return "synth"
+
+    # --- DID ---
     if any(k in m for k in ("did", "diff", "callaway", "sun_abraham",
                              "stagger", "imputation", "wooldridge",
                              "chaisemartin", "changes-in-changes",
                              "cic", "stacked")):
         return "did"
 
+    # --- RD ---
     if any(k in m for k in ("rd", "discontinuity", "rdrobust", "kink",
                              "rdit")):
         return "rd"
 
-    if any(k in m for k in ("iv ", "2sls", "instrumental", "bartik",
+    # --- IV (match bare "iv" and "iv ..." via word boundary) ---
+    if any(k in m for k in ("2sls", "instrumental", "bartik",
                              "deepiv", "shift-share")):
         return "iv"
+    import re
+    if re.search(r'\biv\b', m):
+        return "iv"
 
+    # --- Matching / IPW ---
     if any(k in m for k in ("match", "psm", "cem", "mahalanobis",
                              "ipw", "entropy", "aipw", "tmle")):
         return "matching"
 
-    if any(k in m for k in ("synth", "scm", "sdid", "augsynth")):
-        return "synth"
-
+    # --- DML ---
     if any(k in m for k in ("dml", "double", "debiased")):
         return "dml"
 
+    # --- HTE / Meta-learners ---
     if any(k in m for k in ("metalearner", "slearner", "tlearner",
                              "xlearner", "rlearner", "drlearner",
                              "causal forest", "bcf", "cate",
                              "tarnet", "cfrnet", "dragonnet")):
         return "hte"
 
+    # --- Mediation ---
     if any(k in m for k in ("mediat",)):
         return "mediation"
 
-    if any(k in m for k in ("causal impact", "structural time")):
-        return "synth"
-
     return "generic"

src/statspai/plots/_script_editor.py

@@ -111,7 +111,7 @@ def _on_submit(text):
                 old = target.get_text()
                 target.set_text(text)
                 from .interactive import EditRecord
-                editor.edits.append(EditRecord(
+                editor._record_edit(EditRecord(
                     f'{prefix}.text', 'text', old, text,
                     f"# annotation text changed to {text!r}",
                 ))
@@ -123,8 +123,8 @@ def _on_submit(text):
     textbox.on_submit(_on_submit)
 
     # Generate Code button
-    ax_codebtn = fig.add_axes([0.65, 0.04, 0.15, 0.05])
-    btn_code = Button(ax_codebtn, 'Get Code')
+    ax_codebtn = fig.add_axes([0.62, 0.04, 0.12, 0.05])
+    btn_code = Button(ax_codebtn, 'Code')
 
     def _on_code(event):
         print("\n" + "=" * 50)
@@ -134,15 +134,27 @@ def _on_code(event):
     btn_code.on_clicked(_on_code)
 
     # Undo button
-    ax_undobtn = fig.add_axes([0.82, 0.04, 0.1, 0.05])
+    ax_undobtn = fig.add_axes([0.75, 0.04, 0.1, 0.05])
     btn_undo = Button(ax_undobtn, 'Undo')
 
     def _on_undo(event):
         editor.undo()
-        print("[StatsPAI] Undone last edit")
+        print(f"[StatsPAI] Undone last edit "
+              f"({len(editor._redo_stack)} redo available)")
 
     btn_undo.on_clicked(_on_undo)
 
+    # Redo button
+    ax_redobtn = fig.add_axes([0.86, 0.04, 0.1, 0.05])
+    btn_redo = Button(ax_redobtn, 'Redo')
+
+    def _on_redo(event):
+        editor.redo()
+        print(f"[StatsPAI] Redone edit "
+              f"({len(editor._redo_stack)} redo remaining)")
+
+    btn_redo.on_clicked(_on_redo)
+
     # Grid toggle button
     ax_gridbtn = fig.add_axes([0.65, 0.10, 0.12, 0.04])
     btn_grid = Button(ax_gridbtn, 'Grid')
@@ -155,7 +167,7 @@ def _on_grid(event):
     btn_grid.on_clicked(_on_grid)
 
     # Spine toggle
-    ax_spinebtn = fig.add_axes([0.78, 0.10, 0.14, 0.04])
+    ax_spinebtn = fig.add_axes([0.78, 0.10, 0.18, 0.04])
     btn_spine = Button(ax_spinebtn, 'Spines')
     _spine_state = {'minimal': False}
 
@@ -193,6 +205,7 @@ def _on_close(event):
         'textbox': textbox,
         'btn_code': btn_code,
         'btn_undo': btn_undo,
+        'btn_redo': btn_redo,
         'btn_grid': btn_grid,
         'btn_spine': btn_spine,
     }