build-whisperx-bundle.sh

#!/bin/bash
set -e

# === CONFIG ===
BUNDLE_DIR="whisperx-bundle"
PYTHON_VERSION="3.11"
WHISPER_MODEL="medium"
FFMPEG_URL="https://johnvansickle.com/ffmpeg/releases/ffmpeg-release-amd64-static.tar.xz"
# Use the latest available from astral-sh/python-build-standalone
PYTHON_STANDALONE_URL="https://github.com/astral-sh/python-build-standalone/releases/download/20250712/cpython-3.11.13+20250712-x86_64-unknown-linux-gnu-install_only.tar.gz"
PYTHON_STANDALONE_DIR="python-standalone"
PYTHON_STANDALONE_TAR="python-standalone.tar.gz"

# === Optional: Clean install ===
if [[ "$1" == "--clean" ]]; then
  echo "🧹 Removing previous install..."
  rm -rf "$BUNDLE_DIR" "$PYTHON_STANDALONE_DIR"
fi

# === STEP 0: Download and extract portable Python if needed ===
if [[ ! -x "$PYTHON_STANDALONE_DIR/bin/python3" ]]; then
  echo "🐍 Downloading portable Python..."
  echo "URL: $PYTHON_STANDALONE_URL"
  echo "Output: $PYTHON_STANDALONE_TAR"
  curl -L "$PYTHON_STANDALONE_URL" -o "$PYTHON_STANDALONE_TAR"
  # Check if the file is a valid gzip archive
  if ! file "$PYTHON_STANDALONE_TAR" | grep -q 'gzip compressed data'; then
    echo "❌ Downloaded file is not a valid gzip archive. Check your network or the download URL."
    echo "File type: $(file "$PYTHON_STANDALONE_TAR")"
    exit 1
  fi
  mkdir -p "$PYTHON_STANDALONE_DIR"
  tar -xzf "$PYTHON_STANDALONE_TAR" -C "$PYTHON_STANDALONE_DIR" --strip-components=1
  rm "$PYTHON_STANDALONE_TAR"
else
  echo "🐍 Portable Python already present."
fi

PYTHON_BIN="$(pwd)/$PYTHON_STANDALONE_DIR/bin/python3"

# === SAFETY: Prevent running inside wrong virtualenv ===
if [[ -n "$VIRTUAL_ENV" && "$VIRTUAL_ENV" != "$(pwd)/$BUNDLE_DIR/whisperx-env" ]]; then
  echo "❌ You're already inside another virtual environment: $VIRTUAL_ENV"
  echo "Please deactivate it before running this script."
  exit 1
fi

# === STEP 1: Handle existing bundle directory ===
if [[ -d "$BUNDLE_DIR" ]]; then
  echo "⚠️  Bundle directory '$BUNDLE_DIR' already exists."
  read -p "Reuse existing directory and skip recreating it? (y/n): " reuse
  if [[ "$reuse" != "y" ]]; then
    echo "❌ Aborting. Delete the folder or rename it first, or run with --clean."
    exit 1
  fi
else
  echo "📁 Creating bundle directory: $BUNDLE_DIR"
  mkdir -p "$BUNDLE_DIR"
fi

cd "$BUNDLE_DIR"

# === STEP 2: Virtualenv Setup (using portable Python) ===
if [[ ! -d "whisperx-env" ]]; then
  echo "🐍 Creating Python $PYTHON_VERSION virtual environment (standalone)..."
  "$PYTHON_BIN" -m venv whisperx-env
else
  echo "🐍 Virtual environment already exists. Reusing."
fi

source whisperx-env/bin/activate

if [[ "$VIRTUAL_ENV" != "$(pwd)/whisperx-env" ]]; then
  echo "❌ Failed to activate expected virtualenv: whisperx-env"
  deactivate || true
  exit 1
fi

# === STEP 3: Install Python packages ===
echo "⬆️  Upgrading pip..."
pip install --upgrade pip

echo "📦 Installing compatible numpy (pinned to 1.24.4 for wheel support)..."
pip install numpy==1.24.4

echo "📦 Installing PyTorch (CPU)..."
pip install torch torchvision torchaudio

echo "📦 Installing WhisperX..."
pip install git+https://github.com/m-bain/whisperx.git

echo "🧠 Downloading WhisperX model: $WHISPER_MODEL"
python -c "import whisperx; whisperx.load_model('$WHISPER_MODEL', device='cpu', compute_type='int8')"

# === STEP 4: FFmpeg Download ===
if [[ -f "ffmpeg/ffmpeg" ]]; then
  echo "🎞️  FFmpeg already exists. Skipping download."
else
  echo "🎞️  Downloading static FFmpeg..."
  mkdir -p ffmpeg
  curl -L "$FFMPEG_URL" -o ffmpeg.tar.xz
  tar -xf ffmpeg.tar.xz --strip-components=1 -C ffmpeg
  rm ffmpeg.tar.xz
fi

# === DONE ===
deactivate
echo "✅ WhisperX bundle ready at: $(pwd)"
du -sh .