LeonardoAI × Claude

Create tools/leonardo.py with this code and set the API key to YOUR_API_KEY:

#!/usr/bin/env python3
import sys, time, json, urllib.request

API_KEY = "PASTE_YOUR_KEY_HERE"
V2_URL  = "https://cloud.leonardo.ai/api/rest/v2"
V1_URL  = "https://cloud.leonardo.ai/api/rest/v1"
MODEL   = "gemini-image-2"        # Nano Banana Pro

HEADERS = {
    "accept": "application/json",
    "content-type": "application/json",
    "authorization": f"Bearer {API_KEY}",
}

def post(base, path, body):
    req = urllib.request.Request(f"{base}{path}", data=json.dumps(body).encode(), headers=HEADERS, method="POST")
    with urllib.request.urlopen(req) as res: return json.loads(res.read())

def get(base, path):
    req = urllib.request.Request(f"{base}{path}", headers=HEADERS)
    with urllib.request.urlopen(req) as res: return json.loads(res.read())

def generate(prompt):
    print(f"Generating: {prompt!r}")
    resp = post(V2_URL, "/generations", {
        "model": MODEL,
        "parameters": {"prompt": prompt, "width": 1024, "height": 1024, "quantity": 1, "prompt_enhance": "ON"},
        "public": False,
    })
    gen_id = resp["generate"]["generationId"]
    cost   = resp["generate"]["cost"]["amount"]
    print(f"Job ID: {gen_id} | Est. cost: ${cost}")
    for _ in range(30):
        time.sleep(5)
        data   = get(V1_URL, f"/generations/{gen_id}")
        status = data["generations_by_pk"]["status"]
        print(f"  status: {status}")
        if status == "COMPLETE":
            return [img["url"] for img in data["generations_by_pk"]["generated_images"]]
        if status == "FAILED":
            raise RuntimeError("Generation failed.")
    raise TimeoutError("Timed out after 150 seconds.")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print('Usage: python3 tools/leonardo.py "your prompt here"')
        sys.exit(1)
    urls = generate(" ".join(sys.argv[1:]))
    print("\nDone! Image URL(s):")
    for url in urls: print(f"  {url}")

Add this to my CLAUDE.md: every time I ask to generate an image, use Leonardo AI with Nano Banana Pro by running tools/leonardo.py. Do not use any other tool or rewrite the script.