#!/bin/sh
set -eu

if [ -d "/data" ] && [ -w "/data" ]; then
  CACHE_ROOT="/data/.cache/huggingface"
  OUTPUT_ROOT="/data/aibom_output"
else
  CACHE_ROOT="/tmp/.cache/huggingface"
  OUTPUT_ROOT="/tmp/aibom_output"
fi

mkdir -p "${CACHE_ROOT}" "${OUTPUT_ROOT}"

export HF_HOME="${HF_HOME:-${CACHE_ROOT}}"
export TRANSFORMERS_CACHE="${TRANSFORMERS_CACHE:-${CACHE_ROOT}/transformers}"
export AIBOM_OUTPUT_DIR="${AIBOM_OUTPUT_DIR:-${OUTPUT_ROOT}}"
export PORT="${PORT:-7860}"

mkdir -p "${TRANSFORMERS_CACHE}" "${AIBOM_OUTPUT_DIR}"

if [ "$#" -gt 0 ]; then
  exec python -m src.cli "$@"
fi

exec uvicorn src.main:app --host 0.0.0.0 --port "${PORT}"