docker-entrypoint.sh

#!/bin/bash
set -e

echo "=========================================="
echo "Speaches Container Startup"
echo "=========================================="

# Configuration - can be overridden by environment variables
PRELOAD_MODELS="${PRELOAD_MODELS:-speaches-ai/Kokoro-82M-v1.0-ONNX-fp16,deepdml/faster-whisper-large-v3-turbo-ct2}"
SKIP_MODEL_DOWNLOAD="${SKIP_MODEL_DOWNLOAD:-false}"

# Function to check if a model is already cached
model_is_cached() {
    local model_id="$1"
    local cache_path="${HF_HUB_CACHE:-$HOME/.cache/huggingface/hub}"
    local model_dir="models--${model_id//\/\/--}"

    if [ -d "$cache_path/$model_dir" ]; then
        return 0  # Model exists
    else
        return 1  # Model doesn't exist
    fi
}

# Download models if not skipped
if [ "$SKIP_MODEL_DOWNLOAD" != "true" ]; then
    echo "Checking for required models..."

    IFS=',' read -ra MODELS <<< "$PRELOAD_MODELS"
    for model_id in "${MODELS[@]}"; do
        model_id=$(echo "$model_id" | xargs)  # Trim whitespace

        if model_is_cached "$model_id"; then
            echo "✓ Model already cached: $model_id"
        else
            echo "↓ Downloading model: $model_id"
            python model_downloader.py "$model_id" || echo "⚠ Warning: Failed to download $model_id"
        fi
    done

    echo "Model preparation complete!"
else
    echo "Skipping model download (SKIP_MODEL_DOWNLOAD=true)"
fi

echo "=========================================="
echo "Starting Speaches Server..."
echo "=========================================="

# Execute the main command (CMD from Dockerfile)
exec "$@"