From c4f075343f557f278b1bacb4b92891e646f8fb2a Mon Sep 17 00:00:00 2001
From: Michael Foiani <sotech117@michaels-mbp-3.lan>
Date: Tue, 20 Apr 2021 07:55:53 -0400
Subject: Finished up the front end functionality. Some issues with the backend
 (small NaN issues), and some elements need a little more love (styling)
 before demo.

---
 react-frontend/src/components/Visualization.js | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

(limited to 'react-frontend/src/components/Visualization.js')

diff --git a/react-frontend/src/components/Visualization.js b/react-frontend/src/components/Visualization.js
index d374738..96b2027 100644
--- a/react-frontend/src/components/Visualization.js
+++ b/react-frontend/src/components/Visualization.js
@@ -1,6 +1,5 @@
 // JS module imports
-import { useCallback, useEffect, useMemo, useRef, useState } from "react";
-import uuid from 'react-uuid';
+import { useMemo, useState } from "react";
 import Graph from 'vis-react';
 
 // CSS imports
@@ -13,6 +12,8 @@ import '../css/Canvas.css';
  * @returns {import("react").HtmlHTMLAttributes} The canvas to be retured.
  */
 function Visualization(props) {
+  const [key, setKey] = useState(0);
+
   const [graphState, setGraphState] = useState({
     nodes: [],
     edges: []
@@ -27,8 +28,7 @@ function Visualization(props) {
 
   const events = {
     selectNode: event => {
-      console.log(event);
-      //props.setSelectedId(event.nodes[0]);
+      props.setSelectedId(event.nodes[0])
     }
   }
 
@@ -95,12 +95,16 @@ function Visualization(props) {
   }
 
   // Hooks to update graph state when data changes
-  useMemo(() => setGraphState({nodes: getNodes(), edges: getEdges()}), [props.data]);
+  useMemo(() => {
+    setKey(key => key + 1);
+    setGraphState({nodes: getNodes(), edges: getEdges()})
+  }, [JSON.stringify(props.data)]);
+  
 
   return (
     <div className="Map-canvas">
       <Graph
-        key={uuid()} 
+        key={key}
         graph={graphState}
         options={options}
         events={events}>
-- 
cgit v1.2.3-70-g09d2