SillyTavern/SillyTavern-Launcher
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Launcher.git synced 2026-01-26 17:09:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0944e127d8ee411a0f0f8782e1f223592f0c0bcd
SillyTavern-Launcher/install.sh
deffcolony 240badb9c9 update STL for Linux&Mac 
Update STL to make it stable and useable & bring most features from the current Windows version. Features from app installer are work in progress
2025-02-24 01:11:14 +01:00

1158 lines
44 KiB
Bash
Executable File
Raw Blame History

#!/bin/bash
#
# SillyTavern Installer Script
# Created by: Deffcolony
#
# Description:
# This script installs SillyTavern and/or Extras to your Linux system.
#
# Usage:
# chmod +x install.sh && ./install.sh
#
# In automated environments, you may want to run as root.
# If using curl, we recommend using the -fsSL flags.
#
# This script is intended for use on Linux systems. Please
# report any issues or bugs on the GitHub repository.
#
# GitHub: https://github.com/SillyTavern/SillyTavern-Launcher
# Issues: https://github.com/SillyTavern/SillyTavern-Launcher/issues
# ----------------------------------------------------------
# Note: Modify the script as needed to fit your requirements.
# ----------------------------------------------------------
echo -e "\033]0;SillyTavern Installer\007"
# ANSI Escape Code for Colors
reset="\033[0m"
white_fg_strong="\033[90m"
red_fg_strong="\033[91m"
green_fg_strong="\033[92m"
yellow_fg_strong="\033[93m"
blue_fg_strong="\033[94m"
magenta_fg_strong="\033[95m"
cyan_fg_strong="\033[96m"
# Normal Background Colors
red_bg="\033[41m"
blue_bg="\033[44m"
yellow_bg="\033[43m"
# Function to find Miniconda installation directory
find_conda() {
local paths=(
"$HOME/miniconda3"
"$HOME/miniconda"
"/opt/miniconda3"
"/opt/miniconda"
"/usr/local/miniconda3"
"/usr/local/miniconda"
"/usr/miniconda3"
"/usr/miniconda"
"$HOME/anaconda3"
"$HOME/anaconda"
"/opt/anaconda3"
"/opt/anaconda"
"/usr/local/anaconda3"
"/usr/local/anaconda"
"/usr/anaconda3"
"/usr/anaconda"
)
for path in "${paths[@]}"; do
if [ -d "$path" ]; then
echo "$path"
return 0
fi
done
return 1
}
# Function to install Miniconda
install_miniconda() {
# Check if Miniconda is already installed
if command -v conda &>/dev/null; then
echo "Miniconda is already installed. Skipping installation."
installer
return 0 # Exit the function with success status
fi
# Determine Miniconda installer based on OS and architecture
os_name="$(uname -s)"
arch="$(uname -m)"
case "$os_name" in
Linux)
case "$arch" in
x86_64) miniconda_installer="Miniconda3-latest-Linux-x86_64.sh" ;;
aarch64) miniconda_installer="Miniconda3-latest-Linux-aarch64.sh" ;;
*) echo "ERROR: Unsupported architecture: $arch" >&2; return 1 ;;
esac
;;
Darwin)
case "$arch" in
x86_64) miniconda_installer="Miniconda3-latest-MacOSX-x86_64.sh" ;;
arm64) miniconda_installer="Miniconda3-latest-MacOSX-arm64.sh" ;;
*) echo "ERROR: Unsupported architecture: $arch" >&2; return 1 ;;
esac
;;
*)
echo "ERROR: Unsupported operating system: $os_name" >&2
return 1
;;
esac
# Download the Miniconda installer script
wget "https://repo.anaconda.com/miniconda/$miniconda_installer" -O /tmp/miniconda_installer.sh
# Run the installer script
bash /tmp/miniconda_installer.sh -b -p "$HOME/miniconda"
# Add Miniconda to PATH
export PATH="$HOME/miniconda/bin:$PATH"
# Activate Conda environment
source "$HOME/miniconda/etc/profile.d/conda.sh"
# Clean up the Downloaded file
rm /tmp/miniconda_installer.sh
echo "Miniconda installed successfully in $HOME/miniconda"
}
# Function to log messages with timestamps and colors
log_message() {
# This is only time
current_time=$(date +'%H:%M:%S')
# This is with date and time
# current_time=$(date +'%Y-%m-%d %H:%M:%S')
case "$1" in
"INFO")
echo -e "${blue_bg}[$current_time]${reset} ${blue_fg_strong}[INFO]${reset} $2"
;;
"WARN")
echo -e "${yellow_bg}[$current_time]${reset} ${yellow_fg_strong}[WARN]${reset} $2"
;;
"ERROR")
echo -e "${red_bg}[$current_time]${reset} ${red_fg_strong}[ERROR]${reset} $2"
;;
*)
echo -e "${blue_bg}[$current_time]${reset} ${blue_fg_strong}[DEBUG]${reset} $2"
;;
esac
}
boxDrawingText()
{
local string=$1
local maxwidth=$2
# empty string
local color=
if [ $# -eq 3 ]; then
color=$3
fi
local stringlength=${#string}
# stringlength < maxwidth ? maxwidth : stringlength
local width=$((stringlength < maxwidth ? maxwidth : stringlength))
local topL="╔"
local bottomL="╚"
local topR="╗"
local bottomR="╝"
local middle="║"
local space=" "
echo -e "$color$middle$string$(printf ' %.0s' $(seq 1 $((width - stringlength))))$middle"
}
# Log your messages test window
#log_message "INFO" "Something has been launched."
#log_message "WARN" "${yellow_fg_strong}Something is not installed on this system.${reset}"
#log_message "ERROR" "${red_fg_strong}An error occurred during the process.${reset}"
#log_message "DEBUG" "This is a debug message."
#read -p "Press Enter to continue..."
# Function to install Git
install_git() {
if ! command -v git &> /dev/null; then
log_message "WARN" "${yellow_fg_strong}Git is not installed on this system${reset}"
if command -v apt-get &>/dev/null; then
# Debian/Ubuntu-based system
log_message "INFO" "Installing Git using apt..."
sudo apt-get update
sudo apt-get install -y git
elif command -v yum &>/dev/null; then
# Red Hat/Fedora-based system
log_message "INFO" "Installing Git using yum..."
sudo yum install -y git
elif command -v apk &>/dev/null; then
# Alpine Linux-based system
log_message "INFO" "Installing Git using apk..."
sudo apk add git
elif command -v pacman &>/dev/null; then
# Arch Linux-based system
log_message "INFO" "Installing Git using pacman..."
sudo pacman -S --noconfirm git
elif command -v emerge &>/dev/null; then
# Gentoo Linux-based system
log_message "INFO" "Installing Git using emerge..."
sudo emerge --ask dev-vcs/git
elif command -v brew &>/dev/null; then
# macOS
log_message "INFO" "Installing Git using Homebrew..."
brew install git
elif command -v pkg &>/dev/null; then
# Termux on Android
log_message "INFO" "Installing Git using pkg..."
pkg install git
elif command -v zypper &>/dev/null; then
# openSUSE
log_message "INFO" "Installing Git using zypper..."
sudo zypper install git
else
log_message "ERROR" "${red_fg_strong}Unsupported Linux distribution.${reset}"
exit 1
fi
log_message "INFO" "${green_fg_strong}Git is installed.${reset}"
else
echo -e "${blue_fg_strong}[INFO] Git is already installed.${reset}"
fi
}
install_nodejs() {
if ! command -v node &> /dev/null; then
log_message "WARN" "${yellow_fg_strong}Node.js is not installed on this system${reset}"
case "$(uname -s)" in
Linux)
package_manager=$(detect_package_manager)
case "$package_manager" in
apt)
log_message "INFO" "Detected apt (Debian/Ubuntu). Installing Node.js..."
sudo apt update && sudo apt install -y nodejs npm
;;
dnf|yum)
log_message "INFO" "Detected dnf/yum (Red Hat/Fedora). Installing Node.js..."
sudo $package_manager install -y nodejs npm
;;
apk)
log_message "INFO" "Detected apk (Alpine). Installing Node.js..."
sudo apk add nodejs npm
;;
pacman)
log_message "INFO" "Detected pacman (Arch). Installing Node.js..."
sudo pacman -Sy --noconfirm nodejs npm
;;
emerge)
log_message "INFO" "Detected emerge (Gentoo). Installing Node.js..."
sudo emerge --ask nodejs npm
;;
zypper)
log_message "INFO" "Detected zypper (openSUSE). Installing Node.js..."
sudo zypper install -y nodejs npm
;;
xbps)
log_message "INFO" "Detected xbps (Void). Installing Node.js..."
sudo xbps-install -y nodejs npm
;;
nix)
log_message "INFO" "Detected nix (NixOS). Installing Node.js..."
nix-env -iA nixpkgs.nodejs
;;
guix)
log_message "INFO" "Detected guix (Guix). Installing Node.js..."
guix package -i node
;;
*)
log_message "ERROR" "${red_fg_strong}Unsupported package manager or distribution.${reset}"
exit 1
;;
esac
;;
Darwin)
log_message "INFO" "Detected macOS. Installing Node.js via Homebrew..."
if ! command -v brew &> /dev/null; then
log_message "ERROR" "${red_fg_strong}Homebrew not installed. Install it first: https://brew.sh/${reset}"
exit 1
fi
brew install node
;;
*)
log_message "ERROR" "${red_fg_strong}Unsupported operating system.${reset}"
exit 1
;;
esac
log_message "INFO" "${green_fg_strong}Node.js is installed.${reset}"
else
log_message "INFO" "${blue_fg_strong}Node.js is already installed.${reset}"
fi
}
# Function to install all options from the installer
install_all() {
echo -e "\033]0;SillyTavern [INSTALL-SILLYTAVERN-EXTRAS-XTTS]\007"
clear
echo -e "${blue_fg_strong}/ Installer / SillyTavern + Extras + XTTS${reset}"
echo "---------------------------------------------------------------"
echo ""
echo -e "${blue_bg}╔════ INSTALL SUMMARY ══════════════════════════════════════════════════════════════════════════╗${reset}"
echo -e "${blue_bg}║ You are about to install all options from the installer. ║${reset}"
echo -e "${blue_bg}║ This will include the following options: SillyTavern, SillyTavern-Extras and XTTS ║${reset}"
echo -e "${blue_bg}║ Below is a list of package requirements that will get installed: ║${reset}"
echo -e "${blue_bg}║ * SillyTavern [Size: 478 MB] ║${reset}"
echo -e "${blue_bg}║ * SillyTavern-extras [Size: 65 MB] ║${reset}"
echo -e "${blue_bg}║ * xtts [Size: 1.74 GB] ║${reset}"
echo -e "${blue_bg}║ * Miniconda3 [INSTALLED] [Size: 630 MB] ║${reset}"
echo -e "${blue_bg}║ * Miniconda3 env - xtts [Size: 6.98 GB] ║${reset}"
echo -e "${blue_bg}║ * Miniconda3 env - extras [Size: 9.98 GB] ║${reset}"
echo -e "${blue_bg}║ * Git [INSTALLED] [Size: 338 MB] ║${reset}"
echo -e "${blue_bg}║ * Node.js [Size: 87.5 MB] ║${reset}"
echo -e "${blue_bg}║ TOTAL INSTALL SIZE: 20.26 GB ║${reset}"
echo -e "${blue_bg}╚═══════════════════════════════════════════════════════════════════════════════════════════════╝${reset}"
echo ""
echo -n "Are you sure you want to proceed? [Y/N]: "
read confirmation
if [ "$confirmation" = "Y" ] || [ "$confirmation" = "y" ]; then
install_all_y
else
installer
fi
}
install_all_y() {
# Ask the user about the GPU
echo -e "What is your GPU?"
echo -e "1. NVIDIA"
echo -e "2. AMD"
echo -e "3. None (CPU-only mode)"
echo -e "0. Cancel install"
# Get GPU information
gpu_info=""
while IFS= read -r line; do
gpu_info+=" $line"
done < <(lspci | grep VGA | cut -d ':' -f3)
echo ""
echo -e "${blue_bg}╔════ GPU INFO ═════════════════════════════════════════════════════════════╗${reset}"
boxDrawingText "" 75 $blue_bg
boxDrawingText "* ${gpu_info:1}" 75 $blue_bg
boxDrawingText "" 75 $blue_bg
echo -e "${blue_bg}╚═══════════════════════════════════════════════════════════════════════════╝${reset}"
echo ""
# Prompt for GPU choice
read -p "Enter number corresponding to your GPU: " gpu_choice
# GPU menu - Backend
# Set the GPU choice in an environment variable for choice callback
export GPU_CHOICE=$gpu_choice
# Check the user's response
if [ "$gpu_choice" == "1" ]; then
log_message "INFO" "GPU choice set to NVIDIA"
install_all_pre
elif [ "$gpu_choice" == "2" ]; then
log_message "INFO" "GPU choice set to AMD"
install_all_pre
elif [ "$gpu_choice" == "3" ]; then
log_message "INFO" "Using CPU-only mode"
install_all_pre
elif [ "$gpu_choice" == "0" ]; then
installer
else
log_message "ERROR" "${red_fg_strong}Invalid number. Please enter a valid number.${reset}"
read -p "Press Enter to continue..."
install_all
fi
}
install_all_pre() {
log_message "INFO" "Installing SillyTavern + Extras + XTTS"
echo -e "${cyan_fg_strong}This may take a while. Please be patient.${reset}"
log_message "INFO" "Installing SillyTavern..."
git clone https://github.com/SillyTavern/SillyTavern.git
log_message "INFO" "${green_fg_strong}SillyTavern installed successfully.${reset}"
log_message "INFO" "Installing Extras..."
log_message "INFO" "Cloning SillyTavern-extras repository..."
git clone https://github.com/SillyTavern/SillyTavern-extras.git
# Install script for XTTS
log_message "INFO" "Installing XTTS..."
# Activate the Miniconda installation
log_message "INFO" "Activating Miniconda environment..."
source "$miniconda_path/bin/activate"
# Create a Conda environment named xtts
log_message "INFO" "Creating Conda environment: xtts"
conda create -n xtts python=3.10 git -y
# Activate the xtts environment
log_message "INFO" "Activating Conda environment: xtts"
source activate xtts
# Check if xtts activation was successful
if [ $? -eq 0 ]; then
log_message "INFO" "Successfully activated Conda environment: xtts"
install_all_pre_successful
else
log_message "ERROR" "${red_fg_strong}Failed to activate Conda environment: xtts${reset}"
log_message "INFO" "Press Enter to try again otherwise close the installer and restart."
read -p "Press Enter to try again..."
install_all_pre
fi
}
install_all_pre_successful() {
# Create folders for xtts
log_message "INFO" "Creating xtts folders..."
mkdir "$PWD/xtts"
mkdir "$PWD/xtts/speakers"
mkdir "$PWD/xtts/output"
# Clone the xtts-api-server repository for voice examples
log_message "INFO" "Cloning xtts-api-server repository..."
git clone https://github.com/daswer123/xtts-api-server.git
log_message "INFO" "Adding voice examples to speakers directory..."
cp -r "$PWD/xtts-api-server/example/"* "$PWD/xtts/speakers/"
log_message "INFO" "Removing the xtts-api-server directory..."
rm -rf "$PWD/xtts-api-server"
# Install pip3 requirements
log_message "INFO" "Installing pip3 requirements for xtts..."
pip3 install xtts-api-server
pip3 install pydub
pip3 install stream2sentence
# Use the GPU choice made earlier to install requirements for XTTS
if [ "$GPU_CHOICE" == "1" ]; then
log_message "INFO" "Installing NVIDIA version of PyTorch"
pip3 install torch==2.1.1+cu118 torchvision torchaudio==2.1.1+cu118 --index-url https://download.pytorch.org/whl/cu118
install_all_post
elif [ "$GPU_CHOICE" == "2" ]; then
log_message "INFO" "Installing AMD version of PyTorch"
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.6
install_all_post
elif [ "$GPU_CHOICE" == "3" ]; then
log_message "INFO" "Installing CPU-only version of PyTorch"
pip3 install torch torchvision torchaudio
install_all_post
fi
}
# End of install script for XTTS
install_all_post() {
# Create a Conda environment named extras
log_message "INFO" "Creating Conda environment: extras"
conda create -n extras python=3.11 git -y
log_message "INFO" "Activating Conda environment: extras"
conda activate extras
# Check if extras activation was successful
if [ $? -eq 0 ]; then
log_message "INFO" "Successfully activated Conda environment: extras"
install_all_post_successful
else
log_message "ERROR" "${red_fg_strong}Failed to activate Conda environment: extras${reset}"
log_message "INFO" "Press Enter to try again otherwise close the installer and restart."
read -p "Press Enter to try again..."
install_all_post
fi
}
install_all_post_successful() {
# Navigate to the SillyTavern-extras directory
cd "$PWD/SillyTavern-extras"
log_message "INFO" "Installing pip requirements from requirements-rvc.txt in Conda enviroment: ${cyan_fg_strong}extras${reset}"
pip3 install -r requirements-rvc.txt
pip3 install tensorboardX
# Use the GPU choice made earlier to install requirements for extras
if [ "$GPU_CHOICE" == "1" ]; then
log_message "INFO" "Installing modules for NVIDIA from requirements.txt in extras"
pip3 install -r requirements.txt
conda install -c conda-forge faiss-gpu -y
log_message "INFO" "${green_fg_strong}SillyTavern + Extras + XTTS successfully installed.${reset}"
install_all_final
elif [ "$GPU_CHOICE" == "2" ]; then
log_message "INFO" "Installing modules for AMD from requirements-rocm.txt in extras"
pip3 install -r requirements-rocm.txt
log_message "INFO" "${green_fg_strong}SillyTavern + Extras + XTTS successfully installed.${reset}"
install_all_final
elif [ "$GPU_CHOICE" == "3" ]; then
log_message "INFO" "Installing modules for CPU from requirements-silicon.txt in extras"
pip3 install -r requirements-silicon.txt
log_message "INFO" "${green_fg_strong}SillyTavern + Extras + XTTS successfully installed.${reset}"
install_all_final
fi
}
install_all_final() {
# Ask if the user wants to create a shortcut
read -p "Do you want to create a shortcut on the desktop? [Y/n] " create_shortcut
if [[ "${create_shortcut}" == "Y" || "${create_shortcut}" == "y" ]]; then
# Create the desktop shortcut
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Creating desktop shortcut..."
echo "[Desktop Entry]" > "$desktop_file"
echo "Version=1.0" >> "$desktop_file"
echo "Type=Application" >> "$desktop_file"
echo "Name=$script_name" >> "$desktop_file"
echo "Exec=$script_path" >> "$desktop_file"
echo "Icon=$icon_path" >> "$desktop_file"
echo "Terminal=true" >> "$desktop_file"
echo "Comment=SillyTavern Launcher" >> "$desktop_file"
chmod +x "$desktop_file"
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} ${green_fg_strong}Desktop shortcut created at: $desktop_file${reset}"
fi
# Ask if the user wants to start the launcher
read -p "Start the launcher now? [Y/n] " start_launcher
if [[ "${start_launcher}" == "Y" || "${start_launcher}" == "y" ]]; then
# Run the launcher
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Running launcher in a new window..."
cd "$(dirname "$0")"
chmod +x launcher.sh && ./launcher.sh
fi
installer
}
########################################################################################
########################################################################################
####################### SUPPORT MENU FUNCTIONS ########################################
########################################################################################
########################################################################################
# Functions for Support Menu - Backend
issue_report() {
if [ "$EUID" -eq 0 ]; then
log_message "ERROR" "${red_fg_strong}Cannot run xdg-open as root. Please run the script without root permission.${reset}"
else
if [ "$(uname -s)" == "Darwin" ]; then
open https://github.com/SillyTavern/SillyTavern-Launcher/issues/new/choose
else
xdg-open https://github.com/SillyTavern/SillyTavern-Launcher/issues/new/choose
fi
fi
read -p "Press Enter to continue..."
support
}
documentation() {
if [ "$EUID" -eq 0 ]; then
log_message "ERROR" "${red_fg_strong}Cannot run xdg-open as root. Please run the script without root permission.${reset}"
else
if [ "$(uname -s)" == "Darwin" ]; then
open https://docs.sillytavern.app/
else
xdg-open https://docs.sillytavern.app/
fi
fi
read -p "Press Enter to continue..."
support
}
discord() {
if [ "$EUID" -eq 0 ]; then
log_message "ERROR" "${red_fg_strong}Cannot run xdg-open as root. Please run the script without root permission.${reset}"
else
if [ "$(uname -s)" == "Darwin" ]; then
open https://discord.gg/sillytavern
else
xdg-open https://discord.gg/sillytavern
fi
fi
read -p "Press Enter to continue..."
support
}
############################################################
################# SUPPORT MENU - FRONTEND ##################
############################################################
support() {
echo -e "\033]0;SillyTavern [SUPPORT]\007"
clear
echo -e "${blue_fg_strong}| > / Installer / Support |${reset}"
echo -e "${blue_fg_strong}==============================================================${reset}"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| What would you like to do? |${reset}"
echo " 1. I want to report an issue"
echo " 2. Documentation"
echo " 3. Discord"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| Menu Options: |${reset}"
echo " 0. Back"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| |${reset}"
read -p " Choose Your Destiny: " support_choice
# Support Menu - Backend
case $support_choice in
1) issue_report ;;
2) documentation ;;
3) discord ;;
0) installer ;;
*) echo -e "${yellow_fg_strong}WARNING: Invalid number. Please insert a valid number.${reset}"
read -p "Press Enter to continue..."
support ;;
esac
}
########################################################################################
########################################################################################
####################### INSTALLER MENU FUNCTIONS ######################################
########################################################################################
########################################################################################
# Function to install SillyTavern
install_sillytavern() {
echo -e "\033]0;SillyTavern [INSTALL-ST]\007"
clear
echo -e "${blue_fg_strong}/ Installer / SillyTavern${reset}"
echo "---------------------------------------------------------------"
echo -e "${cyan_fg_strong}This may take a while. Please be patient.${reset}"
log_message "INFO" "Installing SillyTavern..."
log_message "INFO" "Cloning SillyTavern repository..."
git clone https://github.com/SillyTavern/SillyTavern.git
log_message "INFO" "${green_fg_strong}SillyTavern installed successfully.${reset}"
# Ask if the user wants to create desktop shortcuts
read -p "Do you want to create desktop shortcuts? [Y/n] " create_shortcut
if [[ "${create_shortcut}" =~ ^[Yy]$ ]]; then
create_desktop_shortcuts
fi
# Ask if the user wants to start the launcher
read -p "Start the launcher now? [Y/n] " start_launcher
if [[ "${start_launcher}" == "Y" || "${start_launcher}" == "y" ]]; then
# Run the launcher
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Running launcher in a new window..."
cd "$(dirname "$0")"
chmod +x launcher.sh && ./launcher.sh
fi
installer
}
create_desktop_shortcuts() {
# Common variables
stl_root="$(dirname "$(realpath "$0")")"
st_install_path="$stl_root/SillyTavern"
desktop_dir=""
launcher_script="$stl_root/launcher.sh"
sillytavern_script="$st_install_path/start.sh"
icon_launcher="$stl_root/st-launcher.ico"
icon_sillytavern="$stl_root/st.ico"
# Detect desktop environment
case "$(uname -s)" in
Linux)
desktop_dir="$HOME/Desktop"
# Create directory if it doesn't exist
mkdir -p "$desktop_dir"
# Create launcher shortcut
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Creating ST-Launcher shortcut..."
cat << EOF > "$desktop_dir/ST-Launcher.desktop"
[Desktop Entry]
Version=1.0
Type=Application
Name=SillyTavern Launcher
Exec="$launcher_script"
Icon=$icon_launcher
Terminal=true
Comment=Launch SillyTavern AI Interface
Path=$stl_root
EOF
chmod +x "$desktop_dir/ST-Launcher.desktop"
# Create SillyTavern direct start shortcut
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Creating SillyTavern shortcut..."
cat << EOF > "$desktop_dir/SillyTavern.desktop"
[Desktop Entry]
Version=1.0
Type=Application
Name=SillyTavern
Exec="$sillytavern_script"
Icon=$icon_sillytavern
Terminal=true
Comment=Start SillyTavern Directly
Path=$st_install_path
EOF
chmod +x "$desktop_dir/SillyTavern.desktop"
;;
Darwin)
desktop_dir="$HOME/Desktop"
# Create launcher shortcut
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Creating ST-Launcher shortcut..."
cat << EOF > "$desktop_dir/ST-Launcher.command"
#!/bin/zsh
cd "$stl_root" && ./launcher.sh
EOF
chmod +x "$desktop_dir/ST-Launcher.command"
# Create SillyTavern direct start shortcut
echo -e "${blue_bg}[$(date +%T)]${reset} ${blue_fg_strong}[INFO]${reset} Creating SillyTavern shortcut..."
cat << EOF > "$desktop_dir/SillyTavern.command"
#!/bin/zsh
cd "$st_install_path" && ./start.sh
EOF
chmod +x "$desktop_dir/SillyTavern.command"
# Set custom icons (requires osascript)
if [ -f "$icon_launcher" ]; then
osascript << EOF
set iconPath to POSIX file "$icon_launcher"
tell application "Finder" to set label icon of alias file ("$desktop_dir/ST-Launcher.command" as POSIX file) to iconPath
EOF
fi
if [ -f "$icon_sillytavern" ]; then
osascript << EOF
set iconPath to POSIX file "$icon_sillytavern"
tell application "Finder" to set label icon of alias file ("$desktop_dir/SillyTavern.command" as POSIX file) to iconPath
EOF
fi
;;
*)
echo -e "${red_fg_strong}Unsupported operating system for desktop shortcuts${reset}"
return 1
;;
esac
# Verify creation
if [ -f "$desktop_dir/ST-Launcher"* ] && [ -f "$desktop_dir/SillyTavern"* ]; then
echo -e "${green_fg_strong}Shortcuts created successfully on "$desktop_dir/SillyTavern" ${reset}"
else
echo -e "${yellow_fg_strong}Warning: Shortcuts might not have been created correctly${reset}"
fi
}
# Function to install Extras
install_extras() {
echo -e "\033]0;SillyTavern [INSTALL-EXTRAS]\007"
clear
echo -e "${blue_fg_strong}/ Installer / Extras${reset}"
echo "---------------------------------------------------------------"
echo ""
echo -e "${red_bg}╔════ INSTALL SUMMARY ══════════════════════════════════════════════════════════════════════════╗${reset}"
echo -e "${red_bg}║ Extras has been DISCONTINUED since April 2024 and WON'T receive any new updates or modules. ║${reset}"
echo -e "${red_bg}║ The vast majority of modules are available natively in the main SillyTavern application. ║${reset}"
echo -e "${red_bg}║ You may still install and use it but DON'T expect to get support if you face any issues. ║${reset}"
echo -e "${red_bg}║ Below is a list of package requirements that will get installed: ║${reset}"
echo -e "${red_bg}║ * SillyTavern-extras [Size: 65 MB] ║${reset}"
echo -e "${red_bg}║ * Miniconda3 [INSTALLED] [Size: 630 MB] ║${reset}"
echo -e "${red_bg}║ * Miniconda3 env - extras [Size: 9,98 GB] ║${reset}"
echo -e "${red_bg}║ * Git [INSTALLED] [Size: 338 MB] ║${reset}"
echo -e "${red_bg}║ TOTAL INSTALL SIZE: 11 GB ║${reset}"
echo -e "${red_bg}╚═══════════════════════════════════════════════════════════════════════════════════════════════╝${reset}"
echo ""
echo -n "Are you sure you want to proceed? [Y/N]: "
read confirmation
if [ "$confirmation" = "Y" ] || [ "$confirmation" = "y" ]; then
install_extras_y
else
installer
fi
}
install_extras_y() {
clear
# Ask the user about the GPU
echo -e "What is your GPU?"
echo -e "1. NVIDIA"
echo -e "2. AMD"
echo -e "3. None (CPU-only mode)"
echo -e "0. Cancel install"
# Get GPU information
gpu_info=""
while IFS= read -r line; do
gpu_info+=" $line"
done < <(lspci | grep VGA | cut -d ':' -f3)
echo ""
echo -e "${blue_bg}╔════ GPU INFO ═════════════════════════════════════════════════════════════╗${reset}"
boxDrawingText "" 75 $blue_bg
boxDrawingText "* ${gpu_info:1}" 75 $blue_bg
boxDrawingText "" 75 $blue_bg
echo -e "${blue_bg}╚═══════════════════════════════════════════════════════════════════════════╝${reset}"
echo ""
# Prompt for GPU choice
read -p "Enter number corresponding to your GPU: " gpu_choice
# GPU menu - Backend
# Set the GPU choice in an environment variable for choice callback
export GPU_CHOICE=$gpu_choice
# Check the user's response
if [ "$gpu_choice" == "1" ]; then
log_message "INFO" "GPU choice set to NVIDIA"
install_extras_pre
elif [ "$gpu_choice" == "2" ]; then
log_message "INFO" "GPU choice set to AMD"
install_extras_pre
elif [ "$gpu_choice" == "3" ]; then
log_message "INFO" "Using CPU-only mode"
install_extras_pre
elif [ "$gpu_choice" == "0" ]; then
installer
else
log_message "ERROR" "${red_fg_strong}Invalid number. Please enter a valid number.${reset}"
read -p "Press Enter to continue..."
install_extras
fi
}
install_extras_pre() {
log_message "INFO" "Installing Extras..."
log_message "INFO" "Cloning SillyTavern-extras repository..."
git clone https://github.com/SillyTavern/SillyTavern-extras.git
log_message "INFO" "Creating Conda environment: ${cyan_fg_strong}extras${reset}"
conda create -n extras python=3.11 git -y
log_message "INFO" "Activating Conda environment: ${cyan_fg_strong}extras${reset}"
conda init bash
conda activate extras
# Check if extras activation was successful
if [ $? -eq 0 ]; then
log_message "INFO" "Successfully activated Conda environment: extras"
install_extras_successful
else
log_message "ERROR" "${red_fg_strong}Failed to activate Conda environment: extras${reset}"
log_message "INFO" "Press Enter to try again otherwise close the installer and restart."
read -p "Press Enter to try again..."
install_extras_pre
fi
}
install_extras_successful() {
# Navigate to the SillyTavern-extras directory
cd "$PWD/SillyTavern-extras"
log_message "INFO" "Installing pip3 requirements-rvc in Conda environment: ${cyan_fg_strong}extras${reset}"
pip3 install -r requirements-rvc.txt
pip3 install tensorboardX
# Use the GPU choice made earlier to install requirements in the Conda environment extras
if [ "$GPU_CHOICE" == "1" ]; then
log_message "INFO" "Installing modules for NVIDIA from requirements.txt in Conda environment: ${cyan_fg_strong}extras${reset}"
pip3 install -r requirements.txt
conda install -c conda-forge faiss-gpu -y
log_message "INFO" "${green_fg_strong}Extras successfully installed.${reset}"
installer
elif [ "$GPU_CHOICE" == "2" ]; then
log_message "INFO" "Installing modules for AMD from requirements-rocm.txt in Conda environment: ${cyan_fg_strong}extras${reset}"
pip3 install -r requirements-rocm.txt
log_message "INFO" "${green_fg_strong}Extras successfully installed.${reset}"
installer
elif [ "$GPU_CHOICE" == "3" ]; then
log_message "INFO" "Installing modules for CPU from requirements-silicon.txt in Conda environment: ${cyan_fg_strong}extras${reset}"
pip3 install -r requirements-silicon.txt
log_message "INFO" "${green_fg_strong}Extras successfully installed.${reset}"
installer
fi
}
# Function to install XTTS
install_xtts() {
log_message "INFO" "Installing XTTS..."
# Activate the Miniconda installation
log_message "INFO" "Activating Miniconda environment..."
source "$miniconda_path/bin/activate"
# Create a Conda environment named xtts
log_message "INFO" "Creating Conda environment: xtts"
conda create -n xtts python=3.10 git -y
# Activate the xtts environment
log_message "INFO" "Activating Conda environment: xtts"
conda activate xtts
# Check if activation was successful
if [ $? -eq 0 ]; then
log_message "INFO" "Successfully activated Conda environment: xtts"
install_xtts_successful
else
log_message "ERROR" "${red_fg_strong}Failed to activate Conda environment: xtts${reset}"
log_message "INFO" "Press Enter to try again otherwise close the installer and restart."
read -p "Press Enter to try again..."
install_xtts
fi
}
install_xtts_successful() {
# Create folders for xtts
log_message "INFO" "Creating xtts folders..."
mkdir "$PWD/xtts"
mkdir "$PWD/xtts/speakers"
mkdir "$PWD/xtts/output"
# Clone the xtts-api-server repository for voice examples
log_message "INFO" "Cloning xtts-api-server repository..."
git clone https://github.com/daswer123/xtts-api-server.git
log_message "INFO" "Adding voice examples to speakers directory..."
cp -r "$PWD/xtts-api-server/example/"* "$PWD/xtts/speakers/"
log_message "INFO" "Removing the xtts-api-server directory..."
rm -rf "$PWD/xtts-api-server"
# Install pip3 requirements
log_message "INFO" "Installing pip3 requirements for xtts..."
pip3 install xtts-api-server
pip3 install pydub
pip3 install stream2sentence
installer
}
# Exit Function
exit_program() {
clear
echo "Bye!"
exit 0
}
############################################################
################# INSTALLER - FRONTEND #####################
############################################################
installer() {
echo -e "\033]0;SillyTavern [INSTALLER]\007"
clear
echo -e "${blue_fg_strong}| > / Installer |${reset}"
echo -e "${blue_fg_strong}==============================================================${reset}"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| What would you like to do? |${reset}"
echo " 1. Install SillyTavern"
echo " 2. Install XTTS"
echo " 3. Support"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| Discontinued |${reset}"
echo " 4. Install Extras"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| Menu Options: |${reset}"
echo " 0. Exit"
echo -e "${cyan_fg_strong} _____________________________________________________________${reset}"
echo -e "${cyan_fg_strong}| |${reset}"
read -p " Choose Your Destiny (default is 1): " choice
# Default to choice 1 if no input is provided
if [ -z "$choice" ]; then
choice=1
fi
################# INSTALLER - BACKEND #####################
case $choice in
1) install_sillytavern ;;
2) install_xtts ;;
3) support ;;
4) install_extras ;;
# 4) install_all ;;
0) exit_program ;;
*) echo -e "${yellow_fg_strong}WARNING: Invalid number. Please insert a valid number.${reset}"
read -p "Press Enter to continue..."
installer ;;
esac
}
detect_package_manager() {
local package_manager
# Check for macOS first
if [ "$(uname)" = "Darwin" ]; then
if command -v brew &> /dev/null; then
echo "brew"
return 0
else
echo "unknown"
return 1
fi
fi
# Check for Linux package managers
case "$(true; command -v apt dnf yum apk pacman emerge zypper xbps-install guix nix-env 2>/dev/null | head -n1)" in
*apt) package_manager="apt" ;;
*dnf) package_manager="dnf" ;;
*yum) package_manager="yum" ;;
*apk) package_manager="apk" ;;
*pacman) package_manager="pacman" ;;
*emerge) package_manager="emerge" ;;
*zypper) package_manager="zypper" ;;
*xbps-install) package_manager="xbps" ;;
*guix) package_manager="guix" ;;
*nix-env) package_manager="nix" ;;
*)
if [ -f /etc/os-release ]; then
os_id=$(grep -E '^ID=' /etc/os-release | cut -d= -f2 | tr -d '"')
case "$os_id" in
almalinux|centos_stream|ol|rhel|rocky|fedora|fedoraremixforwsl|nobara)
package_manager="yum" ;;
alpine|wolfi)
package_manager="apk" ;;
arch|arch32|archcraft|arcolinux|arkane|artix|aurora|bazzite|blackarch|blendos|bluefin|bodhi|cachyos|chimera|chimeraos|endeavouros|garuda|hyperbola|kaos|manjaro|manjaro-arm|rebornos|steamos|void)
package_manager="pacman" ;;
debian|devuan|elementary|kali|linuxmint|parrot|pika|pisilinux|pop|pureos|raspbian|tails|ubuntu|ubuntu_kylin|vanillaos|zorin)
package_manager="apt" ;;
opensuse-leap|opensuse-tumbleweed|sled|sles)
package_manager="zypper" ;;
gentoo|funtoo)
package_manager="emerge" ;;
nixos)
package_manager="nix" ;;
guix)
package_manager="guix" ;;
clear-linux-os|miraclelinux|photon|solus|tencentos|tinycore|trisquel|ultramarine)
package_manager="unknown" ;; # Custom handling needed
*)
package_manager="unknown" ;;
esac
else
package_manager="unknown"
fi
;;
esac
echo "$package_manager"
}
# Main function to install dependencies based on package manager
install_dependencies() {
PACKAGE_MANAGER=$(detect_package_manager)
case "$PACKAGE_MANAGER" in
apt)
log_message "INFO" "Detected APT on Debian/Ubuntu-based system."
sudo apt update -y
install_git
install_nodejs
install_miniconda
installer
;;
dnf|yum)
log_message "INFO" "Detected DNF/YUM on RHEL-based system."
sudo $PACKAGE_MANAGER install -y epel-release
install_git
install_nodejs
install_miniconda
installer
;;
pacman)
log_message "INFO" "Detected Pacman on Arch-based system."
sudo pacman -Sy --noconfirm
install_git
install_nodejs
install_miniconda
installer
;;
apk)
log_message "INFO" "Detected APK on Alpine Linux."
sudo apk update
install_git
install_nodejs
install_miniconda
installer
;;
zypper)
log_message "INFO" "Detected Zypper on openSUSE."
sudo zypper refresh
install_git
install_nodejs
install_miniconda
installer
;;
emerge)
log_message "INFO" "Detected Portage on Gentoo."
sudo emerge --sync
install_git
install_nodejs
install_miniconda
installer
;;
nix)
log_message "INFO" "Detected Nix Package Manager."
nix-env -iA nixpkgs.git nixpkgs.nodejs
install_miniconda
installer
;;
guix)
log_message "INFO" "Detected Guix Package Manager."
install_git
install_nodejs
install_miniconda
installer
;;
brew)
log_message "INFO" "Detected Homebrew on macOS."
if ! command -v brew &> /dev/null; then
log_message "ERROR" "Homebrew is not installed. Please install it first: https://brew.sh/"
exit 1
fi
install_git
install_nodejs
install_miniconda
installer
;;
*)
log_message "ERROR" "Unknown package manager. Please install dependencies manually."
exit 1
;;
esac
}
# Startup functions
detect_package_manager
install_dependencies
installer
Reference in New Issue View Git Blame Copy Permalink