#!/usr/bin/env bash
set -euo pipefail

echo "=== HYDRA Setup ==="
echo ""

# Check uv
if ! command -v uv &>/dev/null; then
    echo "Installing uv..."
    curl -LsSf https://astral.sh/uv/install.sh | sh
fi

# Install Python dependencies
echo "Installing Python dependencies..."
cd "$(dirname "$0")/.."
uv sync

# Prepare data (download shards + train tokenizer)
echo ""
echo "Preparing data (this may take a few minutes on first run)..."
uv run prepare.py --num-shards 10

echo ""
echo "=== Setup complete ==="
echo "Run experiments with: uv run train.py"
echo "Run orchestrator with: uv run -m harness.orchestrator"
echo "Run Phase 1 subsystems with: bash scripts/run_phase1.sh"