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

echo "=== HYDRA Phase 1: Sequential Subsystem Bring-Up ==="
cd "$(dirname "$0")/.."

SUBSYSTEMS=("mamba3" "mhc" "engram" "hestia" "sdr")

for sub in "${SUBSYSTEMS[@]}"; do
    echo ""
    echo "--- Subsystem: ${sub} ---"
    BRANCH="autoresearch/phase1-${sub}"

    # Create branch if it doesn't exist
    if ! git rev-parse --verify "${BRANCH}" &>/dev/null; then
        git checkout -b "${BRANCH}"
    else
        git checkout "${BRANCH}"
    fi

    echo "Running: uv run subsystems/train_${sub}.py"
    uv run "subsystems/train_${sub}.py" > "run_${sub}.log" 2>&1 || true

    # Extract result
    echo "Result:"
    grep "^val_bpb:" "run_${sub}.log" || echo "  (crashed)"
    grep "^peak_vram_mb:" "run_${sub}.log" || true
done

echo ""
echo "=== Phase 1 complete ==="
git checkout main 2>/dev/null || git checkout master