From 7ef66dd000445bca5944fa03eb3d31096cb10876 Mon Sep 17 00:00:00 2001
From: BenAAndrew <bandrew01@qub.ac.uk>
Date: Sun, 8 Aug 2021 21:43:24 +0100
Subject: [PATCH] Add install script

---
 application/utils.py        |  1 +
 build_exe.py                | 34 ++++++++++++++++++++++++++++++++++
 dataset/transcribe.py       |  1 +
 extra-hooks/hook-resampy.py |  2 ++
 4 files changed, 38 insertions(+)
 create mode 100644 build_exe.py
 create mode 100644 extra-hooks/hook-resampy.py

diff --git a/application/utils.py b/application/utils.py
index 53fb529..b52dd6e 100644
--- a/application/utils.py
+++ b/application/utils.py
@@ -5,6 +5,7 @@ import traceback
 import shutil
 import zipfile
 import librosa
+import resampy  # noqa
 
 from main import socketio
 from dataset.audio_processing import convert_audio
diff --git a/build_exe.py b/build_exe.py
new file mode 100644
index 0000000..f79f6d0
--- /dev/null
+++ b/build_exe.py
@@ -0,0 +1,34 @@
+import os
+
+
+if __name__ == "__main__":
+    import PyInstaller.__main__
+    from PyInstaller.utils.hooks import collect_data_files
+
+    site_packages_folder = "D:\Program Files\\anaconda3\envs\pyinstaller11\Lib\site-packages"
+    deepspeech_dll = os.path.join(site_packages_folder, "deepspeech", "lib", "libdeepspeech.so")
+    llvmlite_dll = os.path.join(site_packages_folder, "llvmlite", "binding", "llvmlite.dll")
+    kaiser_filters = os.path.join(site_packages_folder, "resampy", "data")
+    pytorch_libs = os.path.join(site_packages_folder, "torch", "lib")
+
+    PyInstaller.__main__.run([
+        "main.py",
+        "--onefile",
+        "--clean",
+        "--hidden-import=sklearn.utils._weight_vector",
+        "--hidden-import=sklearn.utils._cython_blas",
+        "--hidden-import=sklearn.neighbors._typedefs",
+        "--hidden-import=sklearn.neighbors._quad_tree",
+        "--hidden-import=sklearn.tree._utils",
+        "--hidden-import=sklearn.tree",
+        "--hidden-import=scipy.special.cython_special",
+        "--icon=application\static\\favicon\\app-icon.ico", 
+        "--add-data=application/static;application/static",
+        "--add-data",
+        deepspeech_dll+";.",
+        "--add-data",
+        llvmlite_dll+";.",
+        "--paths",
+        pytorch_libs,
+        "--additional-hooks=extra-hooks",
+    ])
diff --git a/dataset/transcribe.py b/dataset/transcribe.py
index 1e24836..07dac00 100644
--- a/dataset/transcribe.py
+++ b/dataset/transcribe.py
@@ -8,6 +8,7 @@ import numpy as np
 import deepspeech
 import torch
 import torchaudio  # noqa
+import soundfile  # noqa
 import omegaconf  # noqa
 
 
diff --git a/extra-hooks/hook-resampy.py b/extra-hooks/hook-resampy.py
new file mode 100644
index 0000000..b00a58d
--- /dev/null
+++ b/extra-hooks/hook-resampy.py
@@ -0,0 +1,2 @@
+from PyInstaller.utils.hooks import collect_data_files
+datas = collect_data_files('resampy')
\ No newline at end of file