import os
import time

import requests

API_KEY = os.environ["BFL_API_KEY"]


def generate(payload, name="flux3_image"):
    submit = requests.post("https://api.bfl.ai/v1/flux-3-image",
                           headers={"x-key": API_KEY}, json=payload, timeout=60)
    submit.raise_for_status()
    polling_url = submit.json()["polling_url"]

    while True:
        time.sleep(1.5)
        result = requests.get(polling_url, headers={"x-key": API_KEY}, timeout=60).json()
        if result["status"] not in ("Pending", "Reasoning", "Generating"):
            break
    if result["status"] != "Ready":
        raise RuntimeError(result)

    image = requests.get(result["result"]["sample"], timeout=120)
    with open(name + ".png", "wb") as f:
        f.write(image.content)
    return name + ".png"


generate({
    "prompt": "A two-color risograph narrative illustration with porous ink, soft misregistration, and cheerful simple shapes: a rabbit rides a pumpkin along a winding path through three layers of rolling hills. No text.",
    "aspect_ratio": "16:9",
})

// Node.js 18+
import { writeFile } from "node:fs/promises";

const headers = { "x-key": process.env.BFL_API_KEY, "Content-Type": "application/json" };

const submit = await fetch("https://api.bfl.ai/v1/flux-3-image", {
  method: "POST",
  headers,
  body: JSON.stringify({
    prompt: "A two-color risograph narrative illustration with porous ink, soft misregistration, and cheerful simple shapes: a rabbit rides a pumpkin along a winding path through three layers of rolling hills. No text.",
    aspect_ratio: "16:9",
  }),
});
if (!submit.ok) throw new Error(await submit.text());
const { polling_url } = await submit.json();

let result;
do {
  await new Promise((resolve) => setTimeout(resolve, 1500));
  result = await (await fetch(polling_url, { headers })).json();
} while (["Pending", "Reasoning", "Generating"].includes(result.status));
if (result.status !== "Ready") throw new Error(JSON.stringify(result));

const image = await fetch(result.result.sample);
await writeFile("flux3_image.png", Buffer.from(await image.arrayBuffer()));

POLLING_URL=$(curl -s --fail-with-body https://api.bfl.ai/v1/flux-3-image \
  -H "x-key: $BFL_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"prompt": "A two-color risograph narrative illustration with porous ink, soft misregistration, and cheerful simple shapes: a rabbit rides a pumpkin along a winding path through three layers of rolling hills. No text.", "aspect_ratio": "16:9"}' | jq -r .polling_url)

while true; do
  RESULT=$(curl -s "$POLLING_URL" -H "x-key: $BFL_API_KEY")
  STATUS=$(echo "$RESULT" | jq -r .status)
  [[ "$STATUS" =~ ^(Pending|Reasoning|Generating)$ ]] || break
  sleep 1.5
done

if [ "$STATUS" = "Ready" ]; then
  curl -s -o flux3_image.png "$(echo "$RESULT" | jq -r .result.sample)"
else
  echo "$RESULT"
fi