omle-viewer 0.1.0__py3-none-any.whl

omle_viewer/__init__.py

@@ -0,0 +1,10 @@
+from .display import OMLEDisplay, model_repr_html, show, show_in_browser
+
+try:
+    from ._version import (
+        __version__,  # generated by setuptools-scm at build time
+    )
+except ImportError:  # pragma: no cover - source checkout without a build
+    __version__ = "0.0.0+unknown"
+
+__all__ = ["OMLEDisplay", "show", "show_in_browser", "model_repr_html", "__version__"]

omle_viewer/_version.py

@@ -0,0 +1,24 @@
+# file generated by vcs-versioning
+# don't change, don't track in version control
+from __future__ import annotations
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+version: str
+__version__: str
+__version_tuple__: tuple[int | str, ...]
+version_tuple: tuple[int | str, ...]
+commit_id: str | None
+__commit_id__: str | None
+
+__version__ = version = '0.1.0'
+__version_tuple__ = version_tuple = (0, 1, 0)
+
+__commit_id__ = commit_id = None

omle_viewer/cli.py

@@ -0,0 +1,55 @@
+"""CLI entry point: ``omle-viewer <model-file>``
+
+Opens an OMLE model file in the default web browser using the interactive
+DAG viewer. Supports JSON (.json) and protobuf binary (.omle, .pb, .bin)
+file formats.
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        prog="omle-viewer",
+        description="Open an OMLE model in the interactive DAG viewer.",
+    )
+    parser.add_argument(
+        "model",
+        metavar="MODEL_FILE",
+        help="Path to the model file (.json, .omle, .pb, .bin)",
+    )
+    args = parser.parse_args()
+
+    path = Path(args.model)
+    if not path.exists():
+        print(f"omle-viewer: file not found: {path}", file=sys.stderr)
+        sys.exit(1)
+
+    try:
+        import omle
+    except ImportError:
+        print(
+            "omle-viewer: the 'omle' package is required. "
+            "Install it with: pip install omle",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    # omle.load dispatches on the extension and decodes both encodings; omle
+    # depends on protobuf directly, so binary files need no optional extra.
+    try:
+        model = omle.load(path)
+    except ValueError as e:
+        print(f"omle-viewer: {e}", file=sys.stderr)
+        sys.exit(1)
+
+    from .display import show_in_browser
+    show_in_browser(model)
+
+
+if __name__ == "__main__":
+    main()

omle_viewer/display.py

@@ -0,0 +1,172 @@
+"""Jupyter display helpers for OMLE models.
+
+Primary approach: self-contained HTML iframe (works in every Jupyter environment
+without any widget extensions — JupyterLab, Classic Notebook, VS Code, Colab).
+
+The viewer HTML is built by running ``./build.sh`` at the repo root and embedded
+as a base64 data-URI iframe so it needs no external server or asset loading.
+"""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from omle.ir.model import OMLEModel
+
+_STATIC = Path(__file__).parent / "static"
+_VIEWER_HTML = _STATIC / "viewer.html"
+
+# ── HTML helpers ──────────────────────────────────────────────────────────────
+
+_NOT_BUILT_HTML = """
+<div style="padding:14px 16px;font-family:monospace;font-size:12px;
+color:#b00;border:1px solid #f5c0c0;border-radius:6px;background:#fff5f5">
+<strong>omle-viewer:</strong> The viewer bundle has not been built yet.<br><br>
+Run the following from the repo root, then reinstall the package:<br>
+<code style="display:block;margin-top:6px;padding:6px 8px;background:#fff;
+border:1px solid #ddd;border-radius:4px">
+./build.sh<br>
+pip install -e omle-viewer/python/
+</code>
+</div>
+"""
+
+
+def _build_viewer_html(model: OMLEModel) -> str | None:
+    """Return the full self-contained viewer HTML with the model pre-loaded, or None if not built."""
+    if not _VIEWER_HTML.exists():
+        return None
+    viewer = _VIEWER_HTML.read_text(encoding="utf-8")
+    model_json = json.dumps(model.to_dict())
+    inject = f"<script>window.__OMLE_MODEL__ = {model_json};</script>"
+    return viewer.replace("</head>", inject + "</head>", 1)
+
+
+def _build_iframe_html(model: OMLEModel, height: int) -> str:
+    """Return an ``<iframe srcdoc=...>`` HTML string that embeds the full interactive viewer.
+
+    Uses srcdoc instead of a data: URI because JupyterLab's CSP blocks data: iframes.
+    """
+    viewer = _build_viewer_html(model)
+    if viewer is None:
+        return _NOT_BUILT_HTML
+    import html as _html
+    # srcdoc is same-origin with the parent page so CSP `unsafe-inline` covers it;
+    # data: URIs are blocked by JupyterLab's frame-src CSP directive.
+    encoded = _html.escape(viewer, quote=True)
+    return (
+        f'<iframe srcdoc="{encoded}" '
+        f'style="width:100%;height:{height}px;border:none;border-radius:4px;" '
+        f'allowfullscreen>'
+        f"</iframe>"
+    )
+
+
+def show_in_browser(model: OMLEModel) -> None:
+    """Open the interactive DAG viewer for *model* in the default web browser.
+
+    Writes a temporary self-contained HTML file and opens it via
+    ``webbrowser.open()``. Works from a script, a terminal, or a notebook.
+
+    Parameters
+    ----------
+    model:
+        The OMLEModel instance to visualize.
+    """
+    import tempfile
+    import webbrowser
+
+    viewer = _build_viewer_html(model)
+    if viewer is None:
+        raise RuntimeError(
+            "omle-viewer: viewer bundle not built. "
+            "Run ./build.sh at the repo root first."
+        )
+
+    with tempfile.NamedTemporaryFile(
+        suffix=".html", delete=False, mode="w", encoding="utf-8"
+    ) as f:
+        f.write(viewer)
+        tmp_path = f.name
+
+    webbrowser.open(f"file://{tmp_path}")
+
+
+# ── Public API ────────────────────────────────────────────────────────────────
+
+class OMLEDisplay:
+    """Jupyter display object wrapping an OMLEModel.
+
+    Returned by ``show()`` so callers have a reference, but suppresses the
+    cell's auto-display of the return value (the iframe was already shown by
+    the explicit ``IPython.display.display()`` call inside ``show()``).
+    """
+
+    def __init__(self, model: OMLEModel, height: int = 500) -> None:
+        self._model = model
+        self._height = height
+
+    # Suppress Jupyter's implicit display of the return value.
+    def _repr_mimebundle_(self, **_kwargs):
+        return {}
+
+    def __repr__(self) -> str:  # plain-Python fallback
+        return "<OMLEDisplay — call .show() to redisplay>"
+
+    def show(self, height: int | None = None) -> None:
+        """Re-display the viewer (e.g. after calling .update())."""
+        try:
+            import warnings
+
+            from IPython.display import HTML, display
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore", UserWarning)
+                display(HTML(_build_iframe_html(self._model, height or self._height)))
+        except ImportError:
+            pass
+
+    def update(self, model: OMLEModel, *, height: int | None = None) -> None:
+        """Display an updated model in a new cell output."""
+        self._model = model
+        if height is not None:
+            self._height = height
+        self.show()
+
+
+def model_repr_html(model: OMLEModel, height: int = 500) -> str:
+    """Return the ``_repr_html_`` string for *model* (used by the display hook)."""
+    return _build_iframe_html(model, height)
+
+
+def show(model: OMLEModel, height: int = 500) -> OMLEDisplay:
+    """Display an OMLEModel DAG inline in a Jupyter notebook.
+
+    Importing ``omle_viewer`` already makes models auto-render when
+    returned from a cell, so ``show()`` is mainly useful when you want a
+    non-default height or a handle for later updates::
+
+        w = show(model, height=700)
+        # push a new model without creating a new cell output:
+        w.update(updated_model)
+
+    Parameters
+    ----------
+    model:
+        The OMLEModel instance to visualize.
+    height:
+        Viewer height in pixels (default 500).
+    """
+    handle = OMLEDisplay(model, height)
+    try:
+        import warnings
+
+        from IPython.display import HTML, display
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", UserWarning)
+            display(HTML(_build_iframe_html(model, height)))
+    except ImportError:
+        pass
+    return handle