#!/usr/bin/env bash
# Creates a venv in ./venv and installs Python deps for Bark.
# Safe to re-run — skips install if already done.

set -euo pipefail

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

if [ ! -d "${VENV}" ]; then
	echo "==> creating venv at ${VENV}"
	python3 -m venv "${VENV}"
fi

echo "==> installing Python dependencies"
"${VENV}/bin/pip" install --upgrade pip --quiet
"${VENV}/bin/pip" install transformers accelerate torch numpy
"${VENV}/bin/pip" install chatterbox-tts

chmod +x "${SCRIPT_DIR}/bark-server.py"
chmod +x "${SCRIPT_DIR}/chatterbox-server.py"

echo ""
echo "==> venv ready at ${VENV}"