#!/bin/bash
# Setup script for dual v9 parameter sweep (vanilla + RSI divergence)
# Run this on EPYC server after extracting backtest_v9_dual_sweep.tar.gz

set -e

echo "=== v9 Dual Sweep Setup ==="
echo

# Check Python version
PYTHON_VERSION=$(python3 --version 2>&1)
echo "Python: $PYTHON_VERSION"

# Install python3-venv if needed
if ! dpkg -l | grep -q python3-venv; then
    echo "Installing python3-venv..."
    sudo apt-get update
    sudo apt-get install -y python3-venv
fi

# Create virtual environment
if [ ! -d ".venv" ]; then
    echo "Creating virtual environment..."
    python3 -m venv .venv
fi

# Activate and install dependencies
echo "Installing dependencies..."
source .venv/bin/activate
pip install --upgrade pip
pip install pandas numpy

echo
echo "=== Setup Complete ==="
echo "Virtual environment: $(pwd)/.venv"
echo "Python: $(which python)"
echo "Packages: $(pip list | grep -E 'pandas|numpy')"
echo
echo "Ready to run sweeps!"
echo "  - Vanilla v9:        ./run_sweep_vanilla_epyc.sh"
echo "  - RSI Divergence v9: ./run_sweep_rsi_epyc.sh"