#!/bin/bash

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_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

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
    /root/venvs/sia/bin/ipython kernel install --name=sia
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