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

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

if [[ ! -f .env ]]; then
  echo ".env not found" >&2
  exit 1
fi

while IFS= read -r line; do
  [[ -n "$line" ]] && export "$line"
done < <(
  awk '
    $0 == "# https://api.aoixx.com" { in_block = 1; next }
    in_block && /^#/ { exit }
    in_block && /^OPENAI_API_KEY=/ { print; next }
  ' .env
)

unset http_proxy https_proxy all_proxy HTTP_PROXY HTTPS_PROXY ALL_PROXY

if [[ -z "${OPENAI_API_KEY:-}" ]]; then
  echo "aoixx OPENAI_API_KEY not found in .env" >&2
  exit 1
fi

export OPENAI_BASE_URL="https://api.aoixx.com"
export IMAGE_REQUEST_STYLE="${IMAGE_REQUEST_STYLE:-apicodex}"
export IMAGE_SIZE="${IMAGE_SIZE:-3840x2160}"
export IMAGE_MODEL="${IMAGE_MODEL:-gpt-image-2}"
export IMAGE_REQUEST_TIMEOUT="${IMAGE_REQUEST_TIMEOUT:-900}"
export IMAGE_MAX_POLL_WAIT="${IMAGE_MAX_POLL_WAIT:-1800}"
export IMAGE_INITIAL_POLL_DELAY="${IMAGE_INITIAL_POLL_DELAY:-10}"
export IMAGE_POLL_INTERVAL="${IMAGE_POLL_INTERVAL:-5}"

# Direct generation tests should ignore any edit inputs coming from .env.
for name in \
  REFERENCE_IMAGE \
  REFERENCE_IMAGE_FILE \
  REFERENCE_IMAGE_URL \
  REFERENCE_IMAGE_DATA_URL \
  MASK_IMAGE \
  MASK_IMAGE_FILE \
  MASK_IMAGE_URL \
  MASK_IMAGE_DATA_URL
do
  export "$name="
done

python3 gpt_image2_generate.py "$@"