common_env.sh

#!/bin/bash
# Shared AutoVoice environment bootstrap for Jetson Thor scripts.

# Resolve paths before enabling nounset; shell startup hooks such as RVM can
# assume unset variables during `cd` and otherwise poison command substitution.
AUTOVOICE_SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
AUTOVOICE_PROJECT_ROOT="$(cd "$AUTOVOICE_SCRIPT_DIR/.." && pwd)"

set -euo pipefail
AUTOVOICE_ENV_NAME="${AUTOVOICE_ENV_NAME:-autovoice-thor}"
if [[ -z "${AUTOVOICE_ENV_PREFIX:-}" ]]; then
    if [[ -n "${AUTOVOICE_PYTHON:-}" && -x "${AUTOVOICE_PYTHON}" ]]; then
        AUTOVOICE_ENV_PREFIX="$(cd "$(dirname "${AUTOVOICE_PYTHON}")/.." && pwd)"
    elif [[ -n "${CONDA_PREFIX:-}" && "${CONDA_DEFAULT_ENV:-}" == "$AUTOVOICE_ENV_NAME" ]]; then
        AUTOVOICE_ENV_PREFIX="${CONDA_PREFIX}"
    else
        AUTOVOICE_ENV_PREFIX="${HOME}/anaconda3/envs/${AUTOVOICE_ENV_NAME}"
    fi
fi
AUTOVOICE_PYTHON_DEFAULT="${AUTOVOICE_ENV_PREFIX}/bin/python"
AUTOVOICE_DATA_DIR="${AUTOVOICE_DATA_DIR:-${DATA_DIR:-${AUTOVOICE_PROJECT_ROOT}/data}}"

autovoice_activate_env() {
    local requested_python="${AUTOVOICE_PYTHON:-$AUTOVOICE_PYTHON_DEFAULT}"

    if [[ -x "$requested_python" ]]; then
        export PYTHON="$requested_python"
    elif command -v conda >/dev/null 2>&1; then
        eval "$(conda shell.bash hook)"
        conda activate "$AUTOVOICE_ENV_NAME"
        export PYTHON="$(command -v python)"
    else
        echo "AutoVoice Python interpreter not found: $requested_python" >&2
        return 1
    fi

    export PYTHONNOUSERSITE=1
    export PYTHONPATH="${AUTOVOICE_PROJECT_ROOT}/src${PYTHONPATH:+:${PYTHONPATH}}"
    export CUDA_HOME="${CUDA_HOME:-/usr/local/cuda-13.0}"
    export PATH="${CUDA_HOME}/bin:${PATH}"
    export LD_LIBRARY_PATH="${CUDA_HOME}/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"
    export TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-11.0}"
}