Update examples

changelog.md

@@ -18,7 +18,7 @@
 - Allow setting the theme manually in `VG.render(theme="light")` and `VG.render_widget(theme="dark")`.
 - Use typed nodes and relationship traitlets in GraphWidget, i.e., list of Node and Relationship instead of dictionaries.
 - `render` now allows to pass `layout` as a string as well. Previously expected to be a typed `neo4j_viz.Layout`. 
-- Fixed rendering in Marimo notebooks. Note, you still have to manually pass the marimo theme `VG.render_widget(theme=marimo.app_meta().theme)`
+- Fixed rendering in Marimo notebooks
 
 
 ## Other changes

examples/.gitignore

@@ -1,4 +1,4 @@
 .ipynb_checkpoints
-
+__marimo__
 out/*
-/__marimo__/session/*
+

examples/marimo-example.py

@@ -66,11 +66,11 @@ def _(mo):
 
 
 @app.cell
-def _(VisualizationGraph, mo, nodes, relationships):
+def _(VisualizationGraph, nodes, relationships):
     # Create and render the visualization
     VG = VisualizationGraph(nodes=nodes, relationships=relationships)
     # Note, we need to pass the theme explicitly in Marimo.
-    widget = VG.render_widget(theme=mo.app_meta().theme, renderer="canvas")
+    widget = VG.render_widget(renderer="canvas")
     widget
     return VG, widget
 

js-applet/src/graph-widget.tsx

@@ -37,10 +37,10 @@ export type WidgetData = {
 };
 
 function detectTheme(): "light" | "dark" {
-  if (document.body.classList.contains("vscode-light")) {
+  if (document.body.classList.contains("vscode-light") || document.body.classList.contains("light-theme")) {
     return "light";
   }
-  if (document.body.classList.contains("vscode-dark")) {
+  if (document.body.classList.contains("vscode-dark") || document.body.classList.contains("dark-theme")) {
     return "dark";
   }
 

python-wrapper/.gitignore

@@ -9,3 +9,4 @@ src/neo4j_viz/resources/nvl_entrypoint/*
 !src/neo4j_viz/resources/nvl_entrypoint/style.css
 !src/neo4j_viz/resources/nvl_entrypoint/index.html
 !src/neo4j_viz/resources/nvl_entrypoint/__init__.py
+out/*

python-wrapper/pyproject.toml

@@ -76,7 +76,7 @@ notebook = [
     "palettable>=3.3.3",
     "matplotlib>=3.9.4",
     "snowflake-snowpark-python==1.42.0",
-    "dotenv",
+    "python-dotenv",
     "requests",
     "marimo",
 ]

python-wrapper/src/neo4j_viz/resources/nvl_entrypoint/widget.js

@@ -82204,9 +82204,9 @@ class Hdr extends kr.Component {
   }
 }
 function Vdr() {
-  if (document.body.classList.contains("vscode-light"))
+  if (document.body.classList.contains("vscode-light") || document.body.classList.contains("light-theme"))
     return "light";
-  if (document.body.classList.contains("vscode-dark"))
+  if (document.body.classList.contains("vscode-dark") || document.body.classList.contains("dark-theme"))
     return "dark";
   const r = window.getComputedStyle(document.body, null).getPropertyValue("background-color").match(/\d+/g);
   if (!r || r.length < 3)

python-wrapper/src/neo4j_viz/visualization_graph.py

@@ -84,6 +84,9 @@ def __init__(self, nodes: list[Node], relationships: list[Relationship]) -> None
         self.nodes = nodes
         self.relationships = relationships
 
+    def __str__(self) -> str:
+        return f"VisualizationGraph(nodes={len(self.nodes)}, relationships={len(self.relationships)})"
+
     def _build_render_options(
         self,
         layout: Layout | str | None,

scripts/clean_js_applet.sh

@@ -7,5 +7,4 @@ set -o pipefail
 (
     cd "${GIT_ROOT}/js-applet"
     rm -rf node_modules
-    rm -f yarn.lock
 )