Merge pull request #26 from pyiron/node_colors_themes

Node colors themes

pyironflow/reactflow.py

@@ -1,5 +1,6 @@
 from pyiron_workflow import Workflow
 from pyiron_workflow.channels import NotData
+from pyironflow.themes import get_color
 
 import anywidget
 import pathlib
@@ -145,7 +146,7 @@ def get_node_dict(node, id_num, key=None):
         'type': 'customNode',
         'style': {'border': '1px black solid',
                   'padding': 5,
-                  'background': '#1999', # node.gui_color,  # This requires Tara's branch of pyiron_workflow
+                  'background': get_color(node=node, theme='light'),
                   'borderRadius': '10px',
                   'width': f'{node_width}px'},
         'targetPosition': 'left',

pyironflow/themes.py

@@ -0,0 +1,29 @@
+import typing
+
+from pyiron_workflow.nodes.function import Function
+from pyiron_workflow.nodes.macro import Macro
+from pyiron_workflow.nodes.transform import DataclassNode
+
+if typing.TYPE_CHECKING:
+    from pyiron_workflow.node import Node
+
+
+def get_color(node: Node, theme: typing.Literal['light']):
+
+    if theme == 'light':
+        return light_mode(node)
+    else:
+        raise ValueError(f'Theme must be one of ("light",) but got {theme}')
+
+def light_mode(node: Node):
+    if isinstance(node, Function):
+        color_light_green = "#a2ea9f"
+        return color_light_green
+    elif isinstance(node, Macro):
+        color_light_orange = "#eacf9f"
+        return color_light_orange
+    elif isinstance(node, DataclassNode):
+        color_light_purple = "#cb9fea"
+        return color_light_purple
+    else:
+        return node.color