adapt get-whishper for gpu support

get-whishper.sh

@@ -1,78 +1,121 @@
 #!/bin/bash
 set -e
 
+# Color codes
+RED='\033[0;31m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m' # No Color
+
 # Check if docker is installed
 if ! [ -x "$(command -v docker)" ]; then
-    echo "Docker is not installed. Please install docker first."
-    echo "Run official docker installation script? (y/n)"
-    echo "----> Check out the script: https://get.docker.com"
-    echo "----> I will run: 'curl -fsSL https://get.docker.com | sudo sh'"
+    echo -e "${RED}❌ Docker is not installed. Please install docker first.${NC}"
+    echo -e "> Run official docker installation script? (y/n)"
+    echo -e "---> Check out the script: https://get.docker.com"
+    echo -e "---> I will run: 'curl -fsSL https://get.docker.com | sudo sh'"
     read answer
     if [ "$answer" != "${answer#[Yy]}" ] ;then
-        echo "Running official docker installation script"
+        echo -e "${BLUE}ℹ️  Running official docker installation script${NC}"
         curl -fsSL https://get.docker.com | sudo sh
     else
-        echo "Aborting installation"
+        echo -e "${RED}❌ Aborting installation${NC}"
         exit 1
     fi
 fi
 
 # Ask if user wants to get everything in the current directory or in a new directory
-echo "Do you want to set up everything in the current directory? (y/n)"
+echo -e "${BLUE}> Do you want to set up everything in the current directory? (y/n)${NC}"
 read answer
 if [ "$answer" != "${answer#[Yy]}" ] ;then
-    echo "Setting up everything in the current directory"
+    echo -e "ℹ️  Setting up everything in the current directory"
 else
-    echo "Enter the name of the directory where you want to set up everything: "
+    echo -e "${BLUE}> Enter the name of the directory where you want to set up everything:${NC}"
     read directory
-    echo "----> Setting up everything in the $directory directory"
+    echo -e "ℹ️  Setting up everything in the $directory directory"
     mkdir $directory
     cd $directory
 fi
 
-echo "Getting the docker-compose.yml file from Github"
-curl -o docker-compose.yml https://raw.githubusercontent.com/pluja/whishper/main/docker-compose.yml
+echo ""
+echo -e "ℹ️  For more info about GPU support check out: https://whishper.net/guides/gpu/"
+echo -e "${BLUE}>  Do you want to use the GPU version? (y/n)${NC}"
+read gpu
+if [ "$gpu" != "${gpu#[Yy]}" ] ;then
+    gpu=true
+    if ! [ -x "$(command -v nvidia-container-toolkit)" ]; then
+        echo ""
+        echo -e "${YELLOW}⚠️  WARNING: nvidia-container-toolkit seems to not be installed. You must install it for whishper to work with GPU. Read more at: https://whishper.net/guides/gpu/${NC}"
+        echo -e "${BLUE}>  Do you want to continue anyways? (y/n)"
+        read continue
+        if [ "$gpu" != "${gpu#[Yy]}" ] ;then
+            echo -e "ℹ️  Continuing..."
+        else
+            echo -e "${RED}❌ Aborting installation${NC}"
+            exit 1
+        fi
+    fi
+else
+    gpu=false
+fi
 
-echo "Getting the nginx.conf file from Github"
-curl -o nginx.conf https://raw.githubusercontent.com/pluja/whishper/main/nginx.conf
+echo ""
+echo -e "ℹ️  Getting the docker-compose.yml file from Github"
+curl -o docker-compose.yml https://raw.githubusercontent.com/pluja/whishper/main/docker-compose.yml > /dev/null 2>&1
+sleep 1
+
+echo -e "ℹ️  Getting the nginx.conf file from Github"
+curl -o nginx.conf https://raw.githubusercontent.com/pluja/whishper/main/nginx.conf > /dev/null 2>&1
+sleep 1
 
-# Copy env.example to .env
 # check if .env exists
 if [ -f .env ]; then
-    echo ".env file already exists"
-    echo "Do you want to overwrite it? (y/n)"
+    echo ""
+    echo -e "${YELLOW}⚠️  .env file already exists${NC}"
+    echo -e "${BLUE}> Do you want to overwrite it? (y/n)"
     read answer
     if [ "$answer" != "${answer#[Yy]}" ] ;then
-        echo "Copying env.example to .env"
-        cp env.example .env
+        echo -e "ℹ️  Copying env.example to .env"
+        cp .env .env.backup
+        curl -o .env https://raw.githubusercontent.com/pluja/whishper/main/example.env > /dev/null 2>&1
+        sleep 1
     fi
+    echo ""
 else
-    echo "Getting the default .env file from Github"
-    curl -o .env https://raw.githubusercontent.com/pluja/whishper/main/example.env
+    echo -e "ℹ️  Getting the default .env file from Github"
+    curl -o .env https://raw.githubusercontent.com/pluja/whishper/main/example.env > /dev/null 2>&1
+    sleep 1
+fi
+
+if [ "$gpu" = true ] ;then
+    sed -i 's/COMPOSE_PROFILES=cpu/COMPOSE_PROFILES=gpu/g' .env
 fi
 
 # Create necessary directories for libretranslate
-echo "Creating necessary directories for libretranslate"
+echo -e "ℹ️  Creating necessary directories for libretranslate"
 sudo mkdir -p ./whishper_data/libretranslate/{data,cache}
+sleep 1
 
 # This permissions are for libretranslate docker container
-echo "Setting permissions for libretranslate"
+echo -e "ℹ️  Setting permissions for libretranslate"
 case "$OSTYPE" in
-  darwin*)  echo "macOS detected... Leaving permissions as is." ;; 
+  darwin*)  echo -e "ℹ️  macOS detected... Leaving permissions as is." ;; 
   linux*)   sudo chown -R 1032:1032 ./whishper_data/libretranslate ;;
-  *)        echo "unknown: $OSTYPE" ;;
+  *)        echo -e "${YELLOW}⚠️  unknown: $OSTYPE${NC}" ;;
 esac
+sleep 1
 
-echo "Do you want to pull docker images? (y/n)"
+echo ""
+echo -e "${BLUE}>  Do you want to pull the docker images? (y/n)"
 read answer
 if [ "$answer" != "${answer#[Yy]}" ] ;then
-    echo "Pulling and building docker images"
+    echo -e "ℹ️  Pulling and building docker images"
     sudo docker compose pull
 fi
 
-echo "Do you want to start the containers? (y/n)"
+echo ""
+echo -e "${BLUE}>  Do you want to start the containers now? (y/n)"
 read answer
 if [ "$answer" != "${answer#[Yy]}" ] ;then
-    echo "Starting whishper..."
+    echo -e "ℹ️  Starting whishper..."
     sudo docker compose up -d
 fi