Skip to main content

Image Generation (gpt-image-2)

from openai import OpenAI
import base64

client = OpenAI(api_key="your-api-key", base_url="https://aisupermarket.work/v1")

response = client.images.generate(
model="gpt-image-2",
prompt="An orange cat reading in a cafe, watercolor style",
size="1024x1024",
n=1,
)

# gpt-image-2 returns base64
image_bytes = base64.b64decode(response.data[0].b64_json)
with open("output.png", "wb") as f:
f.write(image_bytes)