example.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "%load_ext autoreload\n",
    "%autoreload 2\n",
    "%env ANYWIDGET_HMR=1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from graphviz_anywidget import GraphvizAnyWidget, graphviz_widget"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "dot_source = \"\"\"\n",
    "\n",
    " digraph world {\n",
    "\tbgcolor=\"#2e3e56\"\n",
    "\tpad=\"0.5\" /* add padding round the edge of the graph */\n",
    "\t/*\n",
    "\t\trankdir=\"LR\"  make graph layout left->right rather than top->bottom \n",
    "\t\tgraph [fontname=\"Helvetica Neue\", fontcolor=\"#fcfcfc\"]\n",
    "\t\tlabelloc=\"t\" label at top \n",
    "\t\tlabel=\"Test Setup\"\n",
    "\n",
    "\t\tdark blue (background): #2e3e56\n",
    "\t\twhite (text/lines): #fcfcfc\n",
    "\t\tdark line (hidden lines): #445773\n",
    "\n",
    "\t\tred: #ea555b - crashes\n",
    "\t\tyellow: #edad56 - nodes in target\n",
    "\t\tgold: #AB6D16 - static libs\n",
    "\t\tdark green: #29b89d \n",
    "\t\tpurple: #9362a8\n",
    "\t\tpink: #f2688d - buckets\n",
    "\t\tgreen: #a5cf80 - 3rd party library\n",
    "\t\tblue: #8eabd9 - start\n",
    "\n",
    "\t\t0.1pt == 3.25px\n",
    "\t*/\n",
    "\n",
    "\tnode [shape=\"circle\", width=\"0.6\", style=\"filled\", fillcolor=\"#edad56\", color=\"#edad56\", penwidth=\"3\", label=\"\"]\n",
    "\tedge [color=\"#fcfcfc\", penwidth=\"2\", fontname=\"helvetica Neue Ultra Light\"]\n",
    "\n",
    "\tS35 [tooltip=\"Go S35\", fillcolor=\"#8eabd9\", color=\"#8eabd9\"];\n",
    "\t23 [shape=\"house\", tooltip=\"Go 23\"];\n",
    "\tT1 [shape=\"box\", tooltip=\"Go T1\", fillcolor=\"#9362a8\", color=\"#9362a8\"];\n",
    "\tT99 [shape=\"rectangle\", tooltip=\"Go T1\", fillcolor=\"#9362a8\", color=\"#9362a8\"];\n",
    "\tP4 [shape=\"ellipse\", tooltip=\"Go P4\", fillcolor=\"#f2688d\", color=\"#f2688d\"];\n",
    "\tT30 [shape=\"circle\", tooltip=\"Go T30\", fillcolor=\"#ea555b\", color=\"#ea555b\"];\n",
    "\n",
    "\t43 [fillcolor=\"#a5cf80\", color=\"#a5cf80\"];\n",
    "\t40 [fillcolor=\"#a5cf80\", color=\"#a5cf80\", label=\"Another\"];\n",
    "\n",
    "\t28 [fillcolor=\"#AB6D16\", color=\"#AB6D16\"];\n",
    "\t19 [fillcolor=\"#AB6D16\", color=\"#AB6D16\"];\n",
    "\n",
    "\tT30 -> P4\n",
    "\n",
    "\tS8 -> 9;\n",
    "\tS24 -> 25;\n",
    "\tS24 -> 27;\n",
    "\tS1 -> 2;\n",
    "\tS1 -> 10;\n",
    "\tS35 -> 43;\n",
    "\tS35 -> 36;\n",
    "\tS30 -> 31;\n",
    "\tS30 -> 33;\n",
    "\t9 -> 42;\n",
    "\t9 -> T1;\n",
    "\t25 -> T1 [color=\"#445773\", weight=\"0\", comment=\"weaklink\"];\n",
    "\t25 -> 26;\n",
    "\t27 -> T24;\n",
    "\t2 -> {3 ; 16 ; 17 ; T1 ; 18}\n",
    "\t10 -> { 11 ; 14 ; T1 ; 13; 12;}\n",
    "\t31 -> T1\n",
    "\t31 -> 32;\n",
    "\t33 -> T30;\n",
    "\t33 -> 34;\n",
    "\t42 -> 4;\n",
    "\t26 -> 4;\n",
    "\t3 -> 4;\n",
    "\t16 -> 15;\n",
    "\t17 -> 19;\n",
    "\t18 -> 29;\n",
    "\t11 -> 4;\n",
    "\t14 -> 15;\n",
    "\t37 -> {39 ; 41 ; 38 ; 40;}\n",
    "\t13 -> 19;\n",
    "\t12 -> 29;\n",
    "\t43 -> 38;\n",
    "\t43 -> 40;\n",
    "\t36 -> 19;\n",
    "\t32 -> 23;\n",
    "\t34 -> 29;\n",
    "\t39 -> 15;\n",
    "\t41 -> 29;\n",
    "\t38 -> 4;\n",
    "\t40 -> 19;\n",
    "\t4 -> 5;\n",
    "\t19 -> {21 ; 20 ; 28;}\n",
    "\t5 -> {6 ; T35 ; 23;}\n",
    "\t21 -> 22;\n",
    "\t20 -> 15;\n",
    "\t28 -> 29;\n",
    "\t6 -> 7;\n",
    "\t15 -> T1;\n",
    "\t22 -> T35;\n",
    "\t22 -> 23;\n",
    "\t29 -> T30;\n",
    "\t7 -> T8;\n",
    "\t23 -> T24;\n",
    "\t23 -> T1;\n",
    "}\n",
    "\"\"\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "graphviz_widget(dot_source)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "GraphvizAnyWidget(dot_source=dot_source)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "`dot_source` can also be generated by the PipeFunc package which is available on PyPI:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from pipefunc import Pipeline, pipefunc\n",
    "\n",
    "\n",
    "@pipefunc(output_name=\"c\")\n",
    "def f(a, b):\n",
    "    return a + b\n",
    "\n",
    "\n",
    "@pipefunc(output_name=\"d\")\n",
    "def g(b, c, x=1):\n",
    "    return b * c * x\n",
    "\n",
    "\n",
    "@pipefunc(output_name=\"e\")\n",
    "def h(c, d, x=1):\n",
    "    return c * d * x\n",
    "\n",
    "\n",
    "pipeline = Pipeline([f, g, h])\n",
    "\n",
    "dot_source = pipeline.visualize_graphviz(return_type=\"graphviz\").source\n",
    "graphviz_widget(dot_source)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "pipefunc",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.13.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}