close add node

Author: HomunMage

crewai-reactflow/src/App.js

@@ -1,6 +1,6 @@
 // App.js
 
-import React, { useState } from 'react';
+import React, { useState, useEffect, useCallback } from 'react';
 import ReactFlow, {
   ReactFlowProvider,
   addEdge,
@@ -57,6 +57,21 @@ const Flow = () => {
     setContextMenu(null);
   };
 
+  const handlePaneClick = useCallback((event) => {
+    if (contextMenu) {
+      setContextMenu(null);
+    }
+  }, [contextMenu]);
+
+  useEffect(() => {
+    const paneElement = document.querySelector('.react-flow__pane');
+    paneElement.addEventListener('mousedown', handlePaneClick);
+
+    return () => {
+      paneElement.removeEventListener('mousedown', handlePaneClick);
+    };
+  }, [handlePaneClick]);
+
   return (
     <div
       style={{ width: '100%', height: '100vh', position: 'relative' }}