batplot 1.7.28__py3-none-any.whl → 1.8.1__py3-none-any.whl

batplot/operando_ec_interactive.py

@@ -18,6 +18,7 @@ from typing import Tuple, Dict, Optional, Any
 import json
 import os
 import time
+import sys
 
 import matplotlib.pyplot as plt
 from matplotlib.colors import LinearSegmentedColormap
@@ -38,7 +39,38 @@ from .color_utils import (
     resolve_color_token,
     manage_user_colors,
 )
-from .utils import choose_style_file
+from .utils import choose_style_file, convert_label_shortcuts
+
+
+class _FilterIMKWarning:
+    """Filter that suppresses macOS IMKCFRunLoopWakeUpReliable warnings while preserving other errors."""
+    def __init__(self, original_stderr):
+        self.original_stderr = original_stderr
+    
+    def write(self, message):
+        # Filter out the harmless macOS IMK warning
+        if 'IMKCFRunLoopWakeUpReliable' not in message:
+            self.original_stderr.write(message)
+    
+    def flush(self):
+        self.original_stderr.flush()
+
+
+def _safe_input(prompt: str = "") -> str:
+    """Wrapper around input() that suppresses macOS IMKCFRunLoopWakeUpReliable warnings.
+    
+    This is a harmless macOS system message that appears when using input() in terminals.
+    """
+    # Filter stderr to hide macOS IMK warnings while preserving other errors
+    original_stderr = sys.stderr
+    sys.stderr = _FilterIMKWarning(original_stderr)
+    try:
+        result = input(prompt)
+        return result
+    except (KeyboardInterrupt, EOFError):
+        raise
+    finally:
+        sys.stderr = original_stderr
 
 
 def _axis_tick_width(axis_obj, which: str = 'major'):
@@ -322,7 +354,7 @@ def _draw_custom_colorbar(cbar_ax, im, label='Intensity', label_mode='normal'):
 
 def _update_custom_colorbar(cbar_ax, im=None, label=None, label_mode=None):
     """Update the custom colorbar when colormap or limits change.
-    
+
     Args:
         cbar_ax: Axes object containing the colorbar
         im: Optional AxesImage object (if None, uses stored reference)
@@ -333,17 +365,82 @@ def _update_custom_colorbar(cbar_ax, im=None, label=None, label_mode=None):
         im = getattr(cbar_ax, '_colorbar_im', None)
         if im is None:
             return
-    
+
     if label is None:
         label = getattr(cbar_ax, '_colorbar_label', 'Intensity')
-    
+
     if label_mode is None:
         label_mode = getattr(cbar_ax, '_colorbar_label_mode', 'normal')
-    
+
     # Redraw the colorbar
     _draw_custom_colorbar(cbar_ax, im, label, label_mode)
 
 
+def _safe_set_clim(im, vmin, vmax):
+    """Safely set color limits without triggering matplotlib colorbar callbacks.
+    
+    This wrapper around im.set_clim() prevents the NotImplementedError: cannot remove artist
+    error when using custom colorbars. The issue occurs because matplotlib's callback system
+    tries to update the colorbar when set_clim() is called, but our custom colorbar drawing
+    has already cleared the axes, causing the update to fail.
+    
+    This function suppresses the callback traceback by redirecting stderr to a null device
+    during the set_clim() call, preventing matplotlib's callback system from printing
+    tracebacks to the terminal.
+    
+    Args:
+        im: AxesImage object
+        vmin: Minimum value for color scale
+        vmax: Maximum value for color scale
+    """
+    import sys
+    import os
+    from io import StringIO
+    
+    # Create a null device for stderr redirection
+    class NullDevice:
+        def write(self, s):
+            pass
+        def flush(self):
+            pass
+        def close(self):
+            pass
+    
+    # Suppress matplotlib's exception printing by redirecting stderr AND excepthook
+    old_stderr = sys.stderr
+    old_excepthook = sys.excepthook
+    null_dev = NullDevice()
+    
+    # Create a no-op excepthook that suppresses all exceptions
+    def suppress_excepthook(exc_type, exc_value, exc_traceback):
+        # Only suppress if it's the specific error we're looking for
+        if exc_type == NotImplementedError and 'cannot remove artist' in str(exc_value).lower():
+            return  # Suppress this specific exception
+        # For any other exception, use the original handler
+        old_excepthook(exc_type, exc_value, exc_traceback)
+    
+    sys.stderr = null_dev
+    sys.excepthook = suppress_excepthook
+    
+    try:
+        # Set the color limits - matplotlib's callback will try to print traceback
+        # but both stderr and excepthook are suppressed
+        im.set_clim(vmin, vmax)
+        # The operation succeeds; any tracebacks are suppressed
+    except NotImplementedError:
+        # Suppress the specific error - color limits were still updated successfully
+        pass
+    except Exception:
+        # For any other unexpected error, restore handlers and re-raise
+        sys.stderr = old_stderr
+        sys.excepthook = old_excepthook
+        raise
+    finally:
+        # Always restore both stderr and excepthook
+        sys.stderr = old_stderr
+        sys.excepthook = old_excepthook
+
+
 def _detach_mpl_colorbar_callbacks(cbar, im) -> None:
     """Detach a Matplotlib Colorbar from its mappable callbacks.
 
@@ -363,7 +460,7 @@ def _detach_mpl_colorbar_callbacks(cbar, im) -> None:
         if cax is not None and getattr(cax, '_bp_detached_mpl_colorbar', False):
             return
 
-        # Matplotlib's Colorbar stores the callback id as `_cid` (most versions).
+        # APPROACH 1: Try to find and disconnect the callback ID
         cid = None
         for attr in ('_cid', '_cid_colorbar', 'cid'):
             try:
