#!/bin/bash

set -e  # stop if any command fails

echo "Setting up KM3NeT Masterclass environment..."

# ---- Repository URLs ----
CALIB_REPO="https://git.km3net.de/education/masterclass_calib.git"
ANALYSIS_REPO="https://git.km3net.de/education/masterclass.git"

# ---- Clone calibration repo ----

if [ ! -d "calibration_masterclass" ]; then
    echo "Cloning calibration repository..."
    git clone "$CALIB_REPO" calibration_masterclass
else
    echo "Calibration repository already exists."
fi

# ---- Clone analysis repo ----

if [ ! -d "analysis_masterclass" ]; then
    echo "Cloning analysis repository..."
    git clone "$ANALYSIS_REPO" analysis_masterclass
else
    echo "Analysis repository already exists."
fi

# ---- Create virtual environment ----

echo "Creating virtual environment..."

python3 -m pip install --user virtualenv
python3 -m virtualenv masterclass_venv

echo "Activating environment..."

source masterclass_venv/bin/activate

echo "Installing required packages..."

pip install --upgrade pip
pip install voila ipywidgets matplotlib numpy pandas scipy mplcursors jupyterlab

echo ""
echo "Environment setup complete."
echo ""
echo "Activate it later with:"
echo "source masterclass_venv/bin/activate"