GPT-Image-2 API 接入教程：5 分钟跑通图像生成（Python 实战）

pip install openai

from openai import OpenAI

client = OpenAI(
    api_key="your-codegateway-api-key",  # CodeGateway API Key
    base_url="https://api.codegateway.dev/v1"
)

response = client.images.generate(
    model="gpt-image-2",
    prompt="一只坐在书堆上的橙色猫咪，在温暖的台灯光下阅读，水彩插画风格",
    n=1,
    size="1024x1024"
)

# 获取图片 URL（60 分钟有效）
image_url = response.data[0].url
print(f"图片 URL: {image_url}")

import base64
import os

response = client.images.generate(
    model="gpt-image-2",
    prompt="现代简约风格的 App 界面截图，蓝色主题，干净的白色背景",
    n=1,
    size="1024x1024",
    quality="high",           # "low" | "medium" | "high"
    output_format="png"       # "jpeg" | "png" | "webp"
)

# 保存 base64 编码的图片
img_data = base64.b64decode(response.data[0].b64_json)
with open("output.png", "wb") as f:
    f.write(img_data)
print("图片已保存到 output.png")

response = client.images.generate(
    model="gpt-image-2",
    prompt="一个简洁的应用程序图标，代表代码网关，深蓝色背景，去除背景",
    n=1,
    size="1024x1024",
    quality="high",
    output_format="png",      # 透明背景必须用 PNG 或 WebP
    background="transparent"  # 开启透明背景
)

import base64
from pathlib import Path

# 读取原始图片和遮罩
with open("original.png", "rb") as f:
    original_b64 = base64.b64encode(f.read()).decode()

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

response = client.images.edit(
    model="gpt-image-2",
    image=original_b64,
    mask=mask_b64,           # 白色区域 = 要修改的部分
    prompt="把背景改成日落海边场景",
    n=1,
    size="1024x1024"
)

response = client.images.generate(
    model="gpt-image-2",
    prompt="科技感强的数据可视化图表，蓝色霓虹灯效果，深色背景",
    n=4,           # 同时生成 4 张变体
    size="1024x1024",
    quality="medium"
)

for i, image_data in enumerate(response.data):
    url = image_data.url
    print(f"变体 {i+1}: {url}")