@@ -374,7 +471,6 @@ def _detach_mpl_colorbar_callbacks(cbar, im) -> None:
             except Exception:
                 pass
 
-        # Disconnect from the ScalarMappable callback registry.
         if cid is not None:
             try:
                 cbreg = getattr(im, 'callbacksSM', None)
@@ -383,7 +479,38 @@ def _detach_mpl_colorbar_callbacks(cbar, im) -> None:
             except Exception:
                 pass
 
-        # Prevent future built-in updates (best-effort; safe for mock colorbars too).
+        # APPROACH 2: Disconnect ALL callbacks from the image (nuclear option)
+        try:
+            cbreg = getattr(im, 'callbacksSM', None)
+            if cbreg is not None:
+                # Try to clear all callbacks
+                if hasattr(cbreg, 'callbacks'):
+                    try:
+                        cbreg.callbacks.clear()
+                    except Exception:
+                        pass
+                # Also try the _signals dict if it exists
+                if hasattr(cbreg, '_signals'):
+                    try:
+                        for signal_dict in cbreg._signals.values():
+                            if hasattr(signal_dict, 'clear'):
+                                signal_dict.clear()
+                    except Exception:
+                        pass
+        except Exception:
+            pass
+
+        # APPROACH 3: Monkey-patch the update_normal method to be a no-op
+        # This is the most reliable approach for preventing the callback
+        try:
+            if hasattr(cbar, 'update_normal'):
+                def _noop_update(*args, **kwargs):
+                    pass
+                cbar.update_normal = _noop_update
+        except Exception:
+            pass
+
+        # APPROACH 4: Prevent future built-in updates by nulling internal state
         try:
             if hasattr(cbar, 'mappable'):
                 cbar.mappable = None
@@ -675,7 +802,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 
                 if intensity_max > intensity_min:
                     # Update color limits
-                    im.set_clim(intensity_min, intensity_max)
+                    _safe_set_clim(im, intensity_min, intensity_max)
                     
                     # Update colorbar if available
                     try:
@@ -1049,7 +1176,10 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 clim = im.get_clim()
             except Exception:
                 clim = None
-            cmap_name = getattr(im.get_cmap(), 'name', None)
+            # Get colormap name: first check if we stored it explicitly, otherwise try to get from colormap object
+            cmap_name = getattr(im, '_operando_cmap_name', None)
+            if cmap_name is None:
+                cmap_name = getattr(im.get_cmap(), 'name', None)
             # EC mode and caches (only if ec_ax exists)
             if ec_ax is not None:
                 mode = getattr(ec_ax, '_ec_y_mode', 'time')
@@ -1263,12 +1393,15 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             cbar.solids = None
                     except Exception:
                         pass
-                    lo, hi = snap['clim']; im.set_clim(float(lo), float(hi))
+                    lo, hi = snap['clim']; _safe_set_clim(im, float(lo), float(hi))
             except Exception:
                 pass
             try:
                 if snap.get('cmap'):
-                    im.set_cmap(snap['cmap'])
+                    cmap_name = snap['cmap']
+                    im.set_cmap(cmap_name)
+                    # Store the colormap name explicitly so it can be retrieved reliably when saving
+                    setattr(im, '_operando_cmap_name', cmap_name)
                     if cbar is not None:
                         _update_custom_colorbar(cbar.ax, im)
             except Exception:
@@ -1733,7 +1866,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             print("Crosshair OFF.")
     while True:
         try:
-            cmd = input("Press a key: ").strip().lower()
+            cmd = _safe_input("Press a key: ").strip().lower()
         except (KeyboardInterrupt, EOFError):
             print("\n\nExiting interactive menu...")
             break
@@ -1741,7 +1874,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             continue
         if cmd == 'q':
             try:
-                ans = input("Quit interactive? Remember to save (e=export, s=save). Quit now? (y/n): ").strip().lower()
+                ans = _safe_input("Quit interactive? Remember to save (e=export, s=save). Quit now? (y/n): ").strip().lower()
             except Exception:
                 ans = 'y'
             if ans == 'y':
@@ -1784,9 +1917,9 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 
                 last_figure_path = getattr(fig, '_last_figure_export_path', None)
                 if last_figure_path:
-                    fname = input("Export filename (default .svg if no extension), number to overwrite, or o to overwrite last (q=cancel): ").strip()
+                    fname = _safe_input("Export filename (default .svg if no extension), number to overwrite, or o to overwrite last (q=cancel): ").strip()
                 else:
-                    fname = input("Export filename (default .svg if no extension) or number to overwrite (q=cancel): ").strip()
+                    fname = _safe_input("Export filename (default .svg if no extension) or number to overwrite (q=cancel): ").strip()
                 if not fname or fname.lower() == 'q':
                     print_menu(); continue
                 
@@ -1799,7 +1932,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     if not os.path.exists(last_figure_path):
                         print(f"Previous export file not found: {last_figure_path}")
                         print_menu(); continue
-                    yn = input(f"Overwrite '{os.path.basename(last_figure_path)}'? (y/n): ").strip().lower()
+                    yn = _safe_input(f"Overwrite '{os.path.basename(last_figure_path)}'? (y/n): ").strip().lower()
                     if yn != 'y':
                         print_menu(); continue
                     target = last_figure_path
@@ -1810,7 +1943,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     idx = int(fname)
                     if 1 <= idx <= len(files):
                         name = files[idx-1]
-                        yn = input(f"Overwrite '{name}'? (y/n): ").strip().lower()
+                        yn = _safe_input(f"Overwrite '{name}'? (y/n): ").strip().lower()
                         if yn != 'y':
                             print_menu(); continue
                         target = file_list[idx-1][1]  # Full path from list
