Cleanup

clarkmcc · Feb 4, 2024 · 4e8e43e · 4e8e43e
1 parent b597692
commit 4e8e43e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 17 deletions.
diff --git a/lib/NodeGraphEditor.tsx b/lib/NodeGraphEditor.tsx
@@ -71,17 +71,17 @@ export const ExampleNodeGraphEditor = forwardRef<
 >(({ nodes, edges, config: _config }: ExampleNodeGraphEditorProps, ref) => {
   const config = useBuildGraphConfig(_config)
   return (
-      <NodeGraphEditor
-        config={config}
-        ref={ref}
-        defaultNodes={nodes}
-        defaultEdges={edges}
-        colorMode="dark"
-      >
-        <Background color="#52525b" variant={BackgroundVariant.Dots} />
-        <MiniMap />
-        <Controls />
-      </NodeGraphEditor>
+    <NodeGraphEditor
+      config={config}
+      ref={ref}
+      defaultNodes={nodes}
+      defaultEdges={edges}
+      colorMode="dark"
+    >
+      <Background color="#52525b" variant={BackgroundVariant.Dots} />
+      <MiniMap />
+      <Controls />
+    </NodeGraphEditor>
   )
 })
 
@@ -165,6 +165,7 @@ const Flow = forwardRef<NodeGraphHandle, FlowProps>(
       >
         <ReactFlow
           {...props}
+          colorMode={props.colorMode ?? 'dark'}
           nodes={nodes}
           edges={edges}
           onNodesChange={onNodesChange}

diff --git a/lib/NodeGraphEditorLayout.stories.tsx b/lib/NodeGraphEditorLayout.stories.tsx
@@ -8,12 +8,14 @@ const meta = {
     const ref = useRef<NodeGraphHandle>(null)
     return (
       <div style={{ width: '100%', height: '100%' }}>
-        <button
-          style={{ position: 'absolute', top: 10, left: 10, zIndex: 1000 }}
-          onClick={() => ref.current!.layout()}
-        >
-          Layout
-        </button>
+        <div className="absolute top-3 left-3 z-10 space-x-3">
+          <button
+            className="bg-neutral-500 text-white px-4 rounded hover:bg-neutral-600"
+            onClick={() => ref.current!.layout()}
+          >
+            Layout
+          </button>
+        </div>
         <ExampleNodeGraphEditor
           ref={ref}
           config={config}