setup.sh

#!/usr/bin/env bash
# setup.sh — Agent Patterns Demo Pack setup script (macOS / Linux)
# Creates a virtual environment, installs dependencies, and prepares .env config.

set -euo pipefail

PROJECT_ROOT="$(cd "$(dirname "$0")" && pwd)"

echo ""
echo "===== Agent Patterns Demo Pack — Setup ====="
echo ""

# --- Python check ---
PYTHON_CMD=""
for candidate in python3 python; do
    if command -v "$candidate" &>/dev/null; then
        ver=$("$candidate" --version 2>&1)
        minor=$(echo "$ver" | grep -oP '(?<=Python 3\.)\d+' || echo "")
        if [ -z "$minor" ]; then
            minor=$(echo "$ver" | sed -n 's/.*Python 3\.\([0-9]*\).*/\1/p')
        fi
        if [ -n "$minor" ] && [ "$minor" -ge 10 ]; then
            PYTHON_CMD="$candidate"
            echo "[OK] Found $ver"
            break
        fi
    fi
done

if [ -z "$PYTHON_CMD" ]; then
    echo "[ERROR] Python 3.10+ is required but was not found."
    echo "        Install from https://www.python.org/downloads/"
    exit 1
fi

# --- Virtual environment ---
VENV_PATH="$PROJECT_ROOT/.venv"
if [ ! -d "$VENV_PATH" ]; then
    echo ""
    echo "Creating virtual environment..."
    "$PYTHON_CMD" -m venv "$VENV_PATH"
    echo "[OK] Virtual environment created at .venv"
else
    echo "[OK] Virtual environment already exists at .venv"
fi

# --- Activate venv ---
# shellcheck disable=SC1091
source "$VENV_PATH/bin/activate"

# --- Install dependencies ---
echo ""
echo "Installing dependencies..."
pip install --upgrade pip > /dev/null
pip install -r "$PROJECT_ROOT/requirements.txt" --pre
echo "[OK] Dependencies installed"

# --- .env file ---
ENV_FILE="$PROJECT_ROOT/.env"
ENV_EXAMPLE="$PROJECT_ROOT/.env.example"
if [ ! -f "$ENV_FILE" ]; then
    if [ -f "$ENV_EXAMPLE" ]; then
        cp "$ENV_EXAMPLE" "$ENV_FILE"
        echo "[OK] Created .env from .env.example"
    else
        echo "[WARN] No .env.example found — skipping .env creation"
    fi
else
    echo "[OK] .env already exists"
fi

# --- Foundry Local check ---
echo ""
if command -v foundry &>/dev/null; then
    echo "[OK] Foundry Local is installed"
else
    echo "[INFO] Foundry Local is not installed."
    echo "       Install it with:  brew install foundrylocal   (macOS)"
    echo "       Or visit:         https://foundrylocal.ai"
fi

# --- Done ---
echo ""
echo "===== Setup complete! ====="
echo ""
echo "Next steps:"
echo "  1. Activate the venv:    source .venv/bin/activate"
echo "  2. Start a model:        foundry model run qwen2.5-1.5b"
echo "  3. Launch the app:       python app.py"
echo "  4. Open browser:         http://localhost:8765"
echo ""