GPT Image 2 图片生成¶

from openai import OpenAI
import base64
import requests

client = OpenAI(
    api_key="sk_your_access_key",
    base_url="https://cdn.12ai.org/v1",
)

# 图片生成
result = client.images.generate(
    model="gpt-image-2",
    prompt="一只赛博朋克风格的猫",
    n=2,
    size="1024x1024",
    quality="high",
    response_format="b64_json",
)

for index, item in enumerate(result.data, start=1):
    if item.b64_json:
        image_bytes = base64.b64decode(item.b64_json)
    elif item.url:
        image_resp = requests.get(item.url, timeout=120)
        image_resp.raise_for_status()
        image_bytes = image_resp.content
    else:
        continue

    with open(f"output_{index}.png", "wb") as f:
        f.write(image_bytes)

import requests
import base64

resp = requests.post(
    "https://cdn.12ai.org/v1/images/generations",
    headers={
        "Authorization": "Bearer sk_your_access_key",
        "Content-Type": "application/json",
    },
    json={
        "model": "gpt-image-2",
        "prompt": "一只赛博朋克风格的猫",
        "n": 2,
        "size": "1536x1024",
        "quality": "high",
        "response_format": "b64_json",
    },
    timeout=600,
)

for index, item in enumerate(resp.json()["data"], start=1):
    if item.get("b64_json"):
        image_bytes = base64.b64decode(item["b64_json"])
    elif item.get("url"):
        image_resp = requests.get(item["url"], timeout=120)
        image_resp.raise_for_status()
        image_bytes = image_resp.content
    else:
        continue

    with open(f"output_{index}.png", "wb") as f:
        f.write(image_bytes)

import base64
from contextlib import ExitStack
from datetime import datetime
from io import BytesIO

import requests
from PIL import Image

BASE_URL = "https://cdn.12ai.org"
API_KEY = "sk_your_access_key"

# 本地参考图路径，可按需保留一张或传多张。
REFERENCE_IMAGE_PATHS = [
    "reference_primary.png",
    "reference_secondary.png",
]

# 可选：黑白遮罩图路径。留空表示不使用遮罩。
# 黑色区域会转为透明区域，表示需要编辑；白色区域表示保留。
MASK_PATH = ""  # 例如："edit_mask.png"

data = {
    "model": "gpt-image-2",
    "prompt": "根据参考图调整画面内容，并保持整体构图自然一致",
    "n": 2,
    "size": "1024x1024",
    "quality": "high",
    "response_format": "b64_json",  # 也可以改为 "url"
}

with ExitStack() as stack:
    files = []
    for index, image_path in enumerate(REFERENCE_IMAGE_PATHS, start=1):
        image_file = stack.enter_context(open(image_path, "rb"))
        files.append(("image", (f"reference_{index}.png", image_file, "image/png")))

    if MASK_PATH:
        mask = Image.open(MASK_PATH).convert("L")
        mask_rgba = Image.new("RGBA", mask.size, (255, 255, 255, 255))
        mask_rgba.putalpha(mask)

        mask_buffer = BytesIO()
        stack.callback(mask_buffer.close)
        mask_rgba.save(mask_buffer, format="PNG")
        mask_buffer.seek(0)
        files.append(("mask", ("mask.png", mask_buffer, "image/png")))

    resp = requests.post(
        f"{BASE_URL}/v1/images/edits",
        headers={"Authorization": f"Bearer {API_KEY}"},
        data=data,
        files=files,
        timeout=600,
    )

resp.raise_for_status()
result = resp.json()

for index, item in enumerate(result.get("data", []), start=1):
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"edited_image_{timestamp}_{index}.png"

    if item.get("b64_json"):
        image_bytes = base64.b64decode(item["b64_json"])
    elif item.get("url"):
        image_resp = requests.get(item["url"], timeout=120)
        image_resp.raise_for_status()
        image_bytes = image_resp.content
    else:
        continue

    with open(filename, "wb") as f:
        f.write(image_bytes)
    print(f"图片已保存：{filename}")