if [ -z "$STABILITY_API_KEY" ]; then
    echo "STABILITY_API_KEY environment variable is not set"
    exit 1
fi

OUTPUT_FILE=./out/v1_txt2img.png
BASE_URL=${API_HOST:-https://api.stability.ai}
URL="$BASE_URL/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-image"

curl -f -sS -X POST "$URL" \
  -H 'Content-Type: application/json' \
  -H 'Accept: image/png' \
  -H "Authorization: Bearer $STABILITY_API_KEY" \
  --data-raw '{
    "text_prompts": [
      {
        "text": "A lighthouse on a cliff"
      }
    ],
    "cfg_scale": 7,
    "height": 1024,
    "width": 1024,
    "samples": 1,
    "steps": 30
  }' \
  -o "$OUTPUT_FILE"

import base64
import os
import requests

engine_id = "stable-diffusion-xl-1024-v1-0"
api_host = os.getenv('API_HOST', 'https://api.stability.ai')
api_key = os.getenv("STABILITY_API_KEY")

if api_key is None:
    raise Exception("Missing Stability API key.")

response = requests.post(
    f"{api_host}/v1/generation/{engine_id}/text-to-image",
    headers={
        "Content-Type": "application/json",
        "Accept": "application/json",
        "Authorization": f"Bearer {api_key}"
    },
    json={
        "text_prompts": [
            {
                "text": "A lighthouse on a cliff"
            }
        ],
        "cfg_scale": 7,
        "height": 1024,
        "width": 1024,
        "samples": 1,
        "steps": 30,
    },
)

if response.status_code != 200:
    raise Exception("Non-200 response: " + str(response.text))

data = response.json()

for i, image in enumerate(data["artifacts"]):
    with open(f"./out/v1_txt2img_{i}.png", "wb") as f:
        f.write(base64.b64decode(image["base64"]))

import fetch from 'node-fetch'
import fs from 'node:fs'

const engineId = 'stable-diffusion-xl-1024-v1-0'
const apiHost = process.env.API_HOST ?? 'https://api.stability.ai'
const apiKey = process.env.STABILITY_API_KEY

if (!apiKey) throw new Error('Missing Stability API key.')

const response = await fetch(
  `${apiHost}/v1/generation/${engineId}/text-to-image`,
  {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json',
      Accept: 'application/json',
      Authorization: `Bearer ${apiKey}`,
    },
    body: JSON.stringify({
      text_prompts: [
        {
          text: 'A lighthouse on a cliff',
        },
      ],
      cfg_scale: 7,
      height: 1024,
      width: 1024,
      steps: 30,
      samples: 1,
    }),
  }
)

if (!response.ok) {
  throw new Error(`Non-200 response: ${await response.text()}`)
}

interface GenerationResponse {
  artifacts: Array<{
    base64: string
    seed: number
    finishReason: string
  }>
}

const responseJSON = (await response.json()) as GenerationResponse

responseJSON.artifacts.forEach((image, index) => {
  fs.writeFileSync(
    `./out/v1_txt2img_${index}.png`,
    Buffer.from(image.base64, 'base64')
  )
})

{
  "artifacts": [
    [
      {
        "base64": "...very long string...",
        "finishReason": "SUCCESS",
        "seed": 1050625087
      },
      {
        "base64": "...very long string...",
        "finishReason": "CONTENT_FILTERED",
        "seed": 1229191277
      }
    ]
  ]
}

{
  "id": "296a972f-666a-44a1-a3df-c9c28a1f56c0",
  "name": "bad_request",
  "message": "init_image: is required"
}

{
  "id": "9160aa70-222f-4a36-9eb7-475e2668362a",
  "name": "unauthorized",
  "message": "missing authorization header"
}

{
  "id": "5cf19777-d17f-49fe-9bd9-39ff0ec6bb50",
  "name": "permission_denied",
  "message": "You do not have permission to access this resource"
}

{
  "id": "92b19e7f-22a2-4e71-a821-90edda229293",
  "name": "not_found",
  "message": "The specified engine (ID some-fake-engine) was not found."
}

{
  "id": "f81964d6-619b-453e-97bc-9fd7ac3f04e7",
  "name": "server_error",
  "message": "An unexpected server error occurred, please try again."
}