Image Generation

import base64
from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://api.pomex.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.pomex.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.pomex.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.pomex.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.pomex.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")