Image Generation

import base64
from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://api.routerhub.ai/v1",
)

response = client.chat.completions.create(
    model="openai/gpt-image-2",
    messages=[
        {"role": "user", "content": "A cute baby otter wearing a tiny top hat, watercolor style"}
    ],
)

# Find the image part in the response (may be preceded by text)
for part in response.choices[0].message.content:
    if part["type"] == "image_url":
        img_b64 = part["image_url"]["url"].replace("data:image/png;base64,", "")
        img_bytes = base64.b64decode(img_b64)
        with open("otter.png", "wb") as f:
            f.write(img_bytes)
        print("Saved otter.png")
        break

import base64
from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://api.routerhub.ai/v1",
)

response = client.chat.completions.create(
    model="google/gemini-3.1-flash-image-preview",
    messages=[
        {"role": "user", "content": "A serene mountain landscape at sunset, oil painting style"}
    ],
)

# Find the image part in the response (may be preceded by text)
for part in response.choices[0].message.content:
    if part["type"] == "image_url":
        img_b64 = part["image_url"]["url"].replace("data:image/png;base64,", "")
        img_bytes = base64.b64decode(img_b64)
        with open("landscape.png", "wb") as f:
            f.write(img_bytes)
        print("Saved landscape.png")
        break

import requests
import base64

response = requests.post(
    "https://api.routerhub.ai/v1beta/models/gemini-2.5-flash-image:generateContent",
    headers={
        "Authorization": "Bearer YOUR_API_KEY",
        "Content-Type": "application/json",
    },
    json={
        "contents": [
            {
                "role": "user",
                "parts": [{"text": "A cute baby otter wearing a tiny top hat, watercolor style"}],
            }
        ],
    },
)

data = response.json()
for part in data["candidates"][0]["content"]["parts"]:
    if "inlineData" in part:
        img_bytes = base64.b64decode(part["inlineData"]["data"])
        with open("otter.png", "wb") as f:
            f.write(img_bytes)
        print("Saved otter.png")
    elif "text" in part:
        print(part["text"])

import requests
import base64

with open("photo.png", "rb") as f:
    img_b64 = base64.b64encode(f.read()).decode()

response = requests.post(
    "https://api.routerhub.ai/v1beta/models/gemini-3-pro-image-preview:generateContent",
    headers={
        "Authorization": "Bearer YOUR_API_KEY",
        "Content-Type": "application/json",
    },
    json={
        "contents": [
            {
                "role": "user",
                "parts": [
                    {"inlineData": {"mimeType": "image/png", "data": img_b64}},
                    {"text": "Remove the background and replace it with a beach scene"},
                ],
            }
        ],
    },
)

data = response.json()
for part in data["candidates"][0]["content"]["parts"]:
    if "inlineData" in part:
        img_bytes = base64.b64decode(part["inlineData"]["data"])
        with open("edited.png", "wb") as f:
            f.write(img_bytes)
        print("Saved edited.png")

import base64
import requests

response = requests.post(
    "https://api.routerhub.ai/v1/images/generations",
    headers={
        "Authorization": "Bearer YOUR_API_KEY",
        "Content-Type": "application/json",
    },
    json={
        "model": "openai/gpt-image-2",
        "prompt": "A cute baby otter wearing a tiny top hat, watercolor style",
        "size": "1024x1024",
        "quality": "medium",
        "output_compression": 100,
        "output_format": "png",
        "n": 1,
    },
)

response.raise_for_status()
payload = response.json()

img_b64 = payload["data"][0]["b64_json"]
img_bytes = base64.b64decode(img_b64)
with open("generated_image_gpt_image_2.png", "wb") as f:
    f.write(img_bytes)

print("Saved generated_image_gpt_image_2.png")

import base64
import json
import requests

with requests.post(
    "https://api.routerhub.ai/v1/images/generations",
    headers={
        "Authorization": "Bearer YOUR_API_KEY",
        "Content-Type": "application/json",
    },
    json={
        "model": "openai/gpt-image-2",
        "prompt": "A cute baby otter wearing a tiny top hat, watercolor style",
        "size": "1024x1024",
        "stream": True,
        "partial_images": 2,
    },
    stream=True,
) as response:
    response.raise_for_status()
    for raw in response.iter_lines():
        if not raw or not raw.startswith(b"data: "):
            continue
        event = json.loads(raw[len(b"data: "):])
        etype = event.get("type")
        if etype == "error":
            print("Stream error:", event.get("error", {}).get("message", "unknown error"))
            break
        if "b64_json" not in event:
            continue
        img_bytes = base64.b64decode(event["b64_json"])
        if etype == "image_generation.partial_image":
            name = f"otter_partial_{event['partial_image_index']}.png"
        elif etype == "image_generation.completed":
            name = "otter_final.png"
        else:
            continue
        with open(name, "wb") as f:
            f.write(img_bytes)
        print("Saved", name)

import base64
import requests

url = "https://api.routerhub.ai/v1/images/edits"
headers = {
    "Authorization": "Bearer YOUR_API_KEY",
}

data = {
    "model": "openai/gpt-image-2.5-sunburst",
    "prompt": "Add a tiny red party hat on top of the otter's head",
    "size": "1024x1024",
    "quality": "medium",
    "output_format": "png",
    "n": 1,
}

# For multi-image edits, append additional ("image", ("name.png", f2, "image/png")) tuples
with open("otter.png", "rb") as f:
    files = [
        ("image", ("otter.png", f, "image/png")),
    ]
    response = requests.post(url, headers=headers, data=data, files=files)
    response.raise_for_status()
    payload = response.json()

img_b64 = payload["data"][0]["b64_json"]
img_bytes = base64.b64decode(img_b64)
with open("edited_otter.png", "wb") as f:
    f.write(img_bytes)

print("Saved edited_otter.png")

import base64
import json
import requests

url = "https://api.routerhub.ai/v1/images/edits"
headers = {
    "Authorization": "Bearer YOUR_API_KEY",
}

data = {
    "model": "openai/gpt-image-2.5-sunburst",
    "prompt": "Add a tiny red party hat on top of the otter's head",
    "size": "1024x1024",
    "stream": "true",
    "partial_images": 2,
}

with open("otter.png", "rb") as f:
    files = [
        ("image", ("otter.png", f, "image/png")),
    ]
    with requests.post(url, headers=headers, data=data, files=files, stream=True) as response:
        response.raise_for_status()
        for raw in response.iter_lines():
            if not raw or not raw.startswith(b"data: "):
                continue
            event = json.loads(raw[len(b"data: "):])
            etype = event.get("type")
            if etype == "error":
                print("Stream error:", event.get("error", {}).get("message", "unknown error"))
                break
            if "b64_json" not in event:
                continue
            img_bytes = base64.b64decode(event["b64_json"])
            if etype == "image_edit.partial_image":
                name = f"edited_partial_{event['partial_image_index']}.png"
            elif etype == "image_edit.completed":
                name = "edited_final.png"
            else:
                continue
            with open(name, "wb") as f:
                f.write(img_bytes)
            print("Saved", name)