#!/usr/bin/env bash
# ─────────────────────────────────────────────────────────────
# ProdAssist RAG — One-shot environment setup
# ─────────────────────────────────────────────────────────────
set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
ROOT="$(dirname "$SCRIPT_DIR")"

echo "=============================================="
echo "  ProdAssist RAG — Setup"
echo "=============================================="

# 1. Create virtual environment
if [ ! -d "$ROOT/.venv" ]; then
    echo "[1/4] Creating virtual environment (.venv)…"
    python3 -m venv "$ROOT/.venv"
else
    echo "[1/4] Virtual environment already exists — skipping."
fi

# 2. Activate
source "$ROOT/.venv/bin/activate"
echo "[2/4] Virtual environment activated."

# 3. Upgrade pip
pip install --upgrade pip --quiet

# 4. Install requirements
echo "[3/4] Installing requirements…"
pip install -r "$ROOT/requirements.txt"

# 5. Create required directories
echo "[4/4] Creating data directories…"
mkdir -p "$ROOT/db/chroma" "$ROOT/db/faiss" "$ROOT/data/uploads"

echo ""
echo "=============================================="
echo "  Setup complete!"
echo ""
echo "  To start the app:"
echo "    source .venv/bin/activate"
echo "    python main.py ui"
echo ""
echo "  Or directly:"
echo "    .venv/bin/python main.py ui"
echo "=============================================="