From 2c244c49ecf3074e19f064e687a8a73771778913 Mon Sep 17 00:00:00 2001
From: lightmare <lightmare@noreply.localhost>
Date: Mon, 6 Mar 2023 21:16:26 +0000
Subject: [PATCH] Added experimental guided setup

---
 requirements.txt |  3 ---
 setup-guided.sh  | 54 ++++++++++++++++++++++++++++++++++++++++++++++++
 start.sh         |  2 ++
 3 files changed, 56 insertions(+), 3 deletions(-)
 create mode 100755 setup-guided.sh

diff --git a/requirements.txt b/requirements.txt
index 5d14006..023e464 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,3 @@
-git+https://github.com/openai/whisper.git
-git+https://github.com/m-bain/whisperx.git
-
 more-itertools
 ffmpeg-python
 gradio
diff --git a/setup-guided.sh b/setup-guided.sh
new file mode 100755
index 0000000..4b33eb3
--- /dev/null
+++ b/setup-guided.sh
@@ -0,0 +1,54 @@
+#!/bin/bash
+
+if ! command -v git &> /dev/null; then
+	if [[ "$(read -e -p 'Could not find git. Continue? [y/N]> '; echo $REPLY)" != [Yy]* ]]; then exit 1; fi
+else
+	printf "git - ok\n"
+fi
+
+# TODO: This could be more precise. e.g. checking for python3, then python, etc.
+if ! command -v python &> /dev/null; then
+	if [[ "$(read -e -p 'Could not find python. Continue? [y/N]> '; echo $REPLY)" != [Yy]* ]]; then exit 1; fi
+else
+	python -c 'import sys; sys.stderr.write("Wrong python version.\n") if sys.version_info.major != 3 else sys.stderr.write("Python 3 - ok\n")'
+fi
+
+printf "Which GPU brand do you have?\n"
+gpus=(nvidia amd)
+
+gpu=""
+while [ "$gpu" = "" ]; do
+	select gpu in $(printf '%s\n' ${gpus[@]}); do break; done
+done
+
+if [ $gpu = "nvidia" ]; then
+	./setup-cuda.sh
+elif [ $gpu = "amd" ]; then
+	./setup-rocm.sh
+fi
+
+source ./venv/bin/activate
+
+printf "Which Whisper backend would you like to use?\n"
+whisper_backends=("openai/whisper" "m-bain/whisperx" "lightmare/whispercpp")
+
+whisper_backend=""
+while [ "$whisper_backend" = "" ]; do
+	select whisper_backend in $(printf '%s\n' ${whisper_backends[@]}); do break; done
+done
+
+if [ $whisper_backend = "openai/whisper" ]; then
+	python -m pip install git+https://github.com/openai/whisper.git
+elif [ $whisper_backend = "m-bain/whisperx" ]; then
+	python -m pip install git+https://github.com/m-bain/whisperx.git
+elif [ $whisper_backend = "lightmare/whispercpp" ]; then
+	# This depends on SemVer
+	# Git > v2.18 for `--sort`
+	# Git > v2.4 for `versionsort.suffix`
+	# For older versions:
+	# git ls-remote --refs --tags https://git.ecker.tech/lightmare/whispercpp.py | cut --delimiter='/' --fields=3 | tr '-' '~' | sort --version-sort | tail --lines=1
+	WHISPERCPP_LATEST=$(git -c 'versionsort.suffix=-' ls-remote --exit-code --refs --sort='version:refname' --tags https://git.ecker.tech/lightmare/whispercpp.py '*.*.*' | tail -n 1 | cut --delimiter='/' --fields=3)
+	python -m pip install git+https://git.ecker.tech/lightmare/whispercpp.py@$WHISPERCPP_LATEST
+fi
+
+deactivate
diff --git a/start.sh b/start.sh
index f1f9947..50173fb 100755
--- a/start.sh
+++ b/start.sh
@@ -1,4 +1,6 @@
 #!/bin/bash
+if [ ! -d "venv" ]; then ./setup-guided.sh; fi
+
 source ./venv/bin/activate
 python3 ./src/main.py "$@"
 deactivate