#!/bin/bash

set -e

echo "=== Preparing SIA environment ==="

echo "Installing ITB tool..."
python3 -m venv "/root/venvs/itb"
/root/venvs/itb/bin/pip install -e /root/sia/tools/itb/

echo "Installing Train tool..."
python3 -m venv "/root/venvs/train"
/root/venvs/train/bin/pip install -e /root/sia/tools/train/

echo "Installing SIA core..."
python3 -m venv "/root/venvs/sia"
/root/venvs/sia/bin/pip install -e /root/sia/

echo "=== Starting SIA ==="

cd "/root/desktop"
while true; do
    sia
    EXIT_CODE=$?
    
    if [ $EXIT_CODE -eq 42 ]; then
        echo "SIA exited with code 42. Restarting in 2 seconds..."
        sleep 2
    else
        echo "SIA exited with code $EXIT_CODE. Not restarting."
        break
    fi
done

exit $EXIT_CODE