@@ -1889,7 +2022,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     cb_h_offset = getattr(cbar.ax, '_cb_h_offset_in', 0.0)
                     ec_h_offset = getattr(ec_ax, '_ec_h_offset_in', 0.0)
                     print(f"Toggle: 1=colorbar, 2=EC panel, 3=both, 4=colorbar label mode, 5=colorbar label text, m=move horizontal position (cb:{cb_h_offset:.3f}\", ec:{ec_h_offset:.3f}\"), q=cancel")
-                    choice = input("v> ").strip().lower()
+                    choice = _safe_input("v> ").strip().lower()
                     if choice == '1':
                         # Toggle colorbar
                         cb_vis = cbar.ax.get_visible()
@@ -1923,7 +2056,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         # Change colorbar label text
                         current_label = getattr(cbar.ax, '_colorbar_label', 'Intensity')
                         print(f"Current colorbar label: {current_label}")
-                        new_label = input("New colorbar label (blank to keep): ").strip()
+                        new_label = _safe_input("New colorbar label (blank to keep): ").strip()
                         if new_label:
                             cbar.ax._colorbar_label = new_label
                             try:
@@ -1944,12 +2077,12 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             if ec_ax is not None:
                                 print(f"  EC panel offset: {ec_h_offset:.3f}\" (positive=right, negative=left)")
                             print("Commands: c=colorbar, e=EC panel, q=back")
-                            sub = input("m> ").strip().lower()
+                            sub = _safe_input("m> ").strip().lower()
                             if not sub or sub == 'q':
                                 break
                             if sub == 'c':
                                 try:
-                                    new_offset = input(f"Enter colorbar horizontal offset in inches (current: {cb_h_offset:.3f}\"): ").strip()
+                                    new_offset = _safe_input(f"Enter colorbar horizontal offset in inches (current: {cb_h_offset:.3f}\"): ").strip()
                                     if new_offset:
                                         cb_h_offset = float(new_offset)
                                         setattr(cbar.ax, '_cb_h_offset_in', cb_h_offset)
@@ -1957,16 +2090,20 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                         _apply_group_layout_inches(fig, ax, cbar.ax, ec_ax, ax_w_in, ax_h_in, cb_w_in, cb_gap_in, ec_gap_in, ec_w_in)
                                         fig.canvas.draw_idle()
                                         print(f"Colorbar horizontal offset set to {cb_h_offset:.3f}\"")
+                                    # Continue in loop to show menu again
+                                    continue
                                 except ValueError:
                                     print("Invalid input. Enter a number.")
+                                    continue
                                 except Exception as e:
                                     print(f"Error: {e}")
+                                    continue
                             elif sub == 'e':
                                 if ec_ax is None:
                                     print("EC panel not available.")
                                     continue
                                 try:
-                                    new_offset = input(f"Enter EC panel horizontal offset in inches (current: {ec_h_offset:.3f}\"): ").strip()
+                                    new_offset = _safe_input(f"Enter EC panel horizontal offset in inches (current: {ec_h_offset:.3f}\"): ").strip()
                                     if new_offset:
                                         ec_h_offset = float(new_offset)
                                         setattr(ec_ax, '_ec_h_offset_in', ec_h_offset)
@@ -1974,19 +2111,24 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                         _apply_group_layout_inches(fig, ax, cbar.ax, ec_ax, ax_w_in, ax_h_in, cb_w_in, cb_gap_in, ec_gap_in, ec_w_in)
                                         fig.canvas.draw_idle()
                                         print(f"EC panel horizontal offset set to {ec_h_offset:.3f}\"")
+                                    # Continue in loop to show menu again
+                                    continue
                                 except ValueError:
                                     print("Invalid input. Enter a number.")
+                                    continue
                                 except Exception as e:
                                     print(f"Error: {e}")
+                                    continue
                             else:
                                 print("Invalid choice.")
+                                continue
                     elif choice != 'q':
                         print("Invalid choice")
                 else:
                     # Operando-only mode: toggle colorbar or change label mode
                     cb_h_offset = getattr(cbar.ax, '_cb_h_offset_in', 0.0)
                     print(f"Toggle: 1=colorbar visibility, 2=colorbar label mode, 3=colorbar label text, m=move horizontal position (cb:{cb_h_offset:.3f}\"), q=cancel")
-                    choice = input("v> ").strip().lower()
+                    choice = _safe_input("v> ").strip().lower()
                     if choice == '1':
                         cb_vis = cbar.ax.get_visible()
                         cbar.ax.set_visible(not cb_vis)
@@ -2006,7 +2148,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         # Change colorbar label text
                         current_label = getattr(cbar.ax, '_colorbar_label', 'Intensity')
                         print(f"Current colorbar label: {current_label}")
-                        new_label = input("New colorbar label (blank to keep): ").strip()
+                        new_label = _safe_input("New colorbar label (blank to keep): ").strip()
                         if new_label:
                             cbar.ax._colorbar_label = new_label
                             try:
@@ -2024,12 +2166,12 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print(f"\nHorizontal position (relative to canvas center):")
                             print(f"  Colorbar offset: {cb_h_offset:.3f}\" (positive=right, negative=left)")
                             print("Commands: c=colorbar, q=back")
-                            sub = input("m> ").strip().lower()
+                            sub = _safe_input("m> ").strip().lower()
                             if not sub or sub == 'q':
                                 break
                             if sub == 'c':
                                 try:
-                                    new_offset = input(f"Enter colorbar horizontal offset in inches (current: {cb_h_offset:.3f}\"): ").strip()
+                                    new_offset = _safe_input(f"Enter colorbar horizontal offset in inches (current: {cb_h_offset:.3f}\"): ").strip()
                                     if new_offset:
                                         cb_h_offset = float(new_offset)
                                         setattr(cbar.ax, '_cb_h_offset_in', cb_h_offset)
