setup-models.sh

#!/usr/bin/env bash

# Prepare the local Gemma 4 LiteRT model used by the Android competition build.
# The model is intentionally excluded from git because it is a multi-GB binary.

set -euo pipefail

PROJECT_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
MODEL_DIR="$PROJECT_ROOT/assets/models"
MODEL_FILE="$MODEL_DIR/gemma-4-E2B-it.litertlm"
MIN_EXPECTED_BYTES=$((2 * 1024 * 1024 * 1024))

print_usage() {
  cat <<'EOF'
Usage:
  ./setup-models.sh
  ./setup-models.sh /absolute/path/to/gemma-4-E2B-it.litertlm
  DARKGRIDAI_GEMMA_MODEL=/absolute/path/to/gemma-4-E2B-it.litertlm ./setup-models.sh

The Android release build expects:
  assets/models/gemma-4-E2B-it.litertlm

This file is not committed to git. Keep it beside the submission package,
on external media, or in your model artifact store, then run this script.
EOF
}

file_size() {
  stat -f%z "$1" 2>/dev/null || stat -c%s "$1"
}

verify_model() {
  local file="$1"
  local size
  size="$(file_size "$file")"

  if [ "$size" -lt "$MIN_EXPECTED_BYTES" ]; then
    echo "Model file looks too small: $file ($size bytes)"
    echo "Expected the Gemma 4 LiteRT model, which is over 2 GB."
    exit 1
  fi

  echo "Gemma 4 model ready: $file"
  echo "Size: $size bytes"
}

mkdir -p "$MODEL_DIR"

SOURCE_MODEL="${1:-${DARKGRIDAI_GEMMA_MODEL:-}}"

if [ -n "$SOURCE_MODEL" ]; then
  if [ ! -f "$SOURCE_MODEL" ]; then
    echo "Model source not found: $SOURCE_MODEL"
    print_usage
    exit 1
  fi

  if [ "$SOURCE_MODEL" != "$MODEL_FILE" ]; then
    echo "Copying Gemma 4 model into assets/models..."
    cp "$SOURCE_MODEL" "$MODEL_FILE"
  fi

  verify_model "$MODEL_FILE"
  exit 0
fi

if [ -f "$MODEL_FILE" ]; then
  verify_model "$MODEL_FILE"
  exit 0
fi

echo "Gemma 4 model is missing."
print_usage
exit 1