build-python.sh

#!/usr/bin/env bash
set -euo pipefail

PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"

if [[ -x "$PROJECT_ROOT/.venv/bin/python" ]]; then
  PYTHON_EXE="$PROJECT_ROOT/.venv/bin/python"
else
  PYTHON_EXE="${PYTHON_BIN:-python3}"
fi

if ! "$PYTHON_EXE" -c "import importlib.util, sys; sys.exit(0 if importlib.util.find_spec('PyInstaller') else 1)"; then
  echo "PyInstaller not found. Run 'python -m pip install pyinstaller' before building." >&2
  exit 1
fi

DIST_PATH="$PROJECT_ROOT/build/python-dist"
WORK_PATH="$PROJECT_ROOT/build/pyinstaller/work"
SPEC_PATH="$PROJECT_ROOT/build/pyinstaller/spec"
HOOKS_PATH="$PROJECT_ROOT/scripts/pyinstaller-hooks"
TARGET_ARCH="${PYINSTALLER_TARGET_ARCH:-}"

rm -rf "$DIST_PATH" "$PROJECT_ROOT/build/pyinstaller"
mkdir -p "$DIST_PATH" "$WORK_PATH" "$SPEC_PATH"

COMMON_ARGS=(
  -m PyInstaller
  --noconfirm
  --clean
  --distpath "$DIST_PATH"
  --workpath "$WORK_PATH"
  --specpath "$SPEC_PATH"
  --additional-hooks-dir "$HOOKS_PATH"
)

CUDA_ARGS=()

has_python_module() {
  "$PYTHON_EXE" -c "import importlib.util, sys; sys.exit(0 if importlib.util.find_spec('$1') else 1)" >/dev/null 2>&1
}

for module in nvidia.cublas nvidia.cudnn nvidia.cuda_runtime; do
  if has_python_module "$module"; then
    CUDA_ARGS+=(--collect-all "$module")
  fi
done

if [[ "$OSTYPE" == darwin* && -n "$TARGET_ARCH" ]]; then
  COMMON_ARGS+=(--target-arch "$TARGET_ARCH")
fi

DICTATION_ARGS=("${COMMON_ARGS[@]}")
if ((${#CUDA_ARGS[@]} > 0)); then
  DICTATION_ARGS+=("${CUDA_ARGS[@]}")
fi

"$PYTHON_EXE" "${DICTATION_ARGS[@]}" \
  --name dictation_service \
  --collect-all faster_whisper \
  --collect-all ctranslate2 \
  --collect-all tokenizers \
  --collect-all huggingface_hub \
  --collect-all sounddevice \
  python/dictation_service.py

"$PYTHON_EXE" "${COMMON_ARGS[@]}" \
  --name hotkey_listener \
  --collect-submodules pynput \
  python/hotkey_listener.py

echo "Python workers built in $DIST_PATH"