@@ -2037,12 +2179,17 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                         _apply_group_layout_inches(fig, ax, cbar.ax, ec_ax, ax_w_in, ax_h_in, cb_w_in, cb_gap_in, ec_gap_in, ec_w_in)
                                         fig.canvas.draw_idle()
                                         print(f"Colorbar horizontal offset set to {cb_h_offset:.3f}\"")
+                                    # Continue in loop to show menu again
+                                    continue
                                 except ValueError:
                                     print("Invalid input. Enter a number.")
+                                    continue
                                 except Exception as e:
                                     print(f"Error: {e}")
+                                    continue
                             else:
                                 print("Invalid choice.")
+                                continue
                     elif choice != 'q':
                         print("Invalid choice")
                 fig.canvas.draw_idle()
@@ -2079,7 +2226,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     prompt = "Enter new filename (no ext needed), number to overwrite, or o to overwrite last (q=cancel): "
                 else:
                     prompt = "Enter new filename (no ext needed) or number to overwrite (q=cancel): "
-                choice = input(prompt).strip()
+                choice = _safe_input(prompt).strip()
                 if not choice or choice.lower() == 'q':
                     print_menu(); continue
                 if choice.lower() == 'o':
@@ -2090,7 +2237,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     if not os.path.exists(last_session_path):
                         print(f"Previous save file not found: {last_session_path}")
                         print_menu(); continue
-                    yn = input(f"Overwrite '{os.path.basename(last_session_path)}'? (y/n): ").strip().lower()
+                    yn = _safe_input(f"Overwrite '{os.path.basename(last_session_path)}'? (y/n): ").strip().lower()
                     if yn != 'y':
                         print_menu(); continue
                     dump_operando_session(last_session_path, fig=fig, ax=ax, im=im, cbar=cbar, ec_ax=ec_ax, skip_confirm=True)
@@ -2100,7 +2247,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     idx = int(choice)
                     if 1 <= idx <= len(files):
                         name = files[idx-1]
-                        yn = input(f"Overwrite '{name}'? (y/n): ").strip().lower()
+                        yn = _safe_input(f"Overwrite '{name}'? (y/n): ").strip().lower()
                         if yn != 'y':
                             print_menu(); continue
                         target = os.path.join(folder, name)
@@ -2117,7 +2264,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         name = name + '.pkl'
                     target = name if os.path.isabs(name) else os.path.join(folder, name)
                     if os.path.exists(target):
-                        yn = input(f"'{os.path.basename(target)}' exists. Overwrite? (y/n): ").strip().lower()
+                        yn = _safe_input(f"'{os.path.basename(target)}' exists. Overwrite? (y/n): ").strip().lower()
                         if yn != 'y':
                             print_menu(); continue
                     dump_operando_session(target, fig=fig, ax=ax, im=im, cbar=cbar, ec_ax=ec_ax, skip_confirm=True)
@@ -2142,7 +2289,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             # Always read fresh value from attribute to avoid stale cached value
             ax_h_in = getattr(ax, '_fixed_ax_h_in', ax_h_in)
             print(f"Current height: {ax_h_in:.2f} in")
-            val = input("New height (inches): ").strip()
+            val = _safe_input("New height (inches): ").strip()
             if val:
                 _snapshot("height")
                 try:
@@ -2182,7 +2329,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             print(f"\nFont submenu (current: family='{cur_family}', size={cur_size})")
             print("  f: change family  |  s: change size  |  q: back")
             while True:
-                sub = input("Font> ").strip().lower()
+                sub = _safe_input("Font> ").strip().lower()
                 if not sub:
                     continue
                 if sub == 'q':
@@ -2195,7 +2342,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     for i, font in enumerate(fonts, 1):
                         print(f"  {i}: {font}")
                     print("Or enter custom font name directly.")
-                    choice = input("Font family (number or name): ").strip()
+                    choice = _safe_input("Font family (number or name): ").strip()
                     if not choice:
                         continue
                     _snapshot("font-family")
@@ -2215,7 +2362,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 elif sub == 's':
                     # Show current size and accept direct input
                     cur_size = plt.rcParams.get('font.size', None)
-                    choice = input(f"Font size (current: {cur_size}): ").strip()
+                    choice = _safe_input(f"Font size (current: {cur_size}): ").strip()
                     if not choice:
                         continue
                     _snapshot("font-size")
@@ -2238,7 +2385,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             print(_colorize_inline_commands("  1.5 2.5  - set frame=1.5, ticks=2.5"))
             print(_colorize_inline_commands("  q        - cancel"))
             
-            inp = input("Line widths> ").strip().lower()
+            inp = _safe_input("Line widths> ").strip().lower()
             if not inp or inp == 'q':
                 print_menu()
                 continue
@@ -2301,6 +2448,8 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             # Unified WASD ticks/labels/spines submenu for either pane
             # Import here to avoid scoping issues with nested functions
             from matplotlib.ticker import AutoMinorLocator, NullFormatter, MaxNLocator, NullLocator
+            # Import UI positioning functions locally to ensure they're accessible in nested functions
+            from .ui import position_top_xlabel as _ui_position_top_xlabel, position_bottom_xlabel as _ui_position_bottom_xlabel, position_left_ylabel as _ui_position_left_ylabel, position_right_ylabel as _ui_position_right_ylabel
             
             def _get_tick_state(a):
                 # Unified keys with fallbacks for legacy combined flags
@@ -2386,7 +2535,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     print(_colorize_inline_commands("Choose pane: o=operando, e=ec, q=back"))
                 else:
                     print(_colorize_inline_commands("Choose pane: o=operando, q=back"))
