plot-saver 0.1.0__tar.gz

plot_saver-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,47 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Check out repository
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install build tooling
+        run: python -m pip install --upgrade build
+
+      - name: Build distributions
+        run: python -m build
+
+      - name: Upload distributions
+        uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/*
+
+  publish:
+    needs: build
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write
+
+    steps:
+      - name: Download distributions
+        uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

plot_saver-0.1.0/PKG-INFO

@@ -0,0 +1,115 @@
+Metadata-Version: 2.4
+Name: plot-saver
+Version: 0.1.0
+Summary: Reusable anywidget plot saver configurable with a config.toml.
+Requires-Python: >=3.11
+Requires-Dist: anywidget>=0.9.21
+Requires-Dist: marimo>=0.21.1
+Requires-Dist: traitlets>=5.14.3
+Description-Content-Type: text/markdown
+
+# plot-saver
+
+`plot-saver` is a small reusable package for adding save buttons to Matplotlib figures in marimo notebooks.
+
+It saves plots under a project results directory and can be configured with a project-level `config.toml`.
+
+## Installation
+
+```bash
+pip install plot-saver
+```
+
+or with `uv`:
+
+```bash
+uv add plot-saver
+```
+
+## Config
+
+`plot-saver` looks for a `config.toml` by searching upward from the current working directory.
+
+All configuration keys are optional. If you do not provide a `config.toml`, or if you only provide some keys, `plot-saver` falls back to its built-in defaults.
+
+The default output format is `png`.
+
+You can override the defaults with a section like this:
+
+```toml
+[plot-saver]
+format = "png"
+default_label = "Save"
+save_all_label = "Save all model plots"
+empty_title = "No plots available"
+empty_detail = "Render the notebook plots first."
+saved_title = "Saved"
+saved_many_title = "Saved {count} plots"
+saved_error_title = "Saved with errors"
+failed_title = "Could not save plots"
+toast_detail_color = "#6b7280"
+
+[plot-saver.theme]
+radius = "8px"
+padding_y = "0.45rem"
+padding_x = "0.9rem"
+font_size = "0.9rem"
+light_border = "rgba(107, 114, 128, 0.35)"
+light_background = "rgba(249, 250, 251, 0.95)"
+light_text = "#111827"
+light_hover_background = "rgba(243, 244, 246, 1)"
+light_hover_border = "rgba(107, 114, 128, 0.55)"
+light_disabled_background = "rgba(229, 231, 235, 0.9)"
+light_disabled_border = "rgba(156, 163, 175, 0.35)"
+dark_border = "rgba(156, 163, 175, 0.35)"
+dark_background = "rgba(31, 41, 55, 0.95)"
+dark_text = "#f3f4f6"
+dark_hover_background = "rgba(55, 65, 81, 1)"
+dark_hover_border = "rgba(209, 213, 219, 0.45)"
+dark_disabled_background = "rgba(55, 65, 81, 0.8)"
+dark_disabled_border = "rgba(107, 114, 128, 0.35)"
+```
+
+## Usage
+
+```python
+from pathlib import Path
+
+import marimo as mo
+import matplotlib.pyplot as plt
+from plot_saver import make_plot_saver
+
+fig, ax = plt.subplots()
+ax.plot([0, 1, 2], [1, 3, 2])
+
+save_plot = make_plot_saver(
+    mo,
+    results_dir=Path("results"),
+    config_path=None,
+    task_name="2AFC",
+    model_id="example-model",
+)
+
+button = save_plot(fig, "Example figure")
+save_all = save_plot.save_all_widget()
+```
+
+Saved plots go to:
+
+```text
+results/plots/<task_name>/<model_id>/
+```
+
+`save_plot(fig, "Example figure")` registers that figure and returns an individual save button for it.
+
+`save_plot.save_all_widget()` returns a single button that saves every figure previously registered with that `PlotSaver` instance. This is useful in notebooks where you render several figures for the same task and model and want one action to export all of them together.
+
+## API
+
+Main entry points:
+
+- `make_plot_saver(...)`
+- `save_button(...)`
+- `save_figure(...)`
+- `get_plot_save_format(...)`
+- `find_project_config_path(...)`

plot_saver-0.1.0/README.md

@@ -0,0 +1,105 @@
+# plot-saver
+
+`plot-saver` is a small reusable package for adding save buttons to Matplotlib figures in marimo notebooks.
+
+It saves plots under a project results directory and can be configured with a project-level `config.toml`.
+
+## Installation
+
+```bash
+pip install plot-saver
+```
+
+or with `uv`:
+
+```bash
+uv add plot-saver
+```
+
+## Config
+
+`plot-saver` looks for a `config.toml` by searching upward from the current working directory.
+
+All configuration keys are optional. If you do not provide a `config.toml`, or if you only provide some keys, `plot-saver` falls back to its built-in defaults.
+
+The default output format is `png`.
+
+You can override the defaults with a section like this:
+
+```toml
+[plot-saver]
+format = "png"
+default_label = "Save"
+save_all_label = "Save all model plots"
+empty_title = "No plots available"
+empty_detail = "Render the notebook plots first."
+saved_title = "Saved"
+saved_many_title = "Saved {count} plots"
+saved_error_title = "Saved with errors"
+failed_title = "Could not save plots"
+toast_detail_color = "#6b7280"
+
+[plot-saver.theme]
+radius = "8px"
+padding_y = "0.45rem"
+padding_x = "0.9rem"
+font_size = "0.9rem"
+light_border = "rgba(107, 114, 128, 0.35)"
+light_background = "rgba(249, 250, 251, 0.95)"
+light_text = "#111827"
+light_hover_background = "rgba(243, 244, 246, 1)"
+light_hover_border = "rgba(107, 114, 128, 0.55)"
+light_disabled_background = "rgba(229, 231, 235, 0.9)"
+light_disabled_border = "rgba(156, 163, 175, 0.35)"
+dark_border = "rgba(156, 163, 175, 0.35)"
+dark_background = "rgba(31, 41, 55, 0.95)"
+dark_text = "#f3f4f6"
+dark_hover_background = "rgba(55, 65, 81, 1)"
+dark_hover_border = "rgba(209, 213, 219, 0.45)"
+dark_disabled_background = "rgba(55, 65, 81, 0.8)"
+dark_disabled_border = "rgba(107, 114, 128, 0.35)"
+```
+
+## Usage
+
+```python
+from pathlib import Path
+
+import marimo as mo
+import matplotlib.pyplot as plt
+from plot_saver import make_plot_saver
+
+fig, ax = plt.subplots()
+ax.plot([0, 1, 2], [1, 3, 2])
+
+save_plot = make_plot_saver(
+    mo,
+    results_dir=Path("results"),
+    config_path=None,
+    task_name="2AFC",
+    model_id="example-model",
+)
+
+button = save_plot(fig, "Example figure")
+save_all = save_plot.save_all_widget()
+```
+
+Saved plots go to:
+
+```text
+results/plots/<task_name>/<model_id>/
+```
+
+`save_plot(fig, "Example figure")` registers that figure and returns an individual save button for it.
+
+`save_plot.save_all_widget()` returns a single button that saves every figure previously registered with that `PlotSaver` instance. This is useful in notebooks where you render several figures for the same task and model and want one action to export all of them together.
+
+## API
+
+Main entry points:
+
+- `make_plot_saver(...)`
+- `save_button(...)`
+- `save_figure(...)`
+- `get_plot_save_format(...)`
+- `find_project_config_path(...)`

plot_saver-0.1.0/pyproject.toml

@@ -0,0 +1,18 @@
+[project]
+name = "plot-saver"
+version = "0.1.0"
+description = "Reusable anywidget plot saver configurable with a config.toml."
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "anywidget>=0.9.21",
+    "marimo>=0.21.1",
+    "traitlets>=5.14.3",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/plot_saver"]

plot_saver-0.1.0/src/plot_saver/__init__.py

@@ -0,0 +1,25 @@
+from .save_widget import (
+    PlotSaver,
+    SaveFigureAnyWidget,
+    build_plot_path,
+    find_project_config_path,
+    get_plot_save_format,
+    load_app_config,
+    make_plot_saver,
+    sanitize_stem,
+    save_button,
+    save_figure,
+)
+
+__all__ = [
+    "PlotSaver",
+    "SaveFigureAnyWidget",
+    "build_plot_path",
+    "find_project_config_path",
+    "get_plot_save_format",
+    "load_app_config",
+    "make_plot_saver",
+    "sanitize_stem",
+    "save_button",
+    "save_figure",
+]

plot_saver-0.1.0/src/plot_saver/anywidget_compat.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+import base64
+from functools import lru_cache
+from weakref import WeakKeyDictionary
+
+from marimo._plugins.ui._core.ui_element import UIElement
+from marimo._plugins.ui._impl.from_anywidget import (
+    anywidget as MarimoAnyWidget,
+    get_anywidget_model_id,
+)
+from marimo._utils.code import hash_code
+from marimo._utils.data_uri import build_data_url
+
+
+@lru_cache(maxsize=32)
+def _build_js_data_url(js: str) -> str:
+    encoded = base64.b64encode(js.encode("utf-8"))
+    return build_data_url("text/javascript", encoded)
+
+
+def _resolve_js_url(js: str) -> str:
+    if not js:
+        return ""
+    if js.startswith(("data:", "http://", "https://")):
+        return js
+    return _build_js_data_url(js)
+
+
+class _StableAnyWidget(MarimoAnyWidget):
+    def __init__(self, widget):
+        self.widget = widget
+        self._initialized = False
+
+        js = str(getattr(widget, "_esm", "") or "")
+        js_hash = hash_code(js)
+
+        _ = widget.comm
+        model_id = get_anywidget_model_id(widget)
+
+        UIElement.__init__(
+            self,
+            component_name="marimo-anywidget",
+            initial_value={"model_id": model_id},
+            label=None,
+            args={
+                "js-url": _resolve_js_url(js),
+                "js-hash": js_hash,
+                "model-id": model_id,
+            },
+            on_change=None,
+        )
+
+
+_WIDGET_CACHE: WeakKeyDictionary[object, _StableAnyWidget] = WeakKeyDictionary()
+
+
+def wrap_anywidget(widget):
+    cached = _WIDGET_CACHE.get(widget)
+    if cached is not None:
+        return cached
+
+    wrapped = _StableAnyWidget(widget)
+    _WIDGET_CACHE[widget] = wrapped
+    return wrapped

plot_saver-0.1.0/src/plot_saver/config.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from copy import deepcopy
+from importlib.resources import files
+from pathlib import Path
+import tomllib
+
+
+PROJECT_CONFIG_NAME = "config.toml"
+DEFAULT_CONFIG_RESOURCE = "resources/default_config.toml"
+
+
+def _read_toml(path: Path) -> dict:
+    with path.open("rb") as f:
+        return tomllib.load(f)
+
+
+def _merge_dicts(base: dict, override: dict) -> dict:
+    merged = deepcopy(base)
+    for key, value in override.items():
+        if isinstance(value, dict) and isinstance(merged.get(key), dict):
+            merged[key] = _merge_dicts(merged[key], value)
+        else:
+            merged[key] = deepcopy(value)
+    return merged
+
+
+def find_project_config_path(start: str | Path | None = None) -> Path | None:
+    current = Path(start).expanduser().resolve() if start is not None else Path.cwd().resolve()
+    if current.is_file():
+        current = current.parent
+    for directory in (current, *current.parents):
+        candidate = directory / PROJECT_CONFIG_NAME
+        if candidate.exists():
+            return candidate
+    return None
+
+
+def _load_default_config() -> dict:
+    default_config = files("plot_saver").joinpath(DEFAULT_CONFIG_RESOURCE)
+    with default_config.open("rb") as f:
+        return tomllib.load(f)
+
+
+def load_app_config(config_path: str | Path | None = None) -> dict:
+    cfg = deepcopy(_load_default_config())
+    resolved = Path(config_path).expanduser().resolve() if config_path else find_project_config_path()
+    if resolved is not None and resolved.exists():
+        cfg = _merge_dicts(cfg, _read_toml(resolved))
+    return cfg

plot_saver-0.1.0/src/plot_saver/figure_save_widget.css

@@ -0,0 +1,66 @@
+.save-figure-wrap {
+  display: inline-flex;
+  flex-direction: column;
+  align-items: flex-start;
+  gap: 0.35rem;
+  --save-figure-radius: 8px;
+  --save-figure-padding-y: 0.45rem;
+  --save-figure-padding-x: 0.9rem;
+  --save-figure-font-size: 0.9rem;
+  --save-figure-light-border: rgba(107, 114, 128, 0.35);
+  --save-figure-light-background: rgba(249, 250, 251, 0.95);
+  --save-figure-light-text: #111827;
+  --save-figure-light-hover-background: rgba(243, 244, 246, 1);
+  --save-figure-light-hover-border: rgba(107, 114, 128, 0.55);
+  --save-figure-light-disabled-background: rgba(229, 231, 235, 0.9);
+  --save-figure-light-disabled-border: rgba(156, 163, 175, 0.35);
+  --save-figure-dark-border: rgba(156, 163, 175, 0.35);
+  --save-figure-dark-background: rgba(31, 41, 55, 0.95);
+  --save-figure-dark-text: #f3f4f6;
+  --save-figure-dark-hover-background: rgba(55, 65, 81, 1);
+  --save-figure-dark-hover-border: rgba(209, 213, 219, 0.45);
+  --save-figure-dark-disabled-background: rgba(55, 65, 81, 0.8);
+  --save-figure-dark-disabled-border: rgba(107, 114, 128, 0.35);
+}
+
+.save-figure-btn {
+  border: 1px solid var(--save-figure-light-border);
+  background: var(--save-figure-light-background);
+  color: var(--save-figure-light-text);
+  border-radius: var(--save-figure-radius);
+  padding: var(--save-figure-padding-y) var(--save-figure-padding-x);
+  font-size: var(--save-figure-font-size);
+  line-height: 1;
+  cursor: pointer;
+  transition: background 120ms ease, border-color 120ms ease;
+}
+
+.save-figure-btn:hover {
+  background: var(--save-figure-light-hover-background);
+  border-color: var(--save-figure-light-hover-border);
+}
+
+.save-figure-btn:disabled {
+  cursor: not-allowed;
+  opacity: 0.55;
+  background: var(--save-figure-light-disabled-background);
+  border-color: var(--save-figure-light-disabled-border);
+}
+
+@media (prefers-color-scheme: dark) {
+  .save-figure-btn {
+    background: var(--save-figure-dark-background);
+    color: var(--save-figure-dark-text);
+    border-color: var(--save-figure-dark-border);
+  }
+
+  .save-figure-btn:hover {
+    background: var(--save-figure-dark-hover-background);
+    border-color: var(--save-figure-dark-hover-border);
+  }
+
+  .save-figure-btn:disabled {
+    background: var(--save-figure-dark-disabled-background);
+    border-color: var(--save-figure-dark-disabled-border);
+  }
+}

plot_saver-0.1.0/src/plot_saver/figure_save_widget.js

@@ -0,0 +1,40 @@
+function render({ model, el }) {
+  el.innerHTML = "";
+
+  const wrap = document.createElement("div");
+  wrap.className = "save-figure-wrap";
+
+  const button = document.createElement("button");
+  button.className = "save-figure-btn";
+
+  const applyTheme = () => {
+    const theme = model.get("theme_tokens") || {};
+    for (const [key, value] of Object.entries(theme)) {
+      wrap.style.setProperty(`--save-figure-${key}`, value);
+    }
+  };
+
+  const updateButton = () => {
+    button.textContent = model.get("label");
+    button.disabled = !!model.get("disabled");
+  };
+
+  button.addEventListener("click", () => {
+    if (button.disabled) return;
+    const clicks = model.get("clicks") || 0;
+    model.set("clicks", clicks + 1);
+    model.save_changes();
+  });
+
+  model.on("change:label", updateButton);
+  model.on("change:disabled", updateButton);
+  model.on("change:theme_tokens", applyTheme);
+
+  applyTheme();
+  updateButton();
+
+  wrap.appendChild(button);
+  el.appendChild(wrap);
+}
+
+export default { render };

plot_saver-0.1.0/src/plot_saver/resources/default_config.toml

@@ -0,0 +1,31 @@
+[plot-saver]
+format = "png"
+default_label = "Save"
+save_all_label = "Save all model plots"
+empty_title = "No plots available"
+empty_detail = "Render the notebook plots first."
+saved_title = "Saved"
+saved_many_title = "Saved {count} plots"
+saved_error_title = "Saved with errors"
+failed_title = "Could not save plots"
+toast_detail_color = "#6b7280"
+
+[plot-saver.theme]
+radius = "8px"
+padding_y = "0.45rem"
+padding_x = "0.9rem"
+font_size = "0.9rem"
+light_border = "rgba(107, 114, 128, 0.35)"
+light_background = "rgba(249, 250, 251, 0.95)"
+light_text = "#111827"
+light_hover_background = "rgba(243, 244, 246, 1)"
+light_hover_border = "rgba(107, 114, 128, 0.55)"
+light_disabled_background = "rgba(229, 231, 235, 0.9)"
+light_disabled_border = "rgba(156, 163, 175, 0.35)"
+dark_border = "rgba(156, 163, 175, 0.35)"
+dark_background = "rgba(31, 41, 55, 0.95)"
+dark_text = "#f3f4f6"
+dark_hover_background = "rgba(55, 65, 81, 1)"
+dark_hover_border = "rgba(209, 213, 219, 0.45)"
+dark_disabled_background = "rgba(55, 65, 81, 0.8)"
+dark_disabled_border = "rgba(107, 114, 128, 0.35)"

plot_saver-0.1.0/src/plot_saver/save_widget.py

@@ -0,0 +1,331 @@
+from __future__ import annotations
+
+from pathlib import Path
+import re
+from typing import Any
+
+import anywidget
+import traitlets
+
+from .anywidget_compat import wrap_anywidget
+from .config import find_project_config_path, load_app_config
+
+
+_ASSET_DIR = Path(__file__).parent
+
+
+def _read_asset(name: str) -> str:
+    return (_ASSET_DIR / name).read_text(encoding="utf-8")
+
+
+def _get_save_figure_config(config_path: Path | None) -> dict[str, str]:
+    cfg = load_app_config(config_path)
+    section = cfg.get("plot-saver", {})
+    if not isinstance(section, dict):
+        return {}
+    save_cfg = {str(key): str(value) for key, value in section.items() if not isinstance(value, dict)}
+    theme_section = section.get("theme", {})
+    if isinstance(theme_section, dict):
+        save_cfg.update({str(key): str(value) for key, value in theme_section.items()})
+    return save_cfg
+
+
+def _save_figure_theme_tokens(save_cfg: dict[str, str]) -> dict[str, str]:
+    token_keys = (
+        "radius",
+        "padding_y",
+        "padding_x",
+        "font_size",
+        "light_border",
+        "light_background",
+        "light_text",
+        "light_hover_background",
+        "light_hover_border",
+        "light_disabled_background",
+        "light_disabled_border",
+        "dark_border",
+        "dark_background",
+        "dark_text",
+        "dark_hover_background",
+        "dark_hover_border",
+        "dark_disabled_background",
+        "dark_disabled_border",
+    )
+    return {
+        key.replace("_", "-"): value
+        for key in token_keys
+        if (value := save_cfg.get(key))
+    }
+
+
+def _toast_detail_html(detail: str, save_cfg: dict[str, str]) -> str:
+    color = save_cfg.get("toast_detail_color", "#6b7280")
+    return f"<span style='color:{color}'>{detail}</span>"
+
+
+class SaveFigureAnyWidget(anywidget.AnyWidget):
+    _esm = _read_asset("figure_save_widget.js")
+    _css = _read_asset("figure_save_widget.css")
+
+    clicks = traitlets.Int(0).tag(sync=True)
+    label = traitlets.Unicode("Save").tag(sync=True)
+    disabled = traitlets.Bool(False).tag(sync=True)
+    theme_tokens = traitlets.Dict(default_value={}).tag(sync=True)
+
+
+def get_plot_save_format(config_path: Path | None = None) -> str:
+    cfg = load_app_config(config_path)
+    fmt = str(cfg.get("plot-saver", {}).get("format", "pdf")).lower().strip(". ")
+    if fmt not in {"pdf", "svg", "png"}:
+        fmt = "pdf"
+    return fmt
+
+
+def sanitize_stem(stem: str) -> str:
+    stem = re.sub(r"[^A-Za-z0-9._-]+", "_", stem).strip("._-")
+    return stem or "figure"
+
+
+def build_plot_path(results_dir: Path, task_name: str, model_id: str, stem: str, fmt: str) -> Path:
+    out_dir = results_dir / "plots" / task_name / model_id
+    out_dir.mkdir(parents=True, exist_ok=True)
+    return out_dir / f"{sanitize_stem(stem)}.{fmt}"
+
+
+def _axis_for_location(fig: Any, location: tuple[int, int]):
+    row, col = (int(location[0]), int(location[1]))
+    for ax in getattr(fig, "axes", []):
+        get_subplotspec = getattr(ax, "get_subplotspec", None)
+        if get_subplotspec is None:
+            continue
+        spec = get_subplotspec()
+        if spec is None:
+            continue
+        if row in range(spec.rowspan.start, spec.rowspan.stop) and col in range(spec.colspan.start, spec.colspan.stop):
+            return ax
+    raise ValueError(f"No subplot found at location ({row}, {col}).")
+
+
+def _save_axis(fig: Any, ax: Any, out_path: Path, fmt: str) -> Path:
+    if hasattr(fig, "canvas") and fig.canvas is not None:
+        fig.canvas.draw()
+    renderer = fig.canvas.get_renderer()
+    bbox = ax.get_tightbbox(renderer).transformed(fig.dpi_scale_trans.inverted())
+    save_kwargs = {"bbox_inches": bbox}
+    if fmt != "svg":
+        save_kwargs["dpi"] = 300
+    fig.savefig(out_path, **save_kwargs)
+    return out_path
+
+
+def save_figure(
+    fig,
+    *,
+    results_dir: Path,
+    config_path: Path | None,
+    task_name: str,
+    model_id: str,
+    stem: str,
+    location: tuple[int, int] | None = None,
+) -> Path:
+    fmt = get_plot_save_format(config_path)
+    out_path = build_plot_path(results_dir, task_name, model_id, stem, fmt)
+    if location is not None:
+        ax = _axis_for_location(fig, location)
+        return _save_axis(fig, ax, out_path, fmt)
+    if hasattr(fig, "canvas") and fig.canvas is not None:
+        fig.canvas.draw()
+    save_kwargs = {"bbox_inches": "tight"}
+    if fmt != "svg":
+        save_kwargs["dpi"] = 300
+    fig.savefig(out_path, **save_kwargs)
+    return out_path
+
+
+class PlotSaver:
+    def __init__(self, mo, *, results_dir: Path, config_path: Path | None, task_name: str, model_id: str):
+        self.mo = mo
+        self.results_dir = results_dir
+        self.config_path = config_path or find_project_config_path()
+        self.task_name = task_name
+        self.model_id = model_id
+        self.fmt = get_plot_save_format(self.config_path)
+        self.save_cfg = _get_save_figure_config(self.config_path)
+        self.theme_tokens = _save_figure_theme_tokens(self.save_cfg)
+        self._registry: dict[str, dict[str, object]] = {}
+        self._save_all = SaveFigureAnyWidget(
+            label=self.save_cfg.get("save_all_label", "Save all model plots"),
+            disabled=True,
+            theme_tokens=self.theme_tokens,
+        )
+        self._save_all.observe(self._handle_save_all_click, names="clicks")
+        self._save_all._save_observer = self._handle_save_all_click
+        self._save_all_ui = None
+
+    def _save_one(self, fig, *, stem: str, location: tuple[int, int] | None = None) -> Path:
+        return save_figure(
+            fig,
+            results_dir=self.results_dir,
+            config_path=self.config_path,
+            task_name=self.task_name,
+            model_id=self.model_id,
+            stem=stem,
+            location=location,
+        )
+
+    def _register(self, fig, *, name: str, stem: str, location: tuple[int, int] | None = None) -> None:
+        self._registry[stem] = {
+            "fig": fig,
+            "name": name,
+            "stem": stem,
+            "location": location,
+        }
+        self._save_all.disabled = not bool(self._registry)
+
+    def _saved_message(self, saved_paths: list[Path]) -> str:
+        if not saved_paths:
+            return "No files saved."
+        if len(saved_paths) == 1:
+            return saved_paths[0].name
+        return f"{saved_paths[0].name} + {len(saved_paths) - 1} more"
+
+    def save_all(self) -> tuple[list[Path], list[tuple[str, Exception]]]:
+        saved_paths: list[Path] = []
+        errors: list[tuple[str, Exception]] = []
+        for item in list(self._registry.values()):
+            try:
+                out_path = self._save_one(
+                    item["fig"],
+                    stem=str(item["stem"]),
+                    location=item["location"],
+                )
+                saved_paths.append(out_path)
+            except Exception as exc:
+                errors.append((str(item["name"]), exc))
+        return saved_paths, errors
+
+    def _handle_save_all_click(self, change) -> None:
+        if int(change["new"]) <= int(change["old"]):
+            return
+        if not self._registry:
+            self.mo.status.toast(
+                self.save_cfg.get("empty_title", "No plots available"),
+                _toast_detail_html(
+                    self.save_cfg.get("empty_detail", "Render the notebook plots first."),
+                    self.save_cfg,
+                ),
+                kind="danger",
+            )
+            return
+
+        saved_paths, errors = self.save_all()
+        if errors:
+            detail = self._saved_message(saved_paths)
+            if saved_paths:
+                detail = f"{detail}; {len(errors)} failed"
+            else:
+                detail = f"{len(errors)} failed"
+            self.mo.status.toast(
+                self.save_cfg.get(
+                    "saved_error_title" if saved_paths else "failed_title",
+                    "Saved with errors" if saved_paths else "Could not save plots",
+                ),
+                _toast_detail_html(detail, self.save_cfg),
+                kind="danger",
+            )
+            return
+
+        count = len(saved_paths)
+        title = (
+            self.save_cfg.get("saved_title", "Saved")
+            if count == 1
+            else self.save_cfg.get("saved_many_title", "Saved {count} plots").format(count=count)
+        )
+        self.mo.status.toast(
+            title,
+            _toast_detail_html(self._saved_message(saved_paths), self.save_cfg),
+        )
+
+    def save_all_widget(self, label: str | None = None):
+        self._save_all.label = label or self.save_cfg.get("save_all_label", "Save all model plots")
+        if self._save_all_ui is None:
+            self._save_all_ui = wrap_anywidget(self._save_all)
+        return self._save_all_ui
+
+    def __call__(
+        self,
+        fig,
+        name: str,
+        *,
+        stem: str | None = None,
+        label: str | None = None,
+        location: tuple[int, int] | None = None,
+    ):
+        if location is not None:
+            row, col = (int(location[0]), int(location[1]))
+            default_stem = f"{sanitize_stem(name.lower())}_r{row}_c{col}"
+        else:
+            default_stem = sanitize_stem(name.lower())
+        resolved_stem = stem or default_stem
+        button_label = label or f"{self.save_cfg.get('default_label', 'Save')} .{self.fmt}"
+        self._register(fig, name=name, stem=resolved_stem, location=location)
+        widget = SaveFigureAnyWidget(label=button_label, theme_tokens=self.theme_tokens)
+
+        def _handle_click(change):
+            if int(change["new"]) <= int(change["old"]):
+                return
+            try:
+                out_path = self._save_one(fig, stem=resolved_stem, location=location)
+                self.mo.status.toast(
+                    self.save_cfg.get("saved_title", "Saved"),
+                    _toast_detail_html(out_path.name, self.save_cfg),
+                )
+            except Exception as exc:
+                self.mo.status.toast(
+                    self.save_cfg.get("failed_title", "Could not save plots"),
+                    _toast_detail_html(f"{type(exc).__name__}: {exc}", self.save_cfg),
+                    kind="danger",
+                )
+
+        widget.observe(_handle_click, names="clicks")
+        widget._save_observer = _handle_click
+        return wrap_anywidget(widget)
+
+
+def make_plot_saver(mo, *, results_dir: Path, config_path: Path | None, task_name: str, model_id: str):
+    return PlotSaver(
+        mo,
+        results_dir=results_dir,
+        config_path=config_path,
+        task_name=task_name,
+        model_id=model_id,
+    )
+
+
+def save_button(
+    mo,
+    fig,
+    *,
+    results_dir: Path,
+    config_path: Path | None,
+    task_name: str,
+    model_id: str,
+    stem: str,
+    label: str | None = None,
+    location: tuple[int, int] | None = None,
+):
+    fmt = get_plot_save_format(config_path)
+    default_label = label or _get_save_figure_config(config_path).get("default_label", "Save")
+    return make_plot_saver(
+        mo,
+        results_dir=results_dir,
+        config_path=config_path,
+        task_name=task_name,
+        model_id=model_id,
+    )(
+        fig,
+        name=default_label,
+        stem=stem,
+        label=f"{default_label} .{fmt}",
+        location=location,
+    )