55 lines
1.4 KiB
Bash
Executable File
55 lines
1.4 KiB
Bash
Executable File
#!/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}"
|
|
export IMAGE_STREAM="${IMAGE_STREAM:-true}"
|
|
export IMAGE_PARTIAL_IMAGES="${IMAGE_PARTIAL_IMAGES:-1}"
|
|
|
|
# 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 "$@"
|