-                pane = input("ot> ").strip().lower()
+                pane = _safe_input("ot> ").strip().lower()
                 if not pane:
                     continue
                 if pane == 'q':
@@ -2605,7 +2754,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 print(_colorize_inline_commands("Type 'i' to invert tick direction, 'l' to change tick length, 'list' for state, 'q' to return."))
                 print(_colorize_inline_commands("  p = adjust title offsets (w=top, s=bottom, a=left, d=right)"))
                 while True:
-                    cmd2 = input(_colorize_prompt("Toggle> ")).strip().lower()
+                    cmd2 = _safe_input(_colorize_prompt("Toggle> ")).strip().lower()
                     if not cmd2:
                         continue
                     if cmd2 == 'q':
@@ -2630,7 +2779,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             # Get current major tick length from axes
                             current_major = ax.xaxis.get_major_ticks()[0].tick1line.get_markersize() if ax.xaxis.get_major_ticks() else 4.0
                             print(f"Current major tick length: {current_major}")
-                            new_length_str = input("Enter new major tick length (e.g., 6.0): ").strip()
+                            new_length_str = _safe_input("Enter new major tick length (e.g., 6.0): ").strip()
                             if not new_length_str:
                                 continue
                             new_major = float(new_length_str)
@@ -2723,7 +2872,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                 current_y_px = _px_value('_top_xlabel_manual_offset_y_pts', target)
                                 current_x_px = _px_value('_top_xlabel_manual_offset_x_pts', target)
                                 print(f"Top title offset: Y={current_y_px:+.2f} px (positive=up), X={current_x_px:+.2f} px (positive=right)")
-                                sub = input(_colorize_prompt("top (w=up, s=down, a=left, d=right, 0=reset, q=back): ")).strip().lower()
+                                sub = _safe_input(_colorize_prompt("top (w=up, s=down, a=left, d=right, 0=reset, q=back): ")).strip().lower()
                                 if not sub:
                                     continue
                                 if sub == 'q':
@@ -2760,7 +2909,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             while True:
                                 current_y_px = _px_value('_bottom_xlabel_manual_offset_y_pts', target)
                                 print(f"Bottom title offset: Y={current_y_px:+.2f} px (positive=down)")
-                                sub = input(_colorize_prompt("bottom (s=down, w=up, 0=reset, q=back): ")).strip().lower()
+                                sub = _safe_input(_colorize_prompt("bottom (s=down, w=up, 0=reset, q=back): ")).strip().lower()
                                 if not sub:
                                     continue
                                 if sub == 'q':
@@ -2790,7 +2939,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             while True:
                                 current_x_px = _px_value('_left_ylabel_manual_offset_x_pts', target)
                                 print(f"Left title offset: X={current_x_px:+.2f} px (positive=left)")
-                                sub = input(_colorize_prompt("left (a=left, d=right, 0=reset, q=back): ")).strip().lower()
+                                sub = _safe_input(_colorize_prompt("left (a=left, d=right, 0=reset, q=back): ")).strip().lower()
                                 if not sub:
                                     continue
                                 if sub == 'q':
@@ -2827,7 +2976,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                 current_x_px = _px_value('_right_ylabel_manual_offset_x_pts', target)
                                 current_y_px = _px_value('_right_ylabel_manual_offset_y_pts', target)
                                 print(f"Right title offset: X={current_x_px:+.2f} px (positive=right), Y={current_y_px:+.2f} px (positive=up)")
-                                sub = input(_colorize_prompt("right (d=right, a=left, w=up, s=down, 0=reset, q=back): ")).strip().lower()
+                                sub = _safe_input(_colorize_prompt("right (d=right, a=left, w=up, s=down, 0=reset, q=back): ")).strip().lower()
                                 if not sub:
                                     continue
                                 if sub == 'q':
@@ -2866,7 +3015,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print("  " + _colorize_menu('d : adjust right title (d=right, a=left, w=up, s=down)'))
                             print("  " + _colorize_menu('r : reset all offsets'))
                             print("  " + _colorize_menu('q : return'))
-                            choice = input(_colorize_prompt("p> ")).strip().lower()
+                            choice = _safe_input(_colorize_prompt("p> ")).strip().lower()
                             if not choice:
                                 continue
                             if choice == 'q':
@@ -2982,7 +3131,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
         elif cmd == 'ox':
             while True:
                 cur = ax.get_xlim(); print(f"Current operando X: {cur[0]:.4g} {cur[1]:.4g}")
-                line = input("New X range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                line = _safe_input("New X range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                 if not line or line.lower() == 'q':
                     break
                 if line.lower() == 'w':
@@ -2990,7 +3139,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ax.get_xlim()
                         print(f"Current operando X: {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new upper X limit (current lower: {cur[0]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new upper X limit (current lower: {cur[0]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3009,7 +3158,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ax.get_xlim()
                         print(f"Current operando X: {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new lower X limit (current upper: {cur[1]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new lower X limit (current upper: {cur[1]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3055,7 +3204,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
         elif cmd == 'oy':
             while True:
                 cur = ax.get_ylim(); print(f"Current operando Y: {cur[0]:.4g} {cur[1]:.4g}")
-                line = input("New Y range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                line = _safe_input("New Y range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                 if not line or line.lower() == 'q':
                     break
                 if line.lower() == 'w':
@@ -3063,7 +3212,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ax.get_ylim()
                         print(f"Current operando Y: {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new upper Y limit (current lower: {cur[0]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new upper Y limit (current lower: {cur[0]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3082,7 +3231,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ax.get_ylim()
                         print(f"Current operando Y: {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new lower Y limit (current upper: {cur[1]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new lower Y limit (current upper: {cur[1]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3185,9 +3334,9 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     auto_available = False
                 
                 if auto_available:
-                    line = input("New intensity range (min max, w=upper only, s=lower only, a=auto-fit to visible, q=back): ").strip()
+                    line = _safe_input("New intensity range (min max, w=upper only, s=lower only, a=auto-fit to visible, q=back): ").strip()
                 else:
-                    line = input("New intensity range (min max, w=upper only, s=lower only, q=back): ").strip()
+                    line = _safe_input("New intensity range (min max, w=upper only, s=lower only, q=back): ").strip()
                 
                 if not line or line.lower() == 'q':
                     break
@@ -3201,7 +3350,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         except Exception:
                             print("Could not retrieve current color scale range")
                             break
-                        val = input(f"Enter new upper intensity limit (current lower: {cur[0]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new upper intensity limit (current lower: {cur[0]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3210,7 +3359,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print("Invalid value, ignored.")
                             continue
                         _snapshot("operando-intensity-range")
-                        im.set_clim(cur[0], new_upper)
+                        _safe_set_clim(im, cur[0], new_upper)
                         try:
                             if cbar is not None:
                                 _update_custom_colorbar(cbar.ax, im)
@@ -3227,7 +3376,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         except Exception:
                             print("Could not retrieve current color scale range")
                             break
-                        val = input(f"Enter new lower intensity limit (current upper: {cur[1]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new lower intensity limit (current upper: {cur[1]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3236,7 +3385,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print("Invalid value, ignored.")
                             continue
                         _snapshot("operando-intensity-range")
-                        im.set_clim(new_lower, cur[1])
+                        _safe_set_clim(im, new_lower, cur[1])
                         try:
                             if cbar is not None:
                                 _update_custom_colorbar(cbar.ax, im)
@@ -3250,7 +3399,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     if line.lower() == 'a':
                         # Apply auto-normalization to visible data
                         if auto_available:
-                            im.set_clim(auto_lo, auto_hi)
+                            _safe_set_clim(im, auto_lo, auto_hi)
                             try:
                                 if cbar is not None:
                                     _update_custom_colorbar(cbar.ax, im)
@@ -3262,7 +3411,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print("Auto-fit unavailable: no finite data in visible area")
                     else:
                         lo, hi = map(float, line.split())
-                        im.set_clim(lo, hi)
+                        _safe_set_clim(im, lo, hi)
                         try:
                             if cbar is not None:
                                 _update_custom_colorbar(cbar.ax, im)
@@ -3278,7 +3427,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             while True:
                 ax_w_in = getattr(ax, '_fixed_ax_w_in', ax_w_in)
                 print(f"Current operando width: {ax_w_in:.2f} in")
-                val = input("New width (inches, q=back): ").strip()
+                val = _safe_input("New width (inches, q=back): ").strip()
                 if not val or val.lower() == 'q':
                     break
                 _snapshot("operando-width")
@@ -3298,7 +3447,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             while True:
                 ec_w_in = getattr(ec_ax, '_fixed_ec_w_in', ec_w_in)
                 print(f"Current EC width: {ec_w_in:.2f} in")
-                val = input("New EC width (inches, q=back): ").strip()
+                val = _safe_input("New EC width (inches, q=back): ").strip()
                 if not val or val.lower() == 'q':
                     break
                 _snapshot("ec-width")
@@ -3311,6 +3460,15 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             print_menu()
         elif cmd == 'oc':
             # Change operando colormap (perceptually uniform suggestions)
+            # Show current colormap if one is applied
+            try:
+                current_cmap = getattr(im, '_operando_cmap_name', None)
+                if current_cmap is None:
+                    current_cmap = getattr(im.get_cmap(), 'name', None)
+                if current_cmap:
+                    print(f"Current operando colormap: {current_cmap}")
+            except Exception:
+                pass
             def _refresh_available():
                 return set(name.lower() for name in plt.colormaps())
             available = _refresh_available()
@@ -3341,7 +3499,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             if optional:
                 print("\nOther available: " + ", ".join(optional))
             print(_colorize_inline_commands("Append _r to reverse (e.g., viridis_r or 1_r). Blank to cancel."))
-            choice = input(f"Palette name or number (1-{len(rec_palettes)}): ").strip()
+            choice = _safe_input(f"Palette name or number (1-{len(rec_palettes)}): ").strip()
             if not choice:
                 print_menu(); continue
             try:
@@ -3383,6 +3541,8 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     if reversed_choice:
                         palette_obj = palette_obj.reversed()
                     im.set_cmap(palette_obj)
+                # Store the colormap name explicitly so it can be retrieved reliably when saving
+                setattr(im, '_operando_cmap_name', choice)
                 try:
                     # Update custom colorbar with new colormap
                     if cbar is not None:
@@ -3419,7 +3579,10 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     cb_w_in, cb_gap_in, ec_gap_in, ec_w_in, ax_w_in, ax_h_in = _ensure_fixed_params(fig, ax, cbar.ax, ec_ax)
                     fam = plt.rcParams.get('font.sans-serif', [''])[0]
                     fsize = plt.rcParams.get('font.size', None)
-                    cmap_name = getattr(im.get_cmap(), 'name', None)
+                    # Get colormap name: first check if we stored it explicitly, otherwise try to get from colormap object
+                    cmap_name = getattr(im, '_operando_cmap_name', None)
+                    if cmap_name is None:
+                        cmap_name = getattr(im.get_cmap(), 'name', None)
                     cb_vis = bool(cbar.ax.get_visible())
                     ec_vis = bool(ec_ax.get_visible()) if ec_ax is not None else None
                     cb_label_text = str(getattr(cbar.ax, '_colorbar_label', cbar.ax.get_ylabel() or 'Intensity'))
@@ -3590,7 +3753,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     last_style_path = getattr(fig, '_last_style_export_path', None)
                     if ec_ax is None:
                         print("\nNote: Style export (.bps/.bpsg) is only available in dual-pane mode (with EC file).")
-                        sub = input("Style submenu: (q=return, r=refresh): ").strip().lower()
+                        sub = _safe_input("Style submenu: (q=return, r=refresh): ").strip().lower()
                         if sub == 'q':
                             break
                         if sub == 'r' or sub == '':
@@ -3600,9 +3763,9 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             continue
                     else:
                         if last_style_path:
-                            sub = input("Style submenu: (e=export, o=overwrite last, q=return, r=refresh): ").strip().lower()
+                            sub = _safe_input("Style submenu: (e=export, o=overwrite last, q=return, r=refresh): ").strip().lower()
                         else:
-                            sub = input("Style submenu: (e=export, q=return, r=refresh): ").strip().lower()
+                            sub = _safe_input("Style submenu: (e=export, q=return, r=refresh): ").strip().lower()
                         if sub == 'q':
                             break
                         if sub == 'r' or sub == '':
@@ -3615,7 +3778,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             if not os.path.exists(last_style_path):
                                 print(f"Previous export file not found: {last_style_path}")
                                 continue
-                            yn = input(f"Overwrite '{os.path.basename(last_style_path)}'? (y/n): ").strip().lower()
+                            yn = _safe_input(f"Overwrite '{os.path.basename(last_style_path)}'? (y/n): ").strip().lower()
                             if yn != 'y':
                                 continue
                             # Determine export type from existing file and rebuild config
@@ -3636,7 +3799,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print("Export options:")
                             print("  ps  = style only (.bps)")
                             print("  psg = style + geometry (.bpsg)")
-                            exp_choice = input("Export choice (ps/psg, q=cancel): ").strip().lower()
+                            exp_choice = _safe_input("Export choice (ps/psg, q=cancel): ").strip().lower()
                             if not exp_choice or exp_choice == 'q':
                                 print("Style export canceled.")
                                 continue
@@ -3830,7 +3993,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             else:
                                 print(f"  {_i}: {fname}")
                     
-                    choice_name = input("Enter new filename or number to overwrite (q=cancel): ").strip()
+                    choice_name = _safe_input("Enter new filename or number to overwrite (q=cancel): ").strip()
                     if not choice_name or choice_name.lower() == 'q':
                         print("Style export canceled.")
                         continue
@@ -3839,7 +4002,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         _idx = int(choice_name)
                         if 1 <= _idx <= len(_style_files):
                             name = _style_files[_idx-1]
-                            yn = input(f"Overwrite '{name}'? (y/n): ").strip().lower()
+                            yn = _safe_input(f"Overwrite '{name}'? (y/n): ").strip().lower()
                             if yn == 'y':
                                 target = file_list[_idx-1][1]  # Full path from list
                         else:
@@ -3856,7 +4019,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         else:
                             target = get_organized_path(name, 'style', base_path=save_base)
                         if os.path.exists(target):
-                            yn = input(f"'{os.path.basename(target)}' exists. Overwrite? (y/n): ").strip().lower()
+                            yn = _safe_input(f"'{os.path.basename(target)}' exists. Overwrite? (y/n): ").strip().lower()
                             if yn != 'y':
                                 target = None
                     if target:
@@ -3986,6 +4149,8 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 if cmap:
                     try:
                         im.set_cmap(cmap)
+                        # Store the colormap name explicitly so it can be retrieved reliably when saving
+                        setattr(im, '_operando_cmap_name', cmap)
                         if cbar is not None:
                             _update_custom_colorbar(cbar.ax, im)
                     except Exception:
@@ -4259,7 +4424,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     try:
                         intensity_range = op.get('intensity_range')
                         if intensity_range and isinstance(intensity_range, (list, tuple)) and len(intensity_range) == 2:
-                            im.set_clim(float(intensity_range[0]), float(intensity_range[1]))
+                            _safe_set_clim(im, float(intensity_range[0]), float(intensity_range[1]))
                             print(f"Applied intensity range: {intensity_range[0]:.4g} to {intensity_range[1]:.4g}")
                     except Exception as e:
                         print(f"Warning: Could not apply intensity range: {e}")
@@ -4553,16 +4718,18 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 print("Tip: Use LaTeX/mathtext for special characters:")
                 print("  Subscript: H$_2$O → H₂O  |  Superscript: m$^2$ → m²")
                 print("  Bullet: $\\bullet$ → •   |  Greek: $\\alpha$, $\\beta$  |  Angstrom: $\\AA$ → Å")
+                print("  Shortcuts: g{super(-1)} → g$^{\\mathrm{-1}}$  |  Li{sub(2)}O → Li$_{\\mathrm{2}}$O")
                 while True:
-                    sub = input("or> ").strip().lower()
+                    sub = _safe_input("or> ").strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
                         break
                     if sub == 'x':
                         cur = ax.get_xlabel() or ''
-                        lab = input(f"New operando X label (blank=cancel, current='{cur}'): ")
+                        lab = _safe_input(f"New operando X label (blank=cancel, current='{cur}'): ")
                         if lab:
+                            lab = convert_label_shortcuts(lab)
                             _snapshot("rename-op-x")
                             try:
                                 ax.set_xlabel(lab)
@@ -4574,8 +4741,9 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                 pass
                     elif sub == 'y':
                         cur = ax.get_ylabel() or ''
-                        lab = input(f"New operando Y label (blank=cancel, current='{cur}'): ")
+                        lab = _safe_input(f"New operando Y label (blank=cancel, current='{cur}'): ")
                         if lab:
+                            lab = convert_label_shortcuts(lab)
                             _snapshot("rename-op-y")
                             try:
                                 ax.set_ylabel(lab)
@@ -4605,16 +4773,18 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 print("Tip: Use LaTeX/mathtext for special characters:")
                 print("  Subscript: H$_2$O → H₂O  |  Superscript: m$^2$ → m²")
                 print("  Greek: $\\alpha$, $\\beta$  |  Angstrom: $\\AA$ → Å")
+                print("  Shortcuts: g{super(-1)} → g$^{\\mathrm{-1}}$  |  Li{sub(2)}O → Li$_{\\mathrm{2}}$O")
                 while True:
-                    sub = input("er> ").strip().lower()
+                    sub = _safe_input("er> ").strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
                         break
                     if sub == 'x':
                         cur = ec_ax.get_xlabel() or ''
-                        lab = input(f"New EC X label (blank=cancel, current='{cur}'): ")
+                        lab = _safe_input(f"New EC X label (blank=cancel, current='{cur}'): ")
                         if lab:
+                            lab = convert_label_shortcuts(lab)
                             _snapshot("rename-ec-x")
                             try:
                                 ec_ax.set_xlabel(lab)
@@ -4625,8 +4795,9 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                                 pass
                     elif sub == 'y':
                         cur = ec_ax.get_ylabel() or ''
-                        lab = input(f"New EC Y label (blank=cancel, current='{cur}'): ")
+                        lab = _safe_input(f"New EC Y label (blank=cancel, current='{cur}'): ")
                         if lab:
+                            lab = convert_label_shortcuts(lab)
                             _snapshot("rename-ec-y")
                             try:
                                 ec_ax.set_ylabel(lab)
@@ -4663,7 +4834,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     print_menu(); continue
                 print("EC line submenu: c=color, l=linewidth, q=back")
                 while True:
-                    sub = input("el> ").strip().lower()
+                    sub = _safe_input("el> ").strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
@@ -4679,7 +4850,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                         else:
                             print("\nNo saved colors. Type 'u' to manage saved colors.")
                         print("  (Enter color name/hex, saved color number, or 'u' to manage)")
-                        val = input(f"Color (current={cur}, blank=cancel): ").strip()
+                        val = _safe_input(f"Color (current={cur}, blank=cancel): ").strip()
                         if not val:
                             continue
                         if val.lower() == 'u':
@@ -4696,7 +4867,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             print(f"Invalid color: {e}")
                     elif sub == 'l':
                         cur = ln.get_linewidth()
-                        val = input(f"Line width (current={cur}, blank=cancel): ").strip()
+                        val = _safe_input(f"Line width (current={cur}, blank=cancel): ").strip()
                         if not val:
                             continue
                         _snapshot("ec-line-width")
@@ -4721,7 +4892,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                 continue
             while True:
                 cur = ec_ax.get_ylim(); print(f"Current EC time range (Y): {cur[0]:.4g} {cur[1]:.4g}")
-                line = input("New time range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                line = _safe_input("New time range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                 if not line or line.lower() == 'q':
                     break
                 if line.lower() == 'w':
@@ -4729,7 +4900,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ec_ax.get_ylim()
                         print(f"Current EC time range (Y): {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new upper time limit (current lower: {cur[0]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new upper time limit (current lower: {cur[0]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -4747,7 +4918,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ec_ax.get_ylim()
                         print(f"Current EC time range (Y): {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new lower time limit (current upper: {cur[1]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new lower time limit (current upper: {cur[1]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -4861,7 +5032,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     print("The .mpt file must contain the '<I>/mA' column to use this feature.")
                     print_menu(); continue
                 while True:
-                    sub = input("ey submenu: n=ions, t=time, q=back: ").strip().lower()
+                    sub = _safe_input("ey submenu: n=ions, t=time, q=back: ").strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
@@ -4878,7 +5049,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                             prompt = "Enter mass(mg), capacity-per-ion(mAh g^-1), start-ions (e.g. 4.5 26.8 0), q=cancel: "
                         else:
                             prompt = f"Enter mass,cap-per-ion,start-ions (blank=reuse {mass_mg} {cap_per_ion} {start_ions}; q=cancel): "
-                        s = input(prompt).strip()
+                        s = _safe_input(prompt).strip()
                         if not s:
                             if need_input:
                                 continue
@@ -5164,7 +5335,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             while True:
                 cur = ec_ax.get_xlim()
                 print(f"Current EC X range: {cur[0]:.4g} {cur[1]:.4g}")
-                line = input("New EC X range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                line = _safe_input("New EC X range (min max), w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                 if not line or line.lower() == 'q':
                     break
                 if line.lower() == 'w':
@@ -5172,7 +5343,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ec_ax.get_xlim()
                         print(f"Current EC X range: {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new upper EC X limit (current lower: {cur[0]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new upper EC X limit (current lower: {cur[0]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -5191,7 +5362,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
                     while True:
                         cur = ec_ax.get_xlim()
                         print(f"Current EC X range: {cur[0]:.4g} {cur[1]:.4g}")
-                        val = input(f"Enter new lower EC X limit (current upper: {cur[1]:.4g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new lower EC X limit (current upper: {cur[1]:.4g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -5253,7 +5424,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax, file_paths=None):
             cur_w, cur_h = _get_fig_size(fig)
             print(f"Current canvas size: {cur_w:.2f} x {cur_h:.2f} in (W x H)")
             print("Canvas: only figure size will change; panel widths/gaps are not altered.")
-            line = input("New canvas size 'W H' (blank=cancel): ").strip()
+            line = _safe_input("New canvas size 'W H' (blank=cancel): ").strip()
             if line:
                 _snapshot("canvas-size")
                 try: