#!/bin/bash
set -e

echo "🌌 Synesthesia Phase 6 — Environment Bootstrap"

# 1. Ensure uv is installed
if ! command -v uv &> /dev/null; then
    echo "Installing uv..."
    curl -LsSf https://astral.sh/uv/install.sh | sh
    # If on Linux, refresh path
    export PATH="$HOME/.local/bin:$PATH"
fi

# 2. Create and synchronize virtual environment
echo "Synchronizing dependencies with uv..."
uv venv .venv
source .venv/bin/activate

# 3. Install ROCm-optimized PyTorch (gfx1030 / RX 6700 XT)
echo "Installing ROCm-optimized PyTorch..."
uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm7.2

# 4. Install project requirements (JAX ROCm, ONNX ROCm, etc.)
echo "Installing project-specific requirements..."
uv pip install -r ML_Pipeline/requirements.txt

# 5. Create necessary directories for local models and logs
mkdir -p models/quantized
mkdir -p runtime/logs

echo "✅ Bootstrap Complete. Run with: uv run streamlit run app/streamlit_app.py"