Quick Start: Viggle Image

#!/usr/bin/env bash
set -euo pipefail

response=$(curl -s -X POST "https://apis.viggle.ai/v1/images" \
  -H "Authorization: Bearer $VIGGLE_API_KEY" \
  -F prompt="A watercolor fox reading a book under a maple tree" \
  -F aspect_ratio=16:9 \
  -F resolution=1K \
  -F n=2)
image_id=$(echo "$response" | jq -r '.id')

# A job that hasn't finished within 15 minutes fails on its own.
for _ in $(seq 1 300); do
  job=$(curl -s "https://apis.viggle.ai/v1/images/$image_id" \
    -H "Authorization: Bearer $VIGGLE_API_KEY")
  status=$(echo "$job" | jq -r '.status')

  case "$status" in
    ready)
      i=0
      for url in $(echo "$job" | jq -r '.images[].url'); do
        curl -s -o "image_$i.png" "$url"
        echo "saved image_$i.png"
        i=$((i + 1))
      done
      exit 0
      ;;
    failed)
      echo "Image job failed: $(echo "$job" | jq -c '.error')" >&2
      exit 1
      ;;
  esac
  sleep 3
done

echo "Timed out waiting for image job $image_id" >&2
exit 1

import { writeFile } from "node:fs/promises";

const baseUrl = "https://apis.viggle.ai/v1";
const headers = { Authorization: `Bearer ${process.env.VIGGLE_API_KEY}` };

async function api(path, init = {}) {
  const response = await fetch(`${baseUrl}${path}`, { ...init, headers });
  const body = await response.json().catch(() => ({}));
  if (!response.ok) {
    throw new Error(`HTTP ${response.status}: ${JSON.stringify(body.error ?? body)}`);
  }
  return body;
}

const form = new FormData();
form.append("prompt", "A watercolor fox reading a book under a maple tree");
form.append("aspect_ratio", "16:9");
form.append("resolution", "1K");
form.append("n", "2");

let job = await api("/images", { method: "POST", body: form });

// A job that hasn't finished within 15 minutes fails on its own.
while (job.status === "queued" || job.status === "processing") {
  await new Promise((resolve) => setTimeout(resolve, 3000));
  job = await api(`/images/${job.id}`);
}

if (job.status !== "ready") {
  throw new Error(`Image job failed: ${JSON.stringify(job.error)}`);
}

for (const [i, image] of job.images.entries()) {
  const png = await fetch(image.url);
  await writeFile(`image_${i}.png`, Buffer.from(await png.arrayBuffer()));
  console.log(`saved image_${i}.png (${image.width}x${image.height}, seed ${image.seed})`);
}

import os
import time
import requests

base_url = "https://apis.viggle.ai/v1"
headers = {"Authorization": f"Bearer {os.environ['VIGGLE_API_KEY']}"}

fields = {
    "prompt": "A watercolor fox reading a book under a maple tree",
    "aspect_ratio": "16:9",
    "resolution": "1K",
    "n": "2",
}
# files= makes requests send multipart/form-data, which the endpoint requires.
response = requests.post(
    f"{base_url}/images",
    headers=headers,
    files={name: (None, value) for name, value in fields.items()},
)
response.raise_for_status()
job = response.json()

# A job that hasn't finished within 15 minutes fails on its own.
while job["status"] in ("queued", "processing"):
    time.sleep(3)
    response = requests.get(f"{base_url}/images/{job['id']}", headers=headers)
    response.raise_for_status()
    job = response.json()

if job["status"] != "ready":
    error = job.get("error") or {}
    raise RuntimeError(f"Image job failed: {error.get('code')} {error.get('message')}")

for i, image in enumerate(job["images"]):
    png = requests.get(image["url"])
    png.raise_for_status()
    with open(f"image_{i}.png", "wb") as f:
        f.write(png.content)
    print(f"saved image_{i}.png ({image['width']}x{image['height']}, seed {image['seed']})")

import { Viggle } from "@viggle/sdk";

const viggle = new Viggle({ apiKey: process.env.VIGGLE_API_KEY! });

let job = await viggle.images.create({
  prompt: "A watercolor fox reading a book under a maple tree",
  aspectRatio: "16:9",
  resolution: "1K",
  n: 2,
});
while (job.status === "queued" || job.status === "processing") {
  await new Promise((resolve) => setTimeout(resolve, 3000));
  job = await viggle.images.get(job.id);
}
console.log(job.images.map((image) => image.url));