mirror of
https://github.com/theniceboy/.config.git
synced 2026-07-16 22:01:21 +08:00
imggen tool
This commit is contained in:
parent
fc4221baf8
commit
d05a33d1ba
1 changed files with 93 additions and 0 deletions
93
bin/imggen
Executable file
93
bin/imggen
Executable file
|
|
@ -0,0 +1,93 @@
|
|||
#!/bin/bash
|
||||
set -euo pipefail
|
||||
|
||||
OPEN=false
|
||||
IMAGE_ARGS=()
|
||||
PROMPT=""
|
||||
|
||||
while [ $# -gt 0 ]; do
|
||||
case "$1" in
|
||||
--open)
|
||||
OPEN=true
|
||||
shift
|
||||
;;
|
||||
--image|-i)
|
||||
IMAGE_ARGS+=("$2")
|
||||
shift 2
|
||||
;;
|
||||
-h|--help)
|
||||
echo "usage: imggen [--open] [--image <path>]... <prompt>" >&2
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
PROMPT="$1"
|
||||
shift
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
if [ -z "$PROMPT" ]; then
|
||||
echo "usage: imggen [--open] [--image <path>]... <prompt>" >&2
|
||||
exit 2
|
||||
fi
|
||||
|
||||
SCRIPT="$HOME/.config/opencode/tools/codex-imagen/scripts/codex-imagen.mjs"
|
||||
TMP=$(mktemp)
|
||||
trap 'rm -f "$TMP"' EXIT
|
||||
|
||||
if [ ${#IMAGE_ARGS[@]} -gt 0 ]; then
|
||||
printf 'edit image based on this prompt: "%s"' "$PROMPT" > "$TMP"
|
||||
else
|
||||
printf 'generate image: "%s"' "$PROMPT" > "$TMP"
|
||||
fi
|
||||
|
||||
CMD=(node "$SCRIPT" --prompt-file "$TMP" --json --quiet --out-dir /tmp --model gpt-5.5)
|
||||
if [ ${#IMAGE_ARGS[@]} -gt 0 ]; then
|
||||
for img in "${IMAGE_ARGS[@]}"; do
|
||||
CMD+=(-i "$img")
|
||||
done
|
||||
fi
|
||||
|
||||
if [ "${IMGGEN_DEBUG:-}" = "1" ]; then
|
||||
echo "[debug] prompt file: $TMP" >&2
|
||||
echo "[debug] cmd: ${CMD[*]}" >&2
|
||||
OUT=$("${CMD[@]}" 2>&1)
|
||||
echo "[debug] raw output:" >&2
|
||||
echo "$OUT" >&2
|
||||
else
|
||||
OUT=$("${CMD[@]}" 2>/dev/null)
|
||||
fi
|
||||
|
||||
if [ -z "$OUT" ]; then
|
||||
echo "error: no output from codex-imagen (set IMGGEN_DEBUG=1 for details)" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PATH_OUT=$(echo "$OUT" | python3 -c "
|
||||
import sys, json
|
||||
try:
|
||||
d = json.load(sys.stdin)
|
||||
if d.get('images'):
|
||||
print(d['images'][0]['path'])
|
||||
else:
|
||||
sys.stderr.write('error: no images in response\n')
|
||||
sys.stderr.write(json.dumps(d)[:500] + '\n')
|
||||
sys.exit(1)
|
||||
except json.JSONDecodeError as e:
|
||||
sys.stderr.write('error: invalid JSON: ' + str(e) + '\n')
|
||||
sys.stderr.write(sys.stdin.read()[:500] + '\n')
|
||||
sys.exit(1)
|
||||
" 2>&1)
|
||||
|
||||
if [ $? -ne 0 ]; then
|
||||
echo "error: failed to parse output" >&2
|
||||
echo "$PATH_OUT" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ -n "$PATH_OUT" ]; then
|
||||
echo "$PATH_OUT"
|
||||
if [ "$OPEN" = true ]; then
|
||||
open "$PATH_OUT"
|
||||
fi
|
||||
fi
|
||||
Loading…
Add table
Add a link
Reference in a new issue