figrecipe 0.5.0__py3-none-any.whl → 0.7.4__py3-none-any.whl

figrecipe/_editor/_routes_style.py

@@ -0,0 +1,223 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Style and theme Flask route handlers for the figure editor.
+"""
+
+from flask import jsonify, request
+
+from ._helpers import get_form_values_from_style, render_with_overrides
+
+
+def register_style_routes(app, editor):
+    """Register style/theme routes with the Flask app."""
+    from ._overrides import save_overrides
+
+    @app.route("/style")
+    def get_style():
+        """Get current style configuration."""
+        return jsonify(
+            {
+                "base_style": editor.style_overrides.base_style,
+                "programmatic_style": editor.style_overrides.programmatic_style,
+                "manual_overrides": editor.style_overrides.manual_overrides,
+                "effective_style": editor.get_effective_style(),
+                "has_overrides": editor.style_overrides.has_manual_overrides(),
+                "manual_timestamp": editor.style_overrides.manual_timestamp,
+            }
+        )
+
+    @app.route("/theme")
+    def get_theme():
+        """Get current theme YAML content for display."""
+        import io as yaml_io
+
+        from ruamel.yaml import YAML
+
+        style = editor.get_effective_style()
+        style_name = style.get("_name", "SCITEX")
+
+        yaml = YAML()
+        yaml.default_flow_style = False
+        yaml.indent(mapping=2, sequence=4, offset=2)
+        stream = yaml_io.StringIO()
+        yaml.dump(style, stream)
+        yaml_content = stream.getvalue()
+
+        return jsonify(
+            {
+                "name": style_name,
+                "content": yaml_content,
+            }
+        )
+
+    @app.route("/list_themes")
+    def list_themes():
+        """List available theme presets."""
+        from ..styles._style_loader import list_presets
+
+        presets = list_presets()
+        current = editor.get_effective_style().get("_name", "SCITEX")
+
+        return jsonify(
+            {
+                "themes": presets,
+                "current": current,
+            }
+        )
+
+    @app.route("/switch_theme", methods=["POST"])
+    def switch_theme():
+        """Switch to a different theme preset by reproducing the figure."""
+        from .._reproducer import reproduce_from_record
+        from ..styles._style_loader import load_preset
+
+        data = request.get_json() or {}
+        theme_name = data.get("theme")
+
+        if not theme_name:
+            return jsonify({"error": "No theme specified"}), 400
+
+        try:
+            new_style = load_preset(theme_name)
+
+            if new_style is None:
+                return jsonify({"error": f"Theme '{theme_name}' not found"}), 404
+
+            # Convert nested style to flat style dict with color_palette
+            flat_style = dict(new_style)
+            flat_style["_name"] = theme_name
+
+            # Extract color_palette from nested colors.palette
+            if "colors" in new_style and isinstance(new_style["colors"], dict):
+                colors_dict = new_style["colors"]
+                if "palette" in colors_dict:
+                    flat_style["color_palette"] = list(colors_dict["palette"])
+
+            editor.style_overrides.base_style = flat_style
+
+            if hasattr(editor.fig, "record") and editor.fig.record is not None:
+                editor.fig.record.style = flat_style
+                new_fig, _ = reproduce_from_record(editor.fig.record)
+                editor.fig = new_fig
+                # Keep the new style (don't restore old style)
+
+            mpl_fig = editor.fig.fig if hasattr(editor.fig, "fig") else editor.fig
+            behavior = new_style.get("behavior", {})
+            for ax in mpl_fig.get_axes():
+                hide_top = behavior.get("hide_top_spine", True)
+                hide_right = behavior.get("hide_right_spine", True)
+                ax.spines["top"].set_visible(not hide_top)
+                ax.spines["right"].set_visible(not hide_right)
+
+                if behavior.get("grid", False):
+                    ax.grid(True, alpha=0.3)
+                else:
+                    ax.grid(False)
+
+            base64_img, bboxes, img_size = render_with_overrides(
+                editor.fig,
+                editor.get_effective_style(),
+                editor.dark_mode,
+            )
+
+            form_values = get_form_values_from_style(editor.get_effective_style())
+
+            return jsonify(
+                {
+                    "success": True,
+                    "theme": theme_name,
+                    "image": base64_img,
+                    "bboxes": bboxes,
+                    "img_size": {"width": img_size[0], "height": img_size[1]},
+                    "values": form_values,
+                }
+            )
+
+        except Exception as e:
+            import traceback
+
+            traceback.print_exc()
+            return jsonify({"error": f"Failed to switch theme: {str(e)}"}), 500
+
+    @app.route("/save", methods=["POST"])
+    def save():
+        """Save style overrides (stored separately from recipe)."""
+        data = request.get_json() or {}
+        editor.style_overrides.update_manual_overrides(data.get("overrides", {}))
+
+        if editor.recipe_path:
+            path = save_overrides(editor.style_overrides, editor.recipe_path)
+            return jsonify(
+                {
+                    "success": True,
+                    "path": str(path),
+                    "has_overrides": editor.style_overrides.has_manual_overrides(),
+                    "timestamp": editor.style_overrides.manual_timestamp,
+                }
+            )
+
+        return jsonify(
+            {
+                "success": True,
+                "overrides": editor.overrides,
+                "has_overrides": editor.style_overrides.has_manual_overrides(),
+            }
+        )
+
+    @app.route("/restore", methods=["POST"])
+    def restore():
+        """Restore to original style (clear manual overrides and axes positions)."""
+        from ._bbox import extract_bboxes
+
+        # Clear all manual overrides (including position overrides)
+        editor.style_overrides.clear_manual_overrides()
+
+        # Restore original axes positions
+        editor.restore_axes_positions()
+
+        if editor._initial_base64 and not editor.dark_mode:
+            base64_img = editor._initial_base64
+            import base64 as b64
+            import io
+
+            from PIL import Image
+
+            img_data = b64.b64decode(base64_img)
+            img = Image.open(io.BytesIO(img_data))
+            img_size = img.size
+            mpl_fig = editor.fig.fig if hasattr(editor.fig, "fig") else editor.fig
+            original_dpi = mpl_fig.dpi
+            mpl_fig.set_dpi(150)
+            mpl_fig.canvas.draw()
+            bboxes = extract_bboxes(mpl_fig, img_size[0], img_size[1])
+            mpl_fig.set_dpi(original_dpi)
+        else:
+            base64_img, bboxes, img_size = render_with_overrides(
+                editor.fig,
+                None,
+                editor.dark_mode,
+            )
+
+        return jsonify(
+            {
+                "success": True,
+                "image": base64_img,
+                "bboxes": bboxes,
+                "img_size": {"width": img_size[0], "height": img_size[1]},
+                "original_style": editor.style,
+            }
+        )
+
+    @app.route("/diff")
+    def get_diff():
+        """Get differences between original and manual overrides."""
+        return jsonify(
+            {
+                "diff": editor.style_overrides.get_diff(),
+                "has_overrides": editor.style_overrides.has_manual_overrides(),
+            }
+        )
+
+
+__all__ = ["register_style_routes"]

figrecipe/_editor/_templates/__init__.py

@@ -0,0 +1,152 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+HTML/CSS/JS template builder for figure editor.
+
+Uses YAML-compatible flattened key names from to_subplots_kwargs() as the
+single source of truth. No custom key mapping is needed since all keys
+now match the HTML input IDs directly.
+"""
+
+import base64
+import json
+from datetime import datetime
+from pathlib import Path
+from typing import Any, Dict, Tuple
+
+from ._html import HTML_TEMPLATE
+from ._scripts import SCRIPTS
+from ._styles import STYLES
+
+# Server start time for debugging template reloads
+_SERVER_START_TIME = datetime.now().strftime("%H:%M:%S")
+
+# Load SciTeX icon as base64
+_SCITEX_ICON_PATH = (
+    Path(__file__).parent.parent.parent.parent.parent / "docs" / "scitex-icon.png"
+)
+_SCITEX_ICON_BASE64 = ""
+if _SCITEX_ICON_PATH.exists():
+    with open(_SCITEX_ICON_PATH, "rb") as f:
+        _SCITEX_ICON_BASE64 = base64.b64encode(f.read()).decode("utf-8")
+
+
+def build_html_template(
+    image_base64: str,
+    bboxes: Dict[str, Any],
+    color_map: Dict[str, Any],
+    style: Dict[str, Any],
+    overrides: Dict[str, Any],
+    img_size: Tuple[int, int],
+    style_name: str = "SCITEX",
+    hot_reload: bool = False,
+    dark_mode: bool = False,
+) -> str:
+    """
+    Build complete HTML template for figure editor.
+
+    Style keys are expected to be YAML-compatible flattened names that
+    match the HTML input IDs directly (e.g., 'fonts_axis_label_pt').
+
+    Parameters
+    ----------
+    image_base64 : str
+        Base64-encoded preview image.
+    bboxes : dict
+        Element bounding boxes.
+    color_map : dict
+        Hitmap color-to-element mapping.
+    style : dict
+        Base style configuration with YAML-compatible keys.
+    overrides : dict
+        Current style overrides with YAML-compatible keys.
+    img_size : tuple
+        (width, height) of preview image.
+    style_name : str
+        Name of the applied style preset (e.g., "SCITEX", "MATPLOTLIB").
+    hot_reload : bool
+        Enable hot reload auto-reconnect JavaScript.
+    dark_mode : bool
+        Initial dark mode state from saved preferences.
+
+    Returns
+    -------
+    str
+        Complete HTML document.
+    """
+    # Merge style and overrides for initial values
+    # Keys should already match HTML input IDs (YAML-compatible flattened)
+    initial_values = {**style, **overrides}
+
+    # Hot reload JavaScript for auto-reconnect on server restart
+    hot_reload_script = ""
+    if hot_reload:
+        hot_reload_script = """
+// Hot Reload: Auto-reconnect when server restarts
+(function() {
+    let isReconnecting = false;
+    let pingInterval = null;
+
+    function showReloadBanner(show) {
+        let banner = document.getElementById('hot-reload-banner');
+        if (!banner && show) {
+            banner = document.createElement('div');
+            banner.id = 'hot-reload-banner';
+            banner.style.cssText = 'position:fixed;top:0;left:0;right:0;background:#f59e0b;' +
+                'color:#000;text-align:center;padding:8px;z-index:9999;font-weight:bold;';
+            banner.textContent = 'Server restarting... will reload automatically';
+            document.body.prepend(banner);
+        }
+        if (banner) {
+            banner.style.display = show ? 'block' : 'none';
+        }
+    }
+
+    function ping() {
+        fetch('/ping', {cache: 'no-store'})
+            .then(r => {
+                if (r.ok && isReconnecting) {
+                    // Server is back! Reload the page
+                    console.log('[Hot Reload] Server is back, reloading...');
+                    window.location.reload();
+                }
+            })
+            .catch(() => {
+                if (!isReconnecting) {
+                    console.log('[Hot Reload] Server disconnected, waiting for restart...');
+                    isReconnecting = true;
+                    showReloadBanner(true);
+                }
+            });
+    }
+
+    // Start pinging every 500ms
+    pingInterval = setInterval(ping, 500);
+    console.log('[Hot Reload] Enabled - watching for server restarts');
+})();
+"""
+
+    # Inject data into template
+    html = HTML_TEMPLATE
+    html = html.replace("/* STYLES_PLACEHOLDER */", STYLES)
+    html = html.replace("/* SCRIPTS_PLACEHOLDER */", SCRIPTS + hot_reload_script)
+    html = html.replace("IMAGE_BASE64_PLACEHOLDER", image_base64)
+    html = html.replace("BBOXES_PLACEHOLDER", json.dumps(bboxes))
+    html = html.replace("COLOR_MAP_PLACEHOLDER", json.dumps(color_map))
+    html = html.replace("INITIAL_VALUES_PLACEHOLDER", json.dumps(initial_values))
+    html = html.replace("IMG_WIDTH_PLACEHOLDER", str(img_size[0]))
+    html = html.replace("IMG_HEIGHT_PLACEHOLDER", str(img_size[1]))
+    html = html.replace("STYLE_NAME_PLACEHOLDER", style_name)
+    html = html.replace("SCITEX_ICON_PLACEHOLDER", _SCITEX_ICON_BASE64)
+
+    # Dark mode preference - set initial state
+    html = html.replace("DARK_MODE_THEME_PLACEHOLDER", "dark" if dark_mode else "light")
+    html = html.replace("DARK_MODE_CHECKED_PLACEHOLDER", "checked" if dark_mode else "")
+
+    # Server start time for debugging
+    html = html.replace("SERVER_START_TIME_PLACEHOLDER", _SERVER_START_TIME)
+
+    return html
+
+
+__all__ = ["build_html_template"]