From 21778b22c7922b9c3f0a37744e8fa39af1f8bf06 Mon Sep 17 00:00:00 2001
From: NAmorim <na.amorim@gmail.com>
Date: Tue, 16 Nov 2021 19:45:36 +0000
Subject: [PATCH] Contribution: Enable Tensorflow (Learning Tasks)

---
 package.json |  5 +++--
 tf_setup.js  | 16 ++++++++++++++++
 2 files changed, 19 insertions(+), 2 deletions(-)
 create mode 100644 tf_setup.js

diff --git a/package.json b/package.json
index c8bd8b767..95eb16937 100644
--- a/package.json
+++ b/package.json
@@ -10,7 +10,7 @@
   "scripts": {
     "start": "node platform minMemo noBrowser",
     "presetup": "git checkout develop && git pull upstream develop",
-    "setup": "node setup noShortcuts",
+    "setup": "node setup noShortcuts && node tf_setup.js",
     "postsetup": "npm start",
     "dist": "electron-builder -c build/electron-builder.json",
     "postinstall": "electron-builder install-app-deps"
@@ -35,7 +35,8 @@
     "twitter": "^1.7.1",
     "util": "^0.12.4",
     "web3": "^1.6.0",
-    "ws": "^8.2.3"
+    "ws": "^8.2.3",
+    "@tensorflow/tfjs-node": "^2.8.6"
   },
   "devDependencies": {
     "electron": "^15.1.0",
diff --git a/tf_setup.js b/tf_setup.js
new file mode 100644
index 000000000..f3dc1607a
--- /dev/null
+++ b/tf_setup.js
@@ -0,0 +1,16 @@
+const fs = require('fs')
+const path = require('path')
+
+const srcFolder = './node_modules/@tensorflow/tfjs-node/deps/lib/'
+const destFolder = './node_modules/@tensorflow/tfjs-node/lib/'
+
+const dest = fs.readdirSync(destFolder, { withFileTypes: true })
+.filter(dirent => dirent.isDirectory())
+.map(dirent => dirent.name)
+
+if(process.platform === 'win32') {
+    fs.copyFile(path.join(srcFolder, "tensorflow.dll"), path.join(destFolder, dest[0], "tensorflow.dll"), (err) => {
+        if (err) throw err
+        console.log("tensorflow dll copied")
+    })
+}