#!/bin/sh

set -e

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

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

curl -f -sS -X POST "$URL" \
  -H 'Content-Type: multipart/form-data' \
  -H 'Accept: image/png' \
  -H "Authorization: Bearer $STABILITY_API_KEY" \
  -F 'init_image=@"../init_image.png"' \
  -F 'mask_image=@"../mask_image_black.png"' \
  -F 'mask_source=MASK_IMAGE_BLACK' \
  -F 'text_prompts[0][text]=A large spiral galaxy with a bright central bulge and a ring of stars around it' \
  -F 'cfg_scale=7' \
  -F 'clip_guidance_preset=FAST_BLUE' \
  -F 'samples=1' \
  -F '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}/image-to-image/masking",
    headers={
        "Accept": 'application/json',
        "Authorization": f"Bearer {api_key}"
    },
    files={
        'init_image': open("../init_image.png", 'rb'),
        'mask_image': open("../mask_image_black.png", 'rb'),
    },
    data={
        "mask_source": "MASK_IMAGE_BLACK",
        "text_prompts[0][text]": "A large spiral galaxy with a bright central bulge and a ring of stars around it",
        "cfg_scale": 7,
        "clip_guidance_preset": "FAST_BLUE",
        "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_img2img_masking_{i}.png", "wb") as f:
        f.write(base64.b64decode(image["base64"]))

import fetch from 'node-fetch'
import FormData from 'form-data'
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.')

// NOTE: This example is using a NodeJS FormData library. Browser
// implementations should use their native FormData class. React Native
// implementations should also use their native FormData class.
const formData = new FormData()
formData.append('init_image', fs.readFileSync('../init_image.png'))
formData.append('mask_image', fs.readFileSync('../mask_image_black.png'))
formData.append('mask_source', 'MASK_IMAGE_BLACK')
formData.append(
  'text_prompts[0][text]',
  'A large spiral galaxy with a bright central bulge and a ring of stars around it'
)
formData.append('cfg_scale', '7')
formData.append('clip_guidance_preset', 'FAST_BLUE')
formData.append('samples', 1)
formData.append('steps', 30)

const response = await fetch(
  `${apiHost}/v1/generation/${engineId}/image-to-image/masking`,
  {
    method: 'POST',
    headers: {
      ...formData.getHeaders(),
      Accept: 'application/json',
      Authorization: `Bearer ${apiKey}`,
    },
    body: formData,
  }
)

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_img2img_masking_${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."
}