#!/bin/bash

#SIA_INSTALL_NO_WEB=1
#SIA_INSTALL_NO_LLAMA_CPP=1
#SIA_INSTALL_NO_NOTEBOOK=1
#SIA_INSTALL_NO_CORE=1
#SIA_INSTALL_NO_ITB=1
#SIA_INSTALL_NO_MISTRAL_INFER=1
#SIA_INSTALL_NO_GEMMA_INFER=1
#SIA_INSTALL_NO_GEMMA_TRAIN=1

cd "/root/desktop"

if [ -z "${SIA_INSTALL_NO_WEB}" ]; then
    echo "Building web interface"
    (
        cd "/root/sia/web"
        npm install
        npm run build
        ln -s "/root/sia/web/dist" "/root/static"
    )
fi

if [ -z "${SIA_INSTALL_NO_LLAMA_CPP}" ]; then
    echo "Installing venv for llama.cpp"
    python3 -m venv /root/venvs/llama_cpp
    /root/venvs/llama_cpp/bin/pip install /root/sia/modules/llama.cpp
fi

if [ -z "${SIA_INSTALL_NO_NOTEBOOK}" ]; then
    echo "Installing venv for running notebooks"
    python3 -m venv /root/venvs/notebook
    /root/venvs/notebook/bin/pip install jupyter ipykernel ipywidgets
    /root/venvs/notebook/bin/ipython kernel install --name=notebook
fi

if [ -z "${SIA_INSTALL_NO_CORE}" ]; then
    echo "Installing SIA core"
    python3 -m venv /root/venvs/sia
    /root/venvs/sia/bin/pip install -e /root/sia
fi

if [ -z "${SIA_INSTALL_NO_ITB}" ]; then
    echo "Installing ITB tool"
    python3 -m venv /root/venvs/itb
    /root/venvs/itb/bin/pip install -e /root/sia/tools/itb
fi

if [ -z "${SIA_INSTALL_NO_MISTRAL_INFER}" ]; then
    echo "Installing venv for mistral inference"
    python3 -m venv /root/venvs/mistral_infer
    /root/venvs/mistral_infer/bin/pip install -e /root/sia/tools/mistral_infer
fi

if [ -z "${SIA_INSTALL_NO_GEMMA_INFER}" ]; then
    echo "Installing venv for gemma inference"
    python3 -m venv /root/venvs/gemma_infer
    /root/venvs/gemma_infer/bin/pip install -e /root/sia/tools/gemma_infer
fi

if [ -z "${SIA_INSTALL_NO_GEMMA_TRAIN}" ]; then
    echo "Installing venv for gemma training"
    python3 -m venv /root/venvs/gemma_train
    /root/venvs/gemma_train/bin/pip install -e /root/sia/tools/gemma_train
fi

if ! grep -q "source /root/sia/scripts/add_venvs_to_path.sh" /root/.bashrc; then
    echo 'source /root/sia/scripts/add_venvs_to_path.sh' >> /root/.bashrc
fi

if ! grep -q "source /root/sia/scripts/add_venvs_to_path.sh" /etc/profile; then
    echo 'source /root/sia/scripts/add_venvs_to_path.sh' >> /etc/profile
fi

source /root/sia/scripts/add_venvs_to_path.sh