Use real token and minor fixes (#929)

Signed-off-by: bmyrcha <[email protected]>

neural_insights/README.md

@@ -107,4 +107,3 @@ q_model = fit(
 ## Research Collaborations
 
 Welcome to raise any interesting research ideas on model compression techniques and feel free to reach us ([email protected]). Look forward to our collaborations on Neural Insights!
-

neural_insights/gui/src/App.js

@@ -22,6 +22,7 @@ export const api = axios.create({
 
 function App() {
   document.body.style = 'background: #ececec;';
+  localStorage.setItem('token', window.location.search.replace('?token=', ''));
   return (
     <div className="App">
       <Diagnosis />

neural_insights/gui/src/components/Graph/Graph.js

@@ -36,7 +36,7 @@ export default function Graph({ setSelectedNode, selectedWorkload, selectedOp, s
         path: [selectedWorkload.model_path],
         ...((groupNode.length || groupNodeOpList.length) && { group: [...groupNode, ...groupNodeOpList] })
       };
-      api.post('api/model/graph?token=asd', payload)
+      api.post('api/model/graph?token=' + localStorage.getItem('token'), payload)
         .then(
           response => {
             setGraph(response.data);
@@ -49,7 +49,7 @@ export default function Graph({ setSelectedNode, selectedWorkload, selectedOp, s
 
   useEffect(() => {
     if (selectedOp) {
-      api.post('api/model/graph/highlight_pattern?token=asd', {
+      api.post('api/model/graph/highlight_pattern?token=' + localStorage.getItem('token'), {
         workload_id: selectedWorkload.uuid,
         path: [selectedWorkload.model_path],
         op_name: selectedOp,

neural_insights/gui/src/components/Histogram/Histogram.js

@@ -24,7 +24,7 @@ function Histogram({ selectedWorkload, selectedOp, histogramType, setWarningText
   useEffect(() => {
     if (selectedOp.length && histogramType.length) {
       setHistogramData(null);
-      api.post('api/diagnosis/histogram?token=asd', { workload_id: selectedWorkload.uuid, op_name: selectedOp, type: histogramType })
+      api.post('api/diagnosis/histogram?token=' + localStorage.getItem('token'), { workload_id: selectedWorkload.uuid, op_name: selectedOp, type: histogramType })
         .then(
           response => {
             setHistogramData(response.data);

neural_insights/gui/src/components/OpDetails/OpDetails.js

@@ -35,7 +35,7 @@ export default function OpDetails({ selectedWorkload, selectedOp, setHistogramTy
 
   useEffect(() => {
     if (selectedOp?.length) {
-      api.post('api/diagnosis/op_details?token=asd', { workload_id: selectedWorkload.uuid, op_name: selectedOp })
+      api.post('api/diagnosis/op_details?token=' + localStorage.getItem('token'), { workload_id: selectedWorkload.uuid, op_name: selectedOp })
         .then(
           response => {
             setOpDetails(response.data);

neural_insights/gui/src/components/OpList/OpList.js

@@ -21,13 +21,16 @@ export default function OpList({ selectedWorkload, setSelectedOp, selectedOp, se
 
   useEffect(() => {
     if (selectedWorkload) {
-      api.post('api/diagnosis/op_list?token=asd', { workload_id: selectedWorkload.uuid })
+      setOpList([]);
+      api.post('api/diagnosis/op_list?token=' + localStorage.getItem('token'), { workload_id: selectedWorkload.uuid })
         .then(
           response => {
             setOpList(response.data);
           })
         .catch(error => {
-          setWarningText(error.message + ': ' + error?.response?.data);
+          if (selectedWorkload?.status !== 'wip') {
+            setWarningText(error.message + ': ' + error?.response?.data);
+          }
         });
     }
   }, [selectedWorkload, selectedOp]);
@@ -53,19 +56,21 @@ export default function OpList({ selectedWorkload, setSelectedOp, selectedOp, se
 
   return (
     <div className="overflow-table">
-      <Table className="rounded" hover>
-        <thead>
-          <tr>
-            <th className="header center">OP Name</th>
-            <th className="header center">MSE</th>
-            <th className="header center">Activation Min</th>
-            <th className="header center">Activation Max</th>
-          </tr>
-        </thead>
-        <tbody>
-          {tableContent}
-        </tbody>
-      </Table>
+      {opList.length > 0 &&
+        <Table className="rounded" hover>
+          <thead>
+            <tr>
+              <th className="header center">OP Name</th>
+              <th className="header center">MSE</th>
+              <th className="header center">Activation Min</th>
+              <th className="header center">Activation Max</th>
+            </tr>
+          </thead>
+          <tbody>
+            {tableContent}
+          </tbody>
+        </Table>
+      }
     </div>
   );
 }

neural_insights/gui/src/components/Profiling/Profiling.js

@@ -37,7 +37,7 @@ function ProfilingTable({ selectedWorkload, profilingTable, setProfilingTable, s
 
   useEffect(() => {
     if (selectedWorkload) {
-      api.post('api/profiling?token=asd', { workload_id: selectedWorkload.uuid })
+      api.post('api/profiling?token=' + localStorage.getItem('token'), { workload_id: selectedWorkload.uuid })
         .then(
           response => {
             setProfilingTable(response.data);

neural_insights/gui/src/components/Workloads/Workloads.js

@@ -27,18 +27,20 @@ export default function Workloads({ setSelectedWorkload, selectedWorkload, setWa
 
   let socket = io('/');
   socket.on('Config update', data => {
-    getWorkloads();
+    getWorkloads(false);
   });
 
   useEffect(() => {
-    getWorkloads();
+    getWorkloads(true);
   }, []);
 
-  let getWorkloads = () => {
-    api.get('api/workloads?token=asd')
+  let getWorkloads = (changeSelectedWorkload) => {
+    api.get('api/workloads?token=' + localStorage.getItem('token'))
       .then(
         response => {
-          setSelectedWorkload(response.data.workloads[0]);
+          if (changeSelectedWorkload) {
+            setSelectedWorkload(response.data.workloads[0]);
+          }
           setWorkloads(response.data.workloads);
           setSpinner(false);
         }

neural_insights/web/app/asset-manifest.json

@@ -1,7 +1,7 @@
 {
   "files": {
     "main.css": "/static/css/main.570a647a.css",
-    "main.js": "/static/js/main.f6ad24e2.js",
+    "main.js": "/static/js/main.38959c1a.js",
     "static/js/787.c1112931.chunk.js": "/static/js/787.c1112931.chunk.js",
     "static/media/IntelClear_Rg.ttf": "/static/media/IntelClear_Rg.33af11200cffaf9540ff.ttf",
     "static/media/IntelClear_Lt.ttf": "/static/media/IntelClear_Lt.c5e18e9d5505364da760.ttf",
@@ -11,11 +11,11 @@
     "static/media/intelone-display-regular.ttf": "/static/media/intelone-display-regular.0f8c3ef25c545acb6b7c.ttf",
     "index.html": "/index.html",
     "main.570a647a.css.map": "/static/css/main.570a647a.css.map",
-    "main.f6ad24e2.js.map": "/static/js/main.f6ad24e2.js.map",
+    "main.38959c1a.js.map": "/static/js/main.38959c1a.js.map",
     "787.c1112931.chunk.js.map": "/static/js/787.c1112931.chunk.js.map"
   },
   "entrypoints": [
     "static/css/main.570a647a.css",
-    "static/js/main.f6ad24e2.js"
+    "static/js/main.38959c1a.js"
   ]
 }

neural_insights/web/app/index.html

@@ -1 +1 @@
-<!doctype html><html lang="en"><head><meta charset="utf-8"/><link rel="icon" href="/favicon.ico"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><meta name="description" content="Web site created using create-react-app"/><title>Intel Neural Insights</title><script defer="defer" src="/static/js/main.f6ad24e2.js"></script><link href="/static/css/main.570a647a.css" rel="stylesheet"></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>
+<!doctype html><html lang="en"><head><meta charset="utf-8"/><link rel="icon" href="/favicon.ico"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><meta name="description" content="Web site created using create-react-app"/><title>Neural Insights</title><script defer="defer" src="/static/js/main.38959c1a.js"></script><link href="/static/css/main.570a647a.css" rel="stylesheet"></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>

neural_insights/web/configuration.py

@@ -18,13 +18,14 @@
 import argparse
 import logging
 import os
+import secrets
 import socket
 import sys
 from typing import Dict
 
+from neural_compressor.utils.utility import singleton
 from numpy.random import randint
 
-from neural_compressor.utils.utility import singleton
 from neural_insights.utils.consts import WORKDIR_LOCATION
 from neural_insights.utils.exceptions import NotFoundException
 from neural_insights.utils.logger import log
@@ -65,7 +66,7 @@ def determine_values_from_environment(self) -> None:
         self.url_prefix = self.determine_url_prefix(args)
         self.gui_port = self.determine_gui_port(args)
         self.log_level = self.determine_log_level(args)
-        self.token = "asd"  # secrets.token_hex(16)
+        self.token = secrets.token_hex(16)
         self.allow_insecure_connections = args.get("allow_insecure_connections", False)
         self.tls_certificate = args.get("cert", "")
         self.tls_key = args.get("key", "")