batplot 1.7.25__py3-none-any.whl → 1.7.26__py3-none-any.whl

batplot/cpc_interactive.py

@@ -36,11 +36,45 @@ from __future__ import annotations
 from typing import Dict, Optional
 import json
 import os
+import sys
+import contextlib
+from io import StringIO
 
 import matplotlib.pyplot as plt
 from matplotlib.ticker import AutoMinorLocator, NullFormatter, NullLocator
 import random as _random
 
+
+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
+
 from .ui import (
     resize_plot_frame, resize_canvas,
     update_tick_visibility as _ui_update_tick_visibility,
@@ -57,10 +91,20 @@ from .utils import (
     get_organized_path,
 )
 import time
-from .color_utils import resolve_color_token
+from .color_utils import resolve_color_token, color_block, palette_preview, manage_user_colors, get_user_color_list, ensure_colormap
 
 
 def _legend_no_frame(ax, *args, **kwargs):
+    # Compact legend defaults and labelcolor matching marker/line color
+    kwargs.setdefault('frameon', False)
+    kwargs.setdefault('handlelength', 1.0)
+    kwargs.setdefault('handletextpad', 0.35)
+    kwargs.setdefault('labelspacing', 0.25)
+    kwargs.setdefault('borderaxespad', 0.5)
+    kwargs.setdefault('borderpad', 0.3)
+    kwargs.setdefault('columnspacing', 0.6)
+    # Let matplotlib color legend text from line/marker colors
+    kwargs.setdefault('labelcolor', 'linecolor')
     leg = ax.legend(*args, **kwargs)
     if leg is not None:
         try:
@@ -548,16 +592,19 @@ def _style_snapshot(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_data=Non
         'series': {
             'charge': {
                 'color': _color_of(sc_charge),
+                'marker': getattr(sc_charge, 'get_marker', lambda: 'o')(),
                 'markersize': float(getattr(sc_charge, 'get_sizes', lambda: [32])()[0]) if hasattr(sc_charge, 'get_sizes') else 32.0,
                 'alpha': float(sc_charge.get_alpha()) if sc_charge.get_alpha() is not None else 1.0,
             },
             'discharge': {
                 'color': _color_of(sc_discharge),
+                'marker': getattr(sc_discharge, 'get_marker', lambda: 's')(),
                 'markersize': float(getattr(sc_discharge, 'get_sizes', lambda: [32])()[0]) if hasattr(sc_discharge, 'get_sizes') else 32.0,
                 'alpha': float(sc_discharge.get_alpha()) if sc_discharge.get_alpha() is not None else 1.0,
             },
             'efficiency': {
                 'color': (sc_eff.get_facecolors()[0].tolist() if hasattr(sc_eff, 'get_facecolors') and len(sc_eff.get_facecolors()) else '#2ca02c'),
+                'marker': getattr(sc_eff, 'get_marker', lambda: '^')(),
                 'markersize': float(getattr(sc_eff, 'get_sizes', lambda: [40])()[0]) if hasattr(sc_eff, 'get_sizes') else 40.0,
                 'alpha': float(sc_eff.get_alpha()) if sc_eff.get_alpha() is not None else 1.0,
                 'visible': bool(getattr(sc_eff, 'get_visible', lambda: True)()),
@@ -569,12 +616,18 @@ def _style_snapshot(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_data=Non
     if file_data and isinstance(file_data, list) and len(file_data) > 0:
         multi_files = []
         for f in file_data:
+            sc_chg = f.get('sc_charge')
+            sc_dchg = f.get('sc_discharge')
+            sc_eff = f.get('sc_eff')
             file_info = {
                 'filename': f.get('filename', 'unknown'),
                 'visible': f.get('visible', True),
-                'charge_color': _color_of(f.get('sc_charge')),
-                'discharge_color': _color_of(f.get('sc_discharge')),
-                'efficiency_color': _color_of(f.get('sc_eff')),
+                'charge_color': _color_of(sc_chg),
+                'charge_marker': getattr(sc_chg, 'get_marker', lambda: 'o')() if sc_chg else 'o',
+                'discharge_color': _color_of(sc_dchg),
+                'discharge_marker': getattr(sc_dchg, 'get_marker', lambda: 's')() if sc_dchg else 's',
+                'efficiency_color': _color_of(sc_eff),
+                'efficiency_marker': getattr(sc_eff, 'get_marker', lambda: '^')() if sc_eff else '^',
             }
             # Save legend labels
             try:
@@ -726,6 +779,13 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
         # Apply marker sizes and alpha globally to all files in multi-file mode
         if is_multi_file:
             for f in file_data:
+                # Marker types (global)
+                if ch.get('marker') is not None and hasattr(f['sc_charge'], 'set_marker'):
+                    f['sc_charge'].set_marker(ch['marker'])
+                if dh.get('marker') is not None and hasattr(f['sc_discharge'], 'set_marker'):
+                    f['sc_discharge'].set_marker(dh['marker'])
+                if ef.get('marker') is not None and hasattr(f['sc_eff'], 'set_marker'):
+                    f['sc_eff'].set_marker(ef['marker'])
                 # Marker sizes (global)
                 if ch.get('markersize') is not None and hasattr(f['sc_charge'], 'set_sizes'):
                     f['sc_charge'].set_sizes([float(ch['markersize'])])
@@ -759,6 +819,8 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
             if ch:
                 if ch.get('color') is not None:
                     sc_charge.set_color(ch['color'])
+                if ch.get('marker') is not None and hasattr(sc_charge, 'set_marker'):
+                    sc_charge.set_marker(ch['marker'])
                 if ch.get('markersize') is not None and hasattr(sc_charge, 'set_sizes'):
                     sc_charge.set_sizes([float(ch['markersize'])])
                 if ch.get('alpha') is not None:
@@ -766,6 +828,8 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
             if dh:
                 if dh.get('color') is not None:
                     sc_discharge.set_color(dh['color'])
+                if dh.get('marker') is not None and hasattr(sc_discharge, 'set_marker'):
+                    sc_discharge.set_marker(dh['marker'])
                 if dh.get('markersize') is not None and hasattr(sc_discharge, 'set_sizes'):
                     sc_discharge.set_sizes([float(dh['markersize'])])
                 if dh.get('alpha') is not None:
@@ -776,6 +840,8 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
                         sc_eff.set_color(ef['color'])
                     except Exception:
                         pass
+                if ef.get('marker') is not None and hasattr(sc_eff, 'set_marker'):
+                    sc_eff.set_marker(ef['marker'])
                 if ef.get('markersize') is not None and hasattr(sc_eff, 'set_sizes'):
                     sc_eff.set_sizes([float(ef['markersize'])])
                 if ef.get('alpha') is not None:
@@ -803,6 +869,27 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
                 leg.set_visible(leg_visible)
             if leg_visible:
                 _apply_legend_position()
+                # Re-apply legend label colors to match handles after position/visibility changes
+                try:
+                    leg = ax.get_legend()
+                    if leg is not None:
+                        handles = list(getattr(leg, "legendHandles", []))
+                        for h, txt in zip(handles, leg.get_texts()):
+                            col = _color_of(h)
+                            if col is None and hasattr(h, 'get_edgecolor'):
+                                col = h.get_edgecolor()
+                            if isinstance(col, (list, tuple)) and len(col) and not isinstance(col, str):
+                                col = col[0]
+                            try:
+                                import numpy as _np
+                                if hasattr(col, "__len__") and not isinstance(col, str):
+                                    col = tuple(_np.array(col).ravel().tolist())
+                            except Exception:
+                                pass
+                            if col is not None:
+                                txt.set_color(col)
+                except Exception:
+                    pass
     except Exception:
         pass
     # Apply tick visibility/widths and spines
@@ -1050,6 +1137,42 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
                 for i, f_info in enumerate(multi_files):
                     if i < len(file_data):
                         f = file_data[i]
+                        # Restore colors FIRST (before labels)
+                        if 'charge_color' in f_info and f.get('sc_charge'):
+                            try:
+                                col = f_info['charge_color']
+                                f['sc_charge'].set_color(col)
+                                f['color'] = col
+                                # Force update of facecolors for scatter plots
+                                if hasattr(f['sc_charge'], 'set_facecolors'):
+                                    from matplotlib.colors import to_rgba
+                                    rgba = to_rgba(col)
+                                    f['sc_charge'].set_facecolors(rgba)
+                            except Exception:
+                                pass
+                        if 'discharge_color' in f_info and f.get('sc_discharge'):
+                            try:
+                                col = f_info['discharge_color']
+                                f['sc_discharge'].set_color(col)
+                                # Force update of facecolors for scatter plots
+                                if hasattr(f['sc_discharge'], 'set_facecolors'):
+                                    from matplotlib.colors import to_rgba
+                                    rgba = to_rgba(col)
+                                    f['sc_discharge'].set_facecolors(rgba)
+                            except Exception:
+                                pass
+                        if 'efficiency_color' in f_info and f.get('sc_eff'):
+                            try:
+                                col = f_info['efficiency_color']
+                                f['sc_eff'].set_color(col)
+                                f['eff_color'] = col
+                                # Force update of facecolors for scatter plots
+                                if hasattr(f['sc_eff'], 'set_facecolors'):
+                                    from matplotlib.colors import to_rgba
+                                    rgba = to_rgba(col)
+                                    f['sc_eff'].set_facecolors(rgba)
+                            except Exception:
+                                pass
                         # Restore legend labels
                         if 'charge_label' in f_info and f.get('sc_charge'):
                             try:
@@ -1206,8 +1329,22 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
     if file_data is None:
         # Backward compatibility: create file_data structure from single file
         # This allows the function to work with old code that passes individual artists
+        # Try to get filename from label if available
+        filename = 'Data'
+        try:
+            if hasattr(sc_charge, 'get_label') and sc_charge.get_label():
+                label = sc_charge.get_label()
+                # Extract filename from label like "filename (Chg)" or use label as-is
+                if ' (Chg)' in label:
+                    filename = label.replace(' (Chg)', '')
+                elif ' (Dch)' in label:
+                    filename = label.replace(' (Dch)', '')
+                elif label and label != 'Charge capacity':
+                    filename = label
+        except Exception:
+            pass
         file_data = [{
-            'filename': 'Data',  # Default filename
+            'filename': filename,
             'sc_charge': sc_charge,      # Charge capacity scatter artist
             'sc_discharge': sc_discharge,  # Discharge capacity scatter artist
             'sc_eff': sc_eff,            # Efficiency scatter artist
@@ -1327,6 +1464,27 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 if k in tick_state:
                     tick_state[k] = bool(v)
             _update_ticks()
+            # Re-apply legend text colors after state restore (undo)
+            try:
+                leg = ax.get_legend()
+                if leg is not None:
+                    handles = list(getattr(leg, "legendHandles", []))
+                    for h, txt in zip(handles, leg.get_texts()):
+                        col = _color_of(h)
+                        if col is None and hasattr(h, 'get_edgecolor'):
+                            col = h.get_edgecolor()
+                        if isinstance(col, (list, tuple)) and len(col) and not isinstance(col, str):
+                            col = col[0]
+                        try:
+                            import numpy as _np
+                            if hasattr(col, "__len__") and not isinstance(col, str):
+                                col = tuple(_np.array(col).ravel().tolist())
+                        except Exception:
+                            pass
+                        if col is not None:
+                            txt.set_color(col)
+            except Exception:
+                pass
             try:
                 fig.canvas.draw()
             except Exception:
@@ -1443,7 +1601,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             # Update current file's scatter artists for commands that need them
             sc_charge, sc_discharge, sc_eff = _get_current_file_artists(file_data, current_file_idx)
             
-            key = input("Press a key: ").strip().lower()
+            key = _safe_input("Press a key: ").strip().lower()
         except (KeyboardInterrupt, EOFError):
             print("\n\nExiting interactive menu...")
             break
@@ -1455,7 +1613,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             try:
                 if is_multi_file:
                     _print_file_list(file_data, current_file_idx)
-                    choice = input(f"Toggle visibility for file (1-{len(file_data)}), 'a' for all, or q=cancel: ").strip()
+                    choice = _safe_input(f"Toggle visibility for file (1-{len(file_data)}), 'a' for all, or q=cancel: ").strip()
                     if choice.lower() == 'q':
                         _print_menu()
                         _print_file_list(file_data, current_file_idx)
@@ -1505,7 +1663,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
         
         if key == 'q':
             try:
-                confirm = input(_colorize_prompt("Quit CPC interactive? Remember to save! Quit now? (y/n): ")).strip().lower()
+                confirm = _safe_input(_colorize_prompt("Quit CPC interactive? Remember to save! Quit now? (y/n): ")).strip().lower()
             except Exception:
                 confirm = 'y'
             if confirm == 'y':
@@ -1516,283 +1674,282 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             restore_state()
             _print_menu(); continue
         elif key == 'c':
-            # Colors submenu: ly (left Y series) and ry (right Y efficiency)
+            # Colors submenu: ly (left Y series) and ry (right Y efficiency), with user colors and palettes
             try:
+                # Use same palettes as EC interactive
+                palette_opts = ['tab10', 'Set2', 'Dark2', 'viridis', 'plasma']
+                def _palette_color(name, idx=0, total=1, default_val=0.4):
+                    import matplotlib.cm as cm
+                    import matplotlib.colors as mcolors
+                    import numpy as _np
+                    # Ensure colormap is registered before use
+                    if not ensure_colormap(name):
+                        # Fallback to viridis if colormap can't be registered
+                        name = 'viridis'
+                        ensure_colormap(name)
+                    try:
+                        cmap = cm.get_cmap(name)
+                    except Exception:
+                        # Fallback if get_cmap fails
+                        ensure_colormap('viridis')
+                        cmap = cm.get_cmap('viridis')
+                    
+                    # Special handling for tab10 to match hardcoded colors exactly
+                    if name.lower() == 'tab10':
+                        default_tab10_colors = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd',
+                                               '#8c564b', '#e377c2', '#7f7f7f', '#bcbd22', '#17becf']
+                        return default_tab10_colors[idx % len(default_tab10_colors)]
+                    
+                    # For discrete colormaps (Set2, Dark2), access colors directly
+                    if hasattr(cmap, 'colors') and cmap.colors is not None:
+                        # Discrete colormap: access colors directly by index
+                        colors = cmap.colors
+                        rgb = colors[idx % len(colors)]
+                        if isinstance(rgb, tuple) and len(rgb) >= 3:
+                            return mcolors.rgb2hex(rgb[:3])
+                    
+                    # For continuous colormaps (viridis, plasma), sample evenly
+                    if total == 1:
+                        vals = [0.55]
+                    elif total == 2:
+                        vals = [0.15, 0.85]
+                    else:
+                        vals = _np.linspace(0.08, 0.88, total)
+                    rgb = cmap(vals[idx % len(vals)])
+                    return mcolors.rgb2hex(rgb[:3])
+                def _resolve_color(spec, idx=0, total=1, default_cmap='tab10'):
+                    spec = spec.strip()
+                    if not spec:
+                        return None
+                    if spec.lower() == 'r':
+                        return _palette_color(default_cmap, idx, total, 0.4)
+                    # user colors: u# or plain number referencing saved list
+                    uc = None
+                    if spec.lower().startswith('u') and len(spec) > 1 and spec[1:].isdigit():
+                        uc = resolve_color_token(spec, fig)
+                    elif spec.isdigit():
+                        # number as palette index if within palette list
+                        n = int(spec)
+                        if 1 <= n <= len(palette_opts):
+                            palette_name = palette_opts[n-1]
+                            return _palette_color(palette_name, idx, total, 0.4)
+                    if uc:
+                        return uc
+                    # Check if spec is a palette name (case-insensitive)
+                    spec_lower = spec.lower()
+                    base = spec.rstrip('_r').rstrip('_R')
+                    base_lower = base.lower()
+                    # Check against palette_opts (case-insensitive)
+                    for pal in palette_opts:
+                        if spec_lower == pal.lower() or base_lower == pal.lower() or spec_lower == (pal + '_r').lower():
+                            return _palette_color(pal if not spec.endswith('_r') and not spec.endswith('_R') else spec, idx, total, 0.4)
+                    # Fall back to resolve_color_token for hex colors, named colors, etc.
+                    return resolve_color_token(spec, fig)
+
                 while True:
-                    print("\nColors: ly=capacity curves, ry=efficiency triangles, q=back")
-                    sub = input("Colors> ").strip().lower()
+                    print("\nColors: ly=capacity curves, ry=efficiency triangles, u=user colors, q=back")
+                    sub = _safe_input("Colors> ").strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
                         break
+                    if sub == 'u':
+                        manage_user_colors(fig); continue
                     if sub == 'ly':
-                        if is_multi_file:
-                            # Show file list for selection
-                            print("\nSelect curve to color:")
-                            for i, f in enumerate(file_data, 1):
-                                vis_mark = "●" if f.get('visible', True) else "○"
-                                print(f"  {i}. {vis_mark} {f['filename']}")
-                            choice = input("Enter curve number (1-{}) or 'a' for all, q=cancel: ".format(len(file_data))).strip()
-                            if not choice or choice.lower() == 'q':
-                                continue
-                            
-                            push_state("colors-ly")
-                            if choice.lower() == 'a':
-                                # Apply to all files
-                                print("\nCharge color palettes (discharge will be auto-generated):")
-                                print("  1. Reds: #d62728, #c62828, #b71c1c, #8b0000, #a30000")
-                                print("  2. Oranges: #ff7f0e, #ff6f00, #ff5722, #f4511e, #e64a19")
-                                print("  3. Pinks/Magentas: #e377c2, #d81b60, #c2185b, #ad1457, #880e4f")
-                                print("  4. Purples: #9c27b0, #8e24aa, #7b1fa2, #6a1b9a, #4a148c")
-                                print("  5. Deep oranges/reds: #d84315, #bf360c, #c2185b, #d32f2f, #c62828")
-                                spec = input("Enter color (name/hex), palette number (1-5), or 'r' for random (q=cancel): ").strip()
-                                if not spec or spec.lower() == 'q':
+                        push_state("colors-ly")
+                        print("\nCurrent capacity curves:")
+                        for i, f in enumerate(file_data, 1):
+                            cur = _color_of(f['sc_charge'])
+                            vis_mark = "●" if f.get('visible', True) else "○"
+                            print(f"  {i}. {vis_mark} {f['filename']}  {color_block(cur)} {cur}")
+                        uc = get_user_color_list(fig)
+                        if uc:
+                            print("\nSaved colors (refer as number or u#):")
+                            for i, c in enumerate(uc, 1):
+                                print(f"  {i}: {color_block(c)} {c}")
+                        print("\nPalettes:")
+                        for idx, name in enumerate(palette_opts, 1):
+                            bar = palette_preview(name)
+                            print(f"  {idx}. {name}")
+                            if bar:
+                                print(f"      {bar}")
+                        color_input = _safe_input("Enter file+color pairs (e.g., 1:2 2:3 or 1 2 2 3) or palette/number for all, q=cancel: ").strip()
+                        if not color_input or color_input.lower() == 'q':
+                            continue
+                        tokens = color_input.split()
+                        if len(tokens) == 1:
+                            # Single token: apply palette to all files
+                            spec = tokens[0]
+                            for i, f in enumerate(file_data):
+                                charge_col = _resolve_color(spec, i, len(file_data), default_cmap='tab10')
+                                if not charge_col:
                                     continue
-                                for i, f in enumerate(file_data):
-                                    if spec.lower() == 'r':
-                                        # Use Viridis colormap
-                                        import matplotlib.cm as cm
-                                        import matplotlib.colors as mcolors
-                                        viridis = cm.get_cmap('viridis', 10)
-                                        charge_col = mcolors.rgb2hex(viridis(_random.random())[:3])
-
-                                    elif spec in ['1', '2', '3', '4', '5']:
-                                        # Use selected palette
-                                        charge_palettes = [
-                                            ['#d62728', '#c62828', '#b71c1c', '#8b0000', '#a30000'],
-                                            ['#ff7f0e', '#ff6f00', '#ff5722', '#f4511e', '#e64a19'],
-                                            ['#e377c2', '#d81b60', '#c2185b', '#ad1457', '#880e4f'],
-                                            ['#9c27b0', '#8e24aa', '#7b1fa2', '#6a1b9a', '#4a148c'],
-                                            ['#d84315', '#bf360c', '#c2185b', '#d32f2f', '#c62828']
-                                        ]
-                                        palette = charge_palettes[int(spec) - 1]
-                                        charge_col = palette[i % len(palette)]
+                                discharge_col = _generate_similar_color(charge_col)
+                                try:
+                                    f['sc_charge'].set_color(charge_col)
+                                    f['sc_discharge'].set_color(discharge_col)
+                                    f['color'] = charge_col
+                                    # Force update of facecolors for scatter plots
+                                    if hasattr(f['sc_charge'], 'set_facecolors'):
+                                        from matplotlib.colors import to_rgba
+                                        rgba = to_rgba(charge_col)
+                                        f['sc_charge'].set_facecolors(rgba)
+                                    if hasattr(f['sc_discharge'], 'set_facecolors'):
+                                        from matplotlib.colors import to_rgba
+                                        rgba = to_rgba(discharge_col)
+                                        f['sc_discharge'].set_facecolors(rgba)
+                                except Exception as e:
+                                    print(f"Error setting color: {e}")
+                                    pass
+                        else:
+                            # Multiple tokens: parse file:color pairs
+                            def _apply_manual_entries(tokens):
+                                idx_color_pairs = []
+                                i = 0
+                                while i < len(tokens):
+                                    tok = tokens[i]
+                                    if ':' in tok:
+                                        idx_str, color = tok.split(':', 1)
                                     else:
-                                        charge_col = spec
+                                        if i + 1 >= len(tokens):
+                                            print(f"Skip incomplete entry: {tok}")
+                                            break
+                                        idx_str = tok
+                                        color = tokens[i + 1]
+                                        i += 1
+                                    idx_color_pairs.append((idx_str, color))
+                                    i += 1
+                                for idx_str, color in idx_color_pairs:
+                                    try:
+                                        file_idx = int(idx_str) - 1
+                                    except ValueError:
+                                        print(f"Bad index: {idx_str}")
+                                        continue
+                                    if not (0 <= file_idx < len(file_data)):
+                                        print(f"Index out of range: {idx_str}")
+                                        continue
+                                    resolved = resolve_color_token(color, fig)
+                                    charge_col = resolved if resolved else color
+                                    if not charge_col:
+                                        continue
                                     discharge_col = _generate_similar_color(charge_col)
                                     try:
-                                        f['sc_charge'].set_color(charge_col)
-                                        f['sc_discharge'].set_color(discharge_col)
-                                        f['color'] = charge_col
+                                        file_data[file_idx]['sc_charge'].set_color(charge_col)
+                                        file_data[file_idx]['sc_discharge'].set_color(discharge_col)
+                                        file_data[file_idx]['color'] = charge_col
+                                        # Force update of facecolors for scatter plots
+                                        if hasattr(file_data[file_idx]['sc_charge'], 'set_facecolors'):
+                                            from matplotlib.colors import to_rgba
+                                            rgba = to_rgba(charge_col)
+                                            file_data[file_idx]['sc_charge'].set_facecolors(rgba)
+                                        if hasattr(file_data[file_idx]['sc_discharge'], 'set_facecolors'):
+                                            from matplotlib.colors import to_rgba
+                                            rgba = to_rgba(discharge_col)
+                                            file_data[file_idx]['sc_discharge'].set_facecolors(rgba)
                                     except Exception:
                                         pass
-                            else:
-                                # Apply to selected file
-                                try:
-                                    idx = int(choice) - 1
-                                    if 0 <= idx < len(file_data):
-                                        print("\nCharge color palettes (discharge will be auto-generated):")
-                                        print("  1. Reds: #d62728, #c62828, #b71c1c, #8b0000, #a30000")
-                                        print("  2. Oranges: #ff7f0e, #ff6f00, #ff5722, #f4511e, #e64a19")
-                                        print("  3. Pinks/Magentas: #e377c2, #d81b60, #c2185b, #ad1457, #880e4f")
-                                        print("  4. Purples: #9c27b0, #8e24aa, #7b1fa2, #6a1b9a, #4a148c")
-                                        print("  5. Deep oranges/reds: #d84315, #bf360c, #c2185b, #d32f2f, #c62828")
-                                        spec = input("Enter color (name/hex), palette number (1-5), or 'r' for random (q=cancel): ").strip()
-                                        if not spec or spec.lower() == 'q':
-                                            continue
-                                        if spec.lower() == 'r':
-                                            # Use Viridis colormap
-                                            import matplotlib.cm as cm
-                                            import matplotlib.colors as mcolors
-                                            viridis = cm.get_cmap('viridis', 10)
-                                            charge_col = mcolors.rgb2hex(viridis(_random.random())[:3])
-                                        elif spec in ['1', '2', '3', '4', '5']:
-                                            # Use selected palette
-                                            charge_palettes = [
-                                                ['#d62728', '#c62828', '#b71c1c', '#8b0000', '#a30000'],
-                                                ['#ff7f0e', '#ff6f00', '#ff5722', '#f4511e', '#e64a19'],
-                                                ['#e377c2', '#d81b60', '#c2185b', '#ad1457', '#880e4f'],
-                                                ['#9c27b0', '#8e24aa', '#7b1fa2', '#6a1b9a', '#4a148c'],
-                                                ['#d84315', '#bf360c', '#c2185b', '#d32f2f', '#c62828']
-                                            ]
-                                            palette = charge_palettes[int(spec) - 1]
-                                            charge_col = palette[0]  # Use first color from palette for single file
-                                        else:
-                                            charge_col = spec
-                                        discharge_col = _generate_similar_color(charge_col)
-                                        try:
-                                            file_data[idx]['sc_charge'].set_color(charge_col)
-                                            file_data[idx]['sc_discharge'].set_color(discharge_col)
-                                            file_data[idx]['color'] = charge_col
-                                        except Exception:
-                                            pass
-                                    else:
-                                        print("Invalid file number.")
-                                except ValueError:
-                                    print("Invalid input.")
-                        else:
-                            # Single file mode
-                            push_state("colors-ly")
-                            print("\nCharge color palettes (discharge will be auto-generated):")
-                            print("  1. Reds: #d62728, #c62828, #b71c1c, #8b0000, #a30000")
-                            print("  2. Oranges: #ff7f0e, #ff6f00, #ff5722, #f4511e, #e64a19")
-                            print("  3. Pinks/Magentas: #e377c2, #d81b60, #c2185b, #ad1457, #880e4f")
-                            print("  4. Purples: #9c27b0, #8e24aa, #7b1fa2, #6a1b9a, #4a148c")
-                            print("  5. Deep oranges/reds: #d84315, #bf360c, #c2185b, #d32f2f, #c62828")
-                            spec = input("Enter color (name/hex), palette number (1-5), or 'r' for random (q=cancel): ").strip()
-                            if not spec or spec.lower() == 'q':
-                                continue
-                            if spec.strip().lower() == 'r':
-                                # Use Viridis colormap
-                                import matplotlib.cm as cm
-                                import matplotlib.colors as mcolors
-                                viridis = cm.get_cmap('viridis', 10)
-                                charge_col = mcolors.rgb2hex(viridis(_random.random())[:3])
-                            elif spec in ['1', '2', '3', '4', '5']:
-                                # Use selected palette
-                                charge_palettes = [
-                                    ['#d62728', '#c62828', '#b71c1c', '#8b0000', '#a30000'],
-                                    ['#ff7f0e', '#ff6f00', '#ff5722', '#f4511e', '#e64a19'],
-                                    ['#e377c2', '#d81b60', '#c2185b', '#ad1457', '#880e4f'],
-                                    ['#9c27b0', '#8e24aa', '#7b1fa2', '#6a1b9a', '#4a148c'],
-                                    ['#d84315', '#bf360c', '#c2185b', '#d32f2f', '#c62828']
-                                ]
-                                palette = charge_palettes[int(spec) - 1]
-                                charge_col = palette[0]  # Use first color from palette
-                            else:
-                                charge_col = spec
-                            discharge_col = _generate_similar_color(charge_col)
+                            _apply_manual_entries(tokens)
+                        if not is_multi_file and getattr(fig, '_cpc_spine_auto', False):
                             try:
-                                sc_charge.set_color(charge_col)
-                                sc_discharge.set_color(discharge_col)
-                                # Apply auto colors if enabled
-                                if not is_multi_file and getattr(fig, '_cpc_spine_auto', False):
-                                    _set_spine_color('left', charge_col)
+                                cur_col = _color_of(sc_charge)
+                                if cur_col:
+                                    _set_spine_color('left', cur_col)
                             except Exception:
                                 pass
                         try:
-                            _rebuild_legend(ax, ax2, file_data)
-                            fig.canvas.draw_idle()
+                            _rebuild_legend(ax, ax2, file_data); fig.canvas.draw_idle()
                         except Exception:
                             pass
                     elif sub == 'ry':
                         push_state("colors-ry")
-                        if is_multi_file:
-                            # Show file list for efficiency triangle selection
-                            print("\nSelect curve's efficiency to color:")
-                            for i, f in enumerate(file_data, 1):
-                                vis_mark = "●" if f.get('visible', True) else "○"
-                                print(f"  {i}. {vis_mark} {f['filename']}")
-                            choice = input("Enter curve number (1-{}) or 'a' for all, q=cancel: ".format(len(file_data))).strip()
-                            if not choice or choice.lower() == 'q':
-                                continue
-                            
-                            if choice.lower() == 'a':
-                                print("\nEfficiency color palettes:")
-                                print("  1. Blues: #1f77b4, #1976d2, #1565c0, #0d47a1, #01579b")
-                                print("  2. Cyans/Teals: #17becf, #00acc1, #0097a7, #00838f, #006064")
-                                print("  3. Purples/Indigos: #9467bd, #5e35b1, #512da8, #4527a0, #311b92")
-                                print("  4. Deep blues: #2196f3, #1e88e5, #1976d2, #1565c0, #0d47a1")
-                                print("  5. Dark cyans/purples: #0097a7, #00838f, #006064, #5e35b1, #4527a0")
-                                val = input("Enter color (hex/name), palette number (1-5), or 'r' for random (q=cancel): ").strip()
-                                if not val or val.lower() == 'q':
+                        print("\nCurrent efficiency curves:")
+                        for i, f in enumerate(file_data, 1):
+                            cur = _color_of(f['sc_eff'])
+                            vis_mark = "●" if f.get('visible', True) else "○"
+                            print(f"  {i}. {vis_mark} {f['filename']}  {color_block(cur)} {cur}")
+                        uc = get_user_color_list(fig)
+                        if uc:
+                            print("\nSaved colors (refer as number or u#):")
+                            for i, c in enumerate(uc, 1):
+                                print(f"  {i}: {color_block(c)} {c}")
+                        print("\nPalettes:")
+                        for idx, name in enumerate(palette_opts, 1):
+                            bar = palette_preview(name)
+                            print(f"  {idx}. {name}")
+                            if bar:
+                                print(f"      {bar}")
+                        color_input = _safe_input("Enter file+color pairs (e.g., 1:2 2:3 or 1 2 2 3) or palette/number for all, q=cancel: ").strip()
+                        if not color_input or color_input.lower() == 'q':
+                            continue
+                        tokens = color_input.split()
+                        if len(tokens) == 1:
+                            # Single token: apply palette to all files
+                            spec = tokens[0]
+                            for i, f in enumerate(file_data):
+                                col = _resolve_color(spec, i, len(file_data), default_cmap='viridis')
+                                if not col:
                                     continue
-                                for i, f in enumerate(file_data):
-                                    if val.lower() == 'r':
-                                        # Use Plasma colormap
-                                        import matplotlib.cm as cm
-                                        import matplotlib.colors as mcolors
-                                        plasma = cm.get_cmap('plasma', 10)
-                                        col = mcolors.rgb2hex(plasma(_random.random())[:3])
-                                    elif val in ['1', '2', '3', '4', '5']:
-                                        # Use selected palette
-                                        efficiency_palettes = [
-                                            ['#1f77b4', '#1976d2', '#1565c0', '#0d47a1', '#01579b'],
-                                            ['#17becf', '#00acc1', '#0097a7', '#00838f', '#006064'],
-                                            ['#9467bd', '#5e35b1', '#512da8', '#4527a0', '#311b92'],
-                                            ['#2196f3', '#1e88e5', '#1976d2', '#1565c0', '#0d47a1'],
-                                            ['#0097a7', '#00838f', '#006064', '#5e35b1', '#4527a0']
-                                        ]
-                                        palette = efficiency_palettes[int(val) - 1]
-                                        col = palette[i % len(palette)]
+                                try:
+                                    f['sc_eff'].set_color(col)
+                                    f['eff_color'] = col
+                                    # Force update of facecolors for scatter plots
+                                    if hasattr(f['sc_eff'], 'set_facecolors'):
+                                        from matplotlib.colors import to_rgba
+                                        rgba = to_rgba(col)
+                                        f['sc_eff'].set_facecolors(rgba)
+                                except Exception:
+                                    pass
+                        else:
+                            # Multiple tokens: parse file:color pairs
+                            def _apply_manual_entries_eff(tokens):
+                                idx_color_pairs = []
+                                i = 0
+                                while i < len(tokens):
+                                    tok = tokens[i]
+                                    if ':' in tok:
+                                        idx_str, color = tok.split(':', 1)
                                     else:
-                                        col = val
+                                        if i + 1 >= len(tokens):
+                                            print(f"Skip incomplete entry: {tok}")
+                                            break
+                                        idx_str = tok
+                                        color = tokens[i + 1]
+                                        i += 1
+                                    idx_color_pairs.append((idx_str, color))
+                                    i += 1
+                                for idx_str, color in idx_color_pairs:
+                                    try:
+                                        file_idx = int(idx_str) - 1
+                                    except ValueError:
+                                        print(f"Bad index: {idx_str}")
+                                        continue
+                                    if not (0 <= file_idx < len(file_data)):
+                                        print(f"Index out of range: {idx_str}")
+                                        continue
+                                    resolved = resolve_color_token(color, fig)
+                                    col = resolved if resolved else color
+                                    if not col:
+                                        continue
                                     try:
-                                        f['sc_eff'].set_color(col)
-                                        f['eff_color'] = col  # Store efficiency color
+                                        file_data[file_idx]['sc_eff'].set_color(col)
+                                        file_data[file_idx]['eff_color'] = col
+                                        # Force update of facecolors for scatter plots
+                                        if hasattr(file_data[file_idx]['sc_eff'], 'set_facecolors'):
+                                            from matplotlib.colors import to_rgba
+                                            rgba = to_rgba(col)
+                                            file_data[file_idx]['sc_eff'].set_facecolors(rgba)
                                     except Exception:
                                         pass
-                            else:
-                                try:
-                                    idx = int(choice) - 1
-                                    if 0 <= idx < len(file_data):
-                                        print("\nEfficiency color palettes:")
-                                        print("  1. Blues: #1f77b4, #1976d2, #1565c0, #0d47a1, #01579b")
-                                        print("  2. Cyans/Teals: #17becf, #00acc1, #0097a7, #00838f, #006064")
-                                        print("  3. Purples/Indigos: #9467bd, #5e35b1, #512da8, #4527a0, #311b92")
-                                        print("  4. Deep blues: #2196f3, #1e88e5, #1976d2, #1565c0, #0d47a1")
-                                        print("  5. Dark cyans/purples: #0097a7, #00838f, #006064, #5e35b1, #4527a0")
-                                        val = input("Enter color (hex/name), palette number (1-5), or 'r' for random (q=cancel): ").strip()
-                                        if not val or val.lower() == 'q':
-                                            continue
-                                        if val.lower() == 'r':
-                                            # Use Plasma colormap
-                                            import matplotlib.cm as cm
-                                            import matplotlib.colors as mcolors
-                                            plasma = cm.get_cmap('plasma', 10)
-                                            col = mcolors.rgb2hex(plasma(_random.random())[:3])
-                                        elif val in ['1', '2', '3', '4', '5']:
-                                            # Use selected palette
-                                            efficiency_palettes = [
-                                                ['#1f77b4', '#1976d2', '#1565c0', '#0d47a1', '#01579b'],
-                                                ['#17becf', '#00acc1', '#0097a7', '#00838f', '#006064'],
-                                                ['#9467bd', '#5e35b1', '#512da8', '#4527a0', '#311b92'],
-                                                ['#2196f3', '#1e88e5', '#1976d2', '#1565c0', '#0d47a1'],
-                                                ['#0097a7', '#00838f', '#006064', '#5e35b1', '#4527a0']
-                                            ]
-                                            palette = efficiency_palettes[int(val) - 1]
-                                            col = palette[0]  # Use first color from palette for single file
-                                        else:
-                                            col = val
-                                        try:
-                                            file_data[idx]['sc_eff'].set_color(col)
-                                            file_data[idx]['eff_color'] = col  # Store efficiency color
-                                        except Exception:
-                                            pass
-                                    else:
-                                        print("Invalid file number.")
-                                except ValueError:
-                                    print("Invalid input.")
-                        else:
-                            # Single file mode
-                            print("\nEfficiency color palettes:")
-                            print("  1. Blues: #1f77b4, #1976d2, #1565c0, #0d47a1, #01579b")
-                            print("  2. Cyans/Teals: #17becf, #00acc1, #0097a7, #00838f, #006064")
-                            print("  3. Purples/Indigos: #9467bd, #5e35b1, #512da8, #4527a0, #311b92")
-                            print("  4. Deep blues: #2196f3, #1e88e5, #1976d2, #1565c0, #0d47a1")
-                            print("  5. Dark cyans/purples: #0097a7, #00838f, #006064, #5e35b1, #4527a0")
-                            val = input("Enter color (hex/name), palette number (1-5), or 'r' for random (q=cancel): ").strip()
-                            if not val or val.lower() == 'q':
-                                continue
-                            if val.lower() == 'r':
-                                # Use Plasma colormap
-                                import matplotlib.cm as cm
-                                import matplotlib.colors as mcolors
-                                plasma = cm.get_cmap('plasma', 10)
-                                col = mcolors.rgb2hex(plasma(_random.random())[:3])
-                            elif val in ['1', '2', '3', '4', '5']:
-                                # Use selected palette
-                                efficiency_palettes = [
-                                    ['#1f77b4', '#1976d2', '#1565c0', '#0d47a1', '#01579b'],
-                                    ['#17becf', '#00acc1', '#0097a7', '#00838f', '#006064'],
-                                    ['#9467bd', '#5e35b1', '#512da8', '#4527a0', '#311b92'],
-                                    ['#2196f3', '#1e88e5', '#1976d2', '#1565c0', '#0d47a1'],
-                                    ['#0097a7', '#00838f', '#006064', '#5e35b1', '#4527a0']
-                                ]
-                                palette = efficiency_palettes[int(val) - 1]
-                                col = palette[0]  # Use first color from palette
-                            else:
-                                col = val
+                            _apply_manual_entries_eff(tokens)
+                        if not is_multi_file and getattr(fig, '_cpc_spine_auto', False):
                             try:
-                                sc_eff.set_color(col)
-                                # Apply auto colors if enabled
-                                if not is_multi_file and getattr(fig, '_cpc_spine_auto', False):
-                                    _set_spine_color('right', col)
+                                cur_col = _color_of(sc_eff)
+                                if cur_col:
+                                    _set_spine_color('right', cur_col)
                             except Exception:
                                 pass
                         try:
-                            _rebuild_legend(ax, ax2, file_data)
-                            fig.canvas.draw_idle()
+                            _rebuild_legend(ax, ax2, file_data); fig.canvas.draw_idle()
                         except Exception:
                             pass
                     else:
@@ -1817,7 +1974,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         auto_status = "ON" if auto_enabled else "OFF"
                         print(_colorize_inline_commands(f"  a : auto (apply capacity curve color to left y-axis, efficiency to right y-axis) [{auto_status}]"))
                     print("q: back to main menu")
-                    line = input("Enter mappings (e.g., w:red a:#4561F7) or q: ").strip()
+                    line = _safe_input("Enter mappings (e.g., w:red a:#4561F7) or q: ").strip()
                     if not line or line.lower() == 'q':
                         break
                     # Handle auto toggle when only one file is loaded
@@ -1889,9 +2046,9 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 
                 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
                 
@@ -1903,7 +2060,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     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
@@ -1912,7 +2069,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     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
@@ -1929,10 +2086,17 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     else:
                         target = get_organized_path(fname, 'figure', base_path=base_path)
                     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
                 if target:
+                    # Save current legend position before export (savefig can change layout)
+                    saved_legend_pos = None
+                    try:
+                        saved_legend_pos = getattr(fig, '_cpc_legend_xy_in', None)
+                    except Exception:
+                        pass
+                    
                     # Remove numbering from legend labels before export
                     original_labels = {}
                     if is_multi_file:
@@ -1997,19 +2161,46 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                     ax2.patch.set_alpha(1.0); ax2.patch.set_facecolor(_ax2_fc)
                             except Exception:
                                 pass
+                        print(f"Exported figure to {target}")
+                        fig._last_figure_export_path = target
+                        
+                        # Restore original labels and legend position
+                        if is_multi_file and original_labels:
+                            try:
+                                for artist, label in original_labels.items():
+                                    artist.set_label(label)
+                                _rebuild_legend(ax, ax2, file_data)
+                            except Exception:
+                                pass
+                        # Restore legend position after savefig (which may have changed layout)
+                        if saved_legend_pos is not None:
+                            try:
+                                fig._cpc_legend_xy_in = saved_legend_pos
+                                _rebuild_legend(ax, ax2, file_data)
+                                fig.canvas.draw_idle()
+                            except Exception:
+                                pass
                     else:
                         fig.savefig(target, bbox_inches='tight')
-                    print(f"Exported figure to {target}")
-                    fig._last_figure_export_path = target
-                    
-                    # Restore original labels
-                    if is_multi_file and original_labels:
-                        try:
-                            for artist, label in original_labels.items():
-                                artist.set_label(label)
-                            _rebuild_legend(ax, ax2, file_data)
-                        except Exception:
-                            pass
+                        print(f"Exported figure to {target}")
+                        fig._last_figure_export_path = target
+                        
+                        # Restore original labels and legend position
+                        if is_multi_file and original_labels:
+                            try:
+                                for artist, label in original_labels.items():
+                                    artist.set_label(label)
+                                _rebuild_legend(ax, ax2, file_data)
+                            except Exception:
+                                pass
+                        # Restore legend position after savefig (which may have changed layout)
+                        if saved_legend_pos is not None:
+                            try:
+                                fig._cpc_legend_xy_in = saved_legend_pos
+                                _rebuild_legend(ax, ax2, file_data)
+                                fig.canvas.draw_idle()
+                            except Exception:
+                                pass
             except Exception as e:
                 print(f"Export failed: {e}")
             _print_menu(); continue
@@ -2091,7 +2282,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     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':
@@ -2102,7 +2293,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     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_cpc_session(last_session_path, fig=fig, ax=ax, ax2=ax2, sc_charge=sc_charge, sc_discharge=sc_discharge, sc_eff=sc_eff, file_data=file_data, skip_confirm=True)
@@ -2112,7 +2303,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     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)
@@ -2129,7 +2320,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         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_cpc_session(target, fig=fig, ax=ax, ax2=ax2, sc_charge=sc_charge, sc_discharge=sc_discharge, sc_eff=sc_eff, file_data=file_data, skip_confirm=True)
@@ -2259,9 +2450,9 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     
                     last_style_path = getattr(fig, '_last_style_export_path', None)
                     if last_style_path:
-                        sub = input(_colorize_prompt("Style submenu: (e=export, o=overwrite last, q=return, r=refresh): ")).strip().lower()
+                        sub = _safe_input(_colorize_prompt("Style submenu: (e=export, o=overwrite last, q=return, r=refresh): ")).strip().lower()
                     else:
-                        sub = input(_colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
+                        sub = _safe_input(_colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
                     if sub == 'q':
                         break
                     if sub == 'r' or sub == '':
@@ -2274,7 +2465,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         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
                         # Rebuild config based on current state
@@ -2300,7 +2491,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         print("Export options:")
                         print("  ps  = style only (.bps)")
                         print("  psg = style + geometry (.bpsg)")
-                        exp_choice = input(_colorize_prompt("Export choice (ps/psg, q=cancel): ")).strip().lower()
+                        exp_choice = _safe_input(_colorize_prompt("Export choice (ps/psg, q=cancel): ")).strip().lower()
                         if not exp_choice or exp_choice == 'q':
                             print("Style export canceled.")
                             continue
@@ -2353,9 +2544,9 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                 else:
                                     print(f"  {i}: {fname}")
                         if last_style_path:
-                            choice = input("Enter new filename, number to overwrite, or o to overwrite last (q=cancel): ").strip()
+                            choice = _safe_input("Enter new filename, number to overwrite, or o to overwrite last (q=cancel): ").strip()
                         else:
-                            choice = input("Enter new filename or number to overwrite (q=cancel): ").strip()
+                            choice = _safe_input("Enter new filename or number to overwrite (q=cancel): ").strip()
                         if not choice or choice.lower() == 'q':
                             print("Style export canceled.")
                             continue
@@ -2367,7 +2558,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             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
                             # Rebuild config based on current state
@@ -2393,7 +2584,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             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':
                                     target = file_list[idx-1][1]  # Full path from list
                             else:
@@ -2410,7 +2601,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             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:
@@ -2587,7 +2778,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 xy_in = _sanitize_legend_offset(xy_in) or (0.0, 0.0)
                 print(f"Legend is {'ON' if vis else 'off'}; position (inches from center): x={xy_in[0]:.2f}, y={xy_in[1]:.2f}")
                 while True:
-                    sub = input("Legend: t=toggle, p=set position, q=back: ").strip().lower()
+                    sub = _safe_input("Legend: t=toggle, p=set position, q=back: ").strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
@@ -2617,7 +2808,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             xy_in = getattr(fig, '_cpc_legend_xy_in', (0.0, 0.0))
                             xy_in = _sanitize_legend_offset(xy_in) or (0.0, 0.0)
                             print(f"Current position: x={xy_in[0]:.2f}, y={xy_in[1]:.2f}")
-                            pos_cmd = input("Position: (x y) or x=x only, y=y only, q=back: ").strip().lower()
+                            pos_cmd = _safe_input("Position: (x y) or x=x only, y=y only, q=back: ").strip().lower()
                             if not pos_cmd or pos_cmd == 'q':
                                 break
                             if pos_cmd == 'x':
@@ -2626,7 +2817,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                     xy_in = getattr(fig, '_cpc_legend_xy_in', (0.0, 0.0))
                                     xy_in = _sanitize_legend_offset(xy_in) or (0.0, 0.0)
                                     print(f"Current position: x={xy_in[0]:.2f}, y={xy_in[1]:.2f}")
-                                    val = input(f"Enter new x position (current y: {xy_in[1]:.2f}, q=back): ").strip()
+                                    val = _safe_input(f"Enter new x position (current y: {xy_in[1]:.2f}, q=back): ").strip()
                                     if not val or val.lower() == 'q':
                                         break
                                     try:
@@ -2649,7 +2840,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                     xy_in = getattr(fig, '_cpc_legend_xy_in', (0.0, 0.0))
                                     xy_in = _sanitize_legend_offset(xy_in) or (0.0, 0.0)
                                     print(f"Current position: x={xy_in[0]:.2f}, y={xy_in[1]:.2f}")
-                                    val = input(f"Enter new y position (current x: {xy_in[0]:.2f}, q=back): ").strip()
+                                    val = _safe_input(f"Enter new y position (current x: {xy_in[0]:.2f}, q=back): ").strip()
                                     if not val or val.lower() == 'q':
                                         break
                                     try:
@@ -2690,11 +2881,11 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 pass
             _print_menu(); continue
         elif key == 'f':
-            sub = input("Font: f=family, s=size, q=back: ").strip().lower()
+            sub = _safe_input("Font: f=family, s=size, q=back: ").strip().lower()
             if sub == 'q' or not sub:
                 _print_menu(); continue
             if sub == 'f':
-                fam = input("Enter font family (e.g., Arial, DejaVu Sans): ").strip()
+                fam = _safe_input("Enter font family (e.g., Arial, DejaVu Sans): ").strip()
                 if fam:
                     try:
                         push_state("font-family")
@@ -2750,7 +2941,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     except Exception:
                         pass
             elif sub == 's':
-                val = input("Enter font size (number): ").strip()
+                val = _safe_input("Enter font size (number): ").strip()
                 try:
                     size = float(val)
                     push_state("font-size")
@@ -2844,13 +3035,13 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     print(f"  {_colorize_menu('f  : change frame (axes spines) and tick widths')}")
                     print(f"  {_colorize_menu('g  : toggle grid lines')}")
                     print(f"  {_colorize_menu('q  : return')}")
-                    sub = input(_colorize_prompt("Choose (f/g/q): ")).strip().lower()
+                    sub = _safe_input(_colorize_prompt("Choose (f/g/q): ")).strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
                         break
                     if sub == 'f':
-                        fw_in = input("Enter frame/tick width (e.g., 1.5) or 'm M' (major minor) or q: ").strip()
+                        fw_in = _safe_input("Enter frame/tick width (e.g., 1.5) or 'm M' (major minor) or q: ").strip()
                         if not fw_in or fw_in.lower() == 'q':
                             print("Canceled.")
                             continue
@@ -2922,7 +3113,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 except Exception:
                     e_ms = 40
                 print(f"  charge ms={c_ms}, discharge ms={d_ms}, efficiency ms={e_ms}")
-                spec = input("Set marker size: 'c <ms>', 'd <ms>', 'e <ms>' (q=cancel): ").strip().lower()
+                spec = _safe_input("Set marker size: 'c <ms>', 'd <ms>', 'e <ms>' (q=cancel): ").strip().lower()
                 if not spec or spec == 'q':
                     _print_menu(); continue
                 parts = spec.split()
@@ -3179,7 +3370,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 print(_colorize_inline_commands("Type 'i' to invert tick direction, 'l' to change tick length, 'list' to show current state, 'q' to go back."))
                 print(_colorize_inline_commands("  p = adjust title offsets (w=top, s=bottom, a=left, d=right)"))
                 while True:
-                    cmd = input(_colorize_prompt("t> ")).strip().lower()
+                    cmd = _safe_input(_colorize_prompt("t> ")).strip().lower()
                     if not cmd:
                         continue
                     if cmd == 'q':
@@ -3204,7 +3395,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             # 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)
@@ -3279,7 +3470,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                 current_y_px = _px_value('_top_xlabel_manual_offset_y_pts')
                                 current_x_px = _px_value('_top_xlabel_manual_offset_x_pts')
                                 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':
@@ -3316,7 +3507,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             while True:
                                 current_y_px = _px_value('_bottom_xlabel_manual_offset_y_pts')
                                 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':
@@ -3346,7 +3537,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             while True:
                                 current_x_px = _px_value('_left_ylabel_manual_offset_x_pts')
                                 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':
@@ -3377,7 +3568,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                 current_x_px = _px_value('_right_ylabel_manual_offset_x_pts')
                                 current_y_px = _px_value('_right_ylabel_manual_offset_y_pts')
                                 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':
@@ -3415,7 +3606,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             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':
@@ -3516,7 +3707,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
         elif key == 'g':
             while True:
                 print("Geometry: p=plot frame, c=canvas, q=back")
-                sub = input("Geom> ").strip().lower()
+                sub = _safe_input("Geom> ").strip().lower()
                 if not sub:
                     continue
                 if sub == 'q':
@@ -3542,7 +3733,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             print("  Bullet: $\\bullet$ → •   |  Greek: $\\alpha$, $\\beta$  |  Angstrom: $\\AA$ → Å")
             while True:
                 print("Rename: x=x-axis, ly=left y-axis, ry=right y-axis, l=legend labels, q=back")
-                sub = input("Rename> ").strip().lower()
+                sub = _safe_input("Rename> ").strip().lower()
                 if not sub:
                     continue
                 if sub == 'q':
@@ -3588,7 +3779,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                         break
                         
                         print(f"Current file name in legend: '{base_name}'")
-                        new_name = input("Enter new file name (q=cancel): ").strip()
+                        new_name = _safe_input("Enter new file name (q=cancel): ").strip()
                         if new_name and new_name.lower() != 'q':
                             try:
                                 push_state("rename-legend")
@@ -3656,7 +3847,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         # Multi-file mode: show file list and let user select
                         print("\nAvailable files:")
                         _print_file_list(file_data, current_file_idx)
-                        file_choice = input("Enter file number to rename (q=cancel): ").strip()
+                        file_choice = _safe_input("Enter file number to rename (q=cancel): ").strip()
                         if file_choice and file_choice.lower() != 'q':
                             try:
                                 file_idx = int(file_choice) - 1
@@ -3695,7 +3886,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                                     break
                                     
                                     print(f"Current file name in legend: '{base_name}'")
-                                    new_name = input("Enter new file name (q=cancel): ").strip()
+                                    new_name = _safe_input("Enter new file name (q=cancel): ").strip()
                                     if new_name and new_name.lower() != 'q':
                                         try:
                                             push_state("rename-legend")
@@ -3766,7 +3957,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 elif sub == 'x':
                     current = ax.get_xlabel()
                     print(f"Current x-axis title: '{current}'")
-                    new_title = input("Enter new x-axis title (q=cancel): ")
+                    new_title = _safe_input("Enter new x-axis title (q=cancel): ")
                     if new_title and new_title.lower() != 'q':
                         try:
                             push_state("rename-x")
@@ -3785,7 +3976,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 elif sub == 'ly':
                     current = ax.get_ylabel()
                     print(f"Current left y-axis title: '{current}'")
-                    new_title = input("Enter new left y-axis title (q=cancel): ")
+                    new_title = _safe_input("Enter new left y-axis title (q=cancel): ")
                     if new_title and new_title.lower() != 'q':
                         try:
                             push_state("rename-ly")
@@ -3799,7 +3990,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                 elif sub == 'ry':
                     current = ax2.get_ylabel()
                     print(f"Current right y-axis title: '{current}'")
-                    new_title = input("Enter new right y-axis title (q=cancel): ")
+                    new_title = _safe_input("Enter new right y-axis title (q=cancel): ")
                     if new_title and new_title.lower() != 'q':
                         try:
                             push_state("rename-ry")
@@ -3819,7 +4010,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             while True:
                 current_xlim = ax.get_xlim()
                 print(f"Current X range: {current_xlim[0]:.6g} to {current_xlim[1]:.6g}")
-                rng = input("Enter x-range: min max, w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                rng = _safe_input("Enter x-range: min max, w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                 if not rng or rng.lower() == 'q':
                     break
                 if rng.lower() == 'w':
@@ -3827,7 +4018,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     while True:
                         current_xlim = ax.get_xlim()
                         print(f"Current X range: {current_xlim[0]:.6g} to {current_xlim[1]:.6g}")
-                        val = input(f"Enter new upper X limit (current lower: {current_xlim[0]:.6g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new upper X limit (current lower: {current_xlim[0]:.6g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3854,7 +4045,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     while True:
                         current_xlim = ax.get_xlim()
                         print(f"Current X range: {current_xlim[0]:.6g} to {current_xlim[1]:.6g}")
-                        val = input(f"Enter new lower X limit (current upper: {current_xlim[1]:.6g}, q=back): ").strip()
+                        val = _safe_input(f"Enter new lower X limit (current upper: {current_xlim[1]:.6g}, q=back): ").strip()
                         if not val or val.lower() == 'q':
                             break
                         try:
@@ -3917,7 +4108,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
         elif key == 'y':
             while True:
                 print("Y-ranges: ly=left axis, ry=right axis, q=back")
-                ycmd = input("Y> ").strip().lower()
+                ycmd = _safe_input("Y> ").strip().lower()
                 if not ycmd:
                     continue
                 if ycmd == 'q':
@@ -3926,7 +4117,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     while True:
                         current_ylim = ax.get_ylim()
                         print(f"Current left Y range: {current_ylim[0]:.6g} to {current_ylim[1]:.6g}")
-                        rng = input("Enter left y-range: min max, w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                        rng = _safe_input("Enter left y-range: min max, w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                         if not rng or rng.lower() == 'q':
                             break
                         if rng.lower() == 'w':
@@ -3934,7 +4125,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             while True:
                                 current_ylim = ax.get_ylim()
                                 print(f"Current left Y range: {current_ylim[0]:.6g} to {current_ylim[1]:.6g}")
-                                val = input(f"Enter new upper left Y limit (current lower: {current_ylim[0]:.6g}, q=back): ").strip()
+                                val = _safe_input(f"Enter new upper left Y limit (current lower: {current_ylim[0]:.6g}, q=back): ").strip()
                                 if not val or val.lower() == 'q':
                                     break
                                 try:
@@ -3961,7 +4152,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             while True:
                                 current_ylim = ax.get_ylim()
                                 print(f"Current left Y range: {current_ylim[0]:.6g} to {current_ylim[1]:.6g}")
-                                val = input(f"Enter new lower left Y limit (current upper: {current_ylim[1]:.6g}, q=back): ").strip()
+                                val = _safe_input(f"Enter new lower left Y limit (current upper: {current_ylim[1]:.6g}, q=back): ").strip()
                                 if not val or val.lower() == 'q':
                                     break
                                 try:
@@ -4029,7 +4220,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             break
                         current_ylim = ax2.get_ylim()
                         print(f"Current right Y range: {current_ylim[0]:.6g} to {current_ylim[1]:.6g}")
-                        rng = input("Enter right y-range: min max, w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
+                        rng = _safe_input("Enter right y-range: min max, w=upper only, s=lower only, a=auto (restore original), q=back: ").strip()
                         if not rng or rng.lower() == 'q':
                             break
                         if rng.lower() == 'w':
@@ -4037,7 +4228,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             while True:
                                 current_ylim = ax2.get_ylim()
                                 print(f"Current right Y range: {current_ylim[0]:.6g} to {current_ylim[1]:.6g}")
-                                val = input(f"Enter new upper right Y limit (current lower: {current_ylim[0]:.6g}, q=back): ").strip()
+                                val = _safe_input(f"Enter new upper right Y limit (current lower: {current_ylim[0]:.6g}, q=back): ").strip()
                                 if not val or val.lower() == 'q':
                                     break
                                 try:
@@ -4064,7 +4255,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             while True:
                                 current_ylim = ax2.get_ylim()
                                 print(f"Current right Y range: {current_ylim[0]:.6g} to {current_ylim[1]:.6g}")
-                                val = input(f"Enter new lower right Y limit (current upper: {current_ylim[1]:.6g}, q=back): ").strip()
+                                val = _safe_input(f"Enter new lower right Y limit (current upper: {current_ylim[1]:.6g}, q=back): ").strip()
                                 if not val or val.lower() == 'q':
                                     break
                                 try: