Python Image Generation

from openai import OpenAI

client = OpenAI(api_key="sk-xxx", base_url="https://api.crazyrouter.com/v1")

response = client.images.generate(
    model="dall-e-3",
    prompt="A cat wearing a spacesuit walking on the moon, digital art style",
    size="1024x1024",
    quality="hd",
    n=1
)

image_url = response.data[0].url
print(f"Image URL: {image_url}")

response = client.images.generate(
    model="gpt-image-2",
    prompt="A modern minimalist logo with an artificial intelligence theme",
    size="1024x1024",
    quality="low",
    output_format="png",
    n=1
)

print(response.data[0].url)

# Image editing (requires original image)
response = client.images.edit(
    model="gpt-image-2",
    image=open("original.png", "rb"),
    prompt="Change the background to a beach sunset",
    size="1024x1024",
    quality="low"
)

print(response.data[0].url)

import requests

response = client.images.generate(
    model="dall-e-3",
    prompt="Cyberpunk city nightscape",
    size="1792x1024"
)

# Download image
image_url = response.data[0].url
img_data = requests.get(image_url).content
with open("cyberpunk_city.png", "wb") as f:
    f.write(img_data)
print("Image saved")