29 lines
650 B
Bash
Executable File
29 lines
650 B
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
PROMPT="$1"
|
|
SERVER_IP="192.168.2.154"
|
|
API_URL="http://${SERVER_IP}:5005/v1/images/generations"
|
|
|
|
# Note: Using "flux-schnell" to match our api.py logic
|
|
RESPONSE=$(curl -s -X POST "$API_URL" \
|
|
-H "Content-Type: application/json" \
|
|
-d "{
|
|
\"prompt\": \"$PROMPT\",
|
|
\"model\": \"flux-schnell\"
|
|
}")
|
|
|
|
BASE64=$(echo "$RESPONSE" | jq -r '.data[0].b64_json')
|
|
|
|
if [ "$BASE64" == "null" ]; then
|
|
echo "Error: $(echo "$RESPONSE" | jq -r .detail)"
|
|
exit 1
|
|
fi
|
|
|
|
OUTDIR="downloads/flux_images"
|
|
mkdir -p "$OUTDIR"
|
|
OUTFILE="${OUTDIR}/$(date +%s).png"
|
|
|
|
echo "$BASE64" | base64 -d > "$OUTFILE"
|
|
echo "Image generated: $OUTFILE"
|