install.sh

#!/bin/bash
# Enable error handling
set -e

function install_provider() {
    local provider=$1
    local api_key_var_name=$2
    local pip_package=$3
    local script_name=$4

    read -p "Enter your $provider API Key: " api_key
    if [ -f ".env" ]; then
        # Remove any existing API key and provider settings
        sed -i.bak "/^${api_key_var_name}_API_KEY=/d" .env
        sed -i.bak "/^PROVIDER=/d" .env
        rm -f .env.bak
    fi
    echo "PROVIDER=$provider" >> .env
    echo "${api_key_var_name}_API_KEY=$api_key" >> .env
    echo "Installing $provider module..."
    pip install $pip_package
    scripts/generate-cmd.sh $script_name
}


if [ ! -d "scripts" ]; then
    echo "Please run the script from the root directory of the project."
    exit 1
fi

echo "Checking if Python 3 is installed..."
if ! python3 --version &>/dev/null; then
    echo "Python 3 not found. Please install Python 3 and try again."
    exit 1
fi

PYTHON_VERSION=$(python3 --version 2>&1 | awk '{print $2}')
MIN_VERSION="3.10.0"

if [[ "$(printf '%s\n' "$MIN_VERSION" "$PYTHON_VERSION" | sort -V | head -n1)" != "$MIN_VERSION" ]]; then
    echo "Detected Python version is less than 3.10.0. Please upgrade your Python version."
    exit 1
else
    echo "Python version is compatible."
fi

echo "Checking if \"envsubtrans\" folder exists..."
if [ -d "envsubtrans" ]; then
    echo "\"envsubtrans\" folder already exists."
    read -p "Do you want to perform a clean install? This will delete the existing environment. (Y/N): " user_choice
    if [ "$user_choice" = "Y" ] || [ "$user_choice" = "y" ]; then
        echo "Performing a clean install..."
        rm -rf envsubtrans
        [ -f .env ] && rm .env
    elif [ "$user_choice" != "N" ] && [ "$user_choice" != "n" ]; then
        echo "Invalid choice. Exiting installation."
        exit 1
    fi
fi

python3 -m venv envsubtrans
source envsubtrans/bin/activate

scripts/generate-cmd.sh gui-subtrans
scripts/generate-cmd.sh llm-subtrans

echo "Select which provider you want to install:"
echo "0 = None"
echo "1 = OpenAI"
echo "2 = Google Gemini"
echo "3 = Anthropic Claude"
echo "a = All"
read -p "Enter your choice (0/1/2/3/a): " provider_choice

case $provider_choice in
    0)
        echo "No additional provider selected. Moving forward without any installations."
        ;;
    1)
        install_provider "OpenAI" "OPENAI" "openai" "gpt-subtrans"
        ;;
    2)
        install_provider "Google Gemini" "GEMINI" "google-generativeai" "gemini-subtrans"
        ;;
    3)
        install_provider "Claude" "CLAUDE" "anthropic" "claude-subtrans"
        ;;
    a)
        install_provider "Claude" "CLAUDE" "anthropic" "claude-subtrans"
        install_provider "Google Gemini" "GEMINI" "google-generativeai" "gemini-subtrans"
        install_provider "OpenAI" "OPENAI" "openai" "gpt-subtrans"
        ;;
    *)
        echo "Invalid choice. Exiting installation."
        exit 1
        ;;
esac

echo "Installing required modules..."
pip install --upgrade -r requirements.txt

echo "Setup completed successfully. To uninstall just delete the directory"

exit 0