Reenable search functionality for GraphRAG v1.0.0 (#32)

noworneverev · Dec 27, 2024 · eb6ab21 · eb6ab21
1 parent 343e7ca
commit eb6ab21
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/app/components/GraphViewer.tsx b/src/app/components/GraphViewer.tsx
@@ -21,6 +21,7 @@ import {
 import FullscreenIcon from "@mui/icons-material/Fullscreen";
 import FullscreenExitIcon from "@mui/icons-material/FullscreenExit";
 import SearchIcon from "@mui/icons-material/Search";
+import DeleteIcon from "@mui/icons-material/Delete";
 import Fuse from "fuse.js";
 import {
   CSS2DRenderer,
@@ -524,10 +525,10 @@ const GraphViewer: React.FC<GraphViewerProps> = ({
     ? includeTextUnits && includeCommunities && includeCovariates
     : includeTextUnits && includeCommunities;
 
-  // const clearSearchResults = () => {
-  //   setGraphData(initialGraphData.current);
-  //   setApiSearchResults(null);
-  // };
+  const clearSearchResults = () => {
+    setGraphData(initialGraphData.current);
+    setApiSearchResults(null);
+  };
 
   return (
     <Box
@@ -880,7 +881,7 @@ const GraphViewer: React.FC<GraphViewerProps> = ({
       >
         <Typography variant="body2">Nodes: {nodeCount}</Typography>
         <Typography variant="body2">Relationships: {linkCount}</Typography>
-        {/* <Button
+        <Button
           variant="contained"
           onClick={toggleApiDrawer(true)}
           startIcon={<SearchIcon />}
@@ -895,7 +896,7 @@ const GraphViewer: React.FC<GraphViewerProps> = ({
           disabled={apiSearchResults === null}
         >
           Clear Query Results
-        </Button> */}
+        </Button>
       </Box>
     </Box>
   );