#!/usr/bin/env bash
# Setup script for a fresh Vast.ai instance.
# Assumes: PyTorch + CUDA image, Ubuntu, this repo uploaded to ~/alkyline
set -euo pipefail

echo "=== Arcspan Vast.ai Setup ==="

cd ~/alkyline

# 1. Create venv and install OPF
python3 -m venv .venv
source .venv/bin/activate
pip install --upgrade pip
pip install -e vendor/privacy-filter/

# 2. Verify GPU
python3 -c "import torch; print(f'CUDA: {torch.cuda.is_available()}, Device: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"none\"}')"

# 3. Download base checkpoint (auto on first run, but let's trigger it)
echo "Downloading base checkpoint..."
opf "test" > /dev/null 2>&1 || true

echo ""
echo "=== Setup complete. Run training with: ==="
echo "  source .venv/bin/activate"
echo "  bash scripts/train_cyner.sh 40 cuda 8"
echo ""
echo "Then evaluate with:"
echo "  bash scripts/eval_cyner.sh checkpoints/cyner_v1 cuda"