init_script.sh

#!/usr/bin/env bash
set -e  # Exit immediately if a command fails

echo "🚀 Starting setup..."

# 1. Check if uv is installed, install if missing
if ! command -v uv &> /dev/null; then
    echo "📦 uv not found. Installing uv..."
    curl -LsSf https://astral.sh/uv/install.sh | sh

    # Reload shell so uv is available immediately
    export PATH="$HOME/.local/bin:$PATH"
else
    echo "✅ uv is already installed"
fi

# 2. Install dependencies
echo "📦 Installing dependencies..."
uv sync

# 3. Check if data exists before downloading
DATA_DIR="./data"

if [ ! -d "$DATA_DIR" ] || [ -z "$(ls -A $DATA_DIR 2>/dev/null)" ]; then
    echo "⬇️ Data not found. Preparing dataset..."
    uv run prepare.py
else
    echo "✅ Data already present, skipping prepare step"
fi

# 4. Run training
echo "🏋️ Running training..."
uv run train.py

echo "🎉 Done!"