#!/bin/bash

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_TRAIN=1

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_TRAIN}" ]; then
    echo "Installing Train tool..."
    python3 -m venv /root/venvs/train
    /root/venvs/train/bin/pip install -e /root/sia/tools/train
    /root/venvs/train/bin/ipython kernel install --name=train
fi