Add graph_widget widget

Author: basnijholt

src/graphviz_anywidget/__init__.py

@@ -1,7 +1,9 @@
 import importlib.metadata
 from pathlib import Path
+from typing import Any
 
 import anywidget
+import ipywidgets
 import traitlets
 
 try:
@@ -33,3 +35,60 @@ class GraphvizWidget(anywidget.AnyWidget):
     selected_direction = traitlets.Unicode("bidirectional").tag(sync=True)
     search_type = traitlets.Unicode("included").tag(sync=True)
     case_sensitive = traitlets.Bool(False).tag(sync=True)  # noqa: FBT003
+
+
+def graph_widget(
+    dot_string: str = "digraph { a -> b; b -> c; c -> a; }",
+) -> ipywidgets.VBox:
+    widget = GraphvizWidget(dot_source=dot_string)
+    reset_button = ipywidgets.Button(description="Reset Zoom")
+    direction_selector = ipywidgets.Dropdown(
+        options=["bidirectional", "downstream", "upstream", "single"],
+        value="bidirectional",
+        description="Direction:",
+    )
+    search_input = ipywidgets.Text(
+        placeholder="Search...",
+        description="Search:",
+    )
+    search_type_selector = ipywidgets.Dropdown(
+        options=["exact", "included", "regex"],
+        value="exact",
+        description="Search Type:",
+    )
+    case_toggle = ipywidgets.ToggleButton(
+        value=False,
+        description="Case Sensitive",
+        icon="check",
+    )
+
+    # Define button actions
+    def reset_graph(_: Any) -> None:
+        widget.send({"action": "reset_zoom"})
+
+    def update_direction(change: dict) -> None:
+        widget.selected_direction = change["new"]
+
+    def perform_search(change: dict) -> None:
+        widget.send({"action": "search", "query": change["new"]})
+
+    def update_search_type(change: dict) -> None:
+        widget.search_type = change["new"]
+
+    def toggle_case_sensitive(change: dict) -> None:
+        widget.case_sensitive = change["new"]
+
+    reset_button.on_click(reset_graph)
+    direction_selector.observe(update_direction, names="value")
+    search_input.observe(perform_search, names="value")
+    search_type_selector.observe(update_search_type, names="value")
+    case_toggle.observe(toggle_case_sensitive, names="value")
+
+    # Display ipywidgets
+    return ipywidgets.VBox(
+        [
+            ipywidgets.HBox([reset_button, direction_selector]),
+            ipywidgets.HBox([search_input, search_type_selector, case_toggle]),
+            widget,
+        ],
+    )