stockholm/krebs/5pkgs/simple/stable-generate/default.nix
2024-03-06 11:28:00 +01:00

33 lines
601 B
Nix

{ pkgs, lib, ... }:
pkgs.writers.writeBashBin "stable-generate" ''
set -efu
export PATH=${lib.makeBinPath [
pkgs.coreutils
pkgs.curl
pkgs.jq
]}
STABLE_URL=''${STABLE_URL:-http://stable-confusion.r}
PAYLOAD=$(jq -cn --arg prompt "$*" '{
prompt: $prompt
}')
filename=$(mktemp)
curl -Ssf "$STABLE_URL/sdapi/v1/txt2img" \
-X POST \
--header 'Content-Type: application/json' \
--data "$PAYLOAD" |
jq -r '.images[0]' |
base64 --decode > "$filename"
if test -t 1; then
echo "$filename"
else
cat "$filename"
rm "$filename"
fi
''