GPT Image 2 API：令牌分组、计费、生成与编辑

export LAOZHANG_API_KEY="替换为你的老张API密钥"
export BASE_URL="https://api.laozhang.ai/v1"
python -m pip install pillow

curl --fail-with-body "$BASE_URL/images/generations" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-image-2",
    "prompt": "生成一张白色陶瓷马克杯放在灰色桌面上的产品图，柔和自然光，简洁背景"
  }' \
  -o generation.json

curl --fail-with-body "$BASE_URL/images/generations" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-image-2-vip",
    "prompt": "生成一张白色陶瓷马克杯放在灰色桌面上的产品图，柔和自然光，简洁背景",
    "size": "2048x2048",
    "quality": "high"
  }' \
  -o generation.json

curl --fail-with-body "$BASE_URL/images/generations" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-image-2",
    "prompt": "生成一张白色陶瓷马克杯放在灰色桌面上的产品图，柔和自然光，简洁背景",
    "size": "1536x1024",
    "quality": "high"
  }' \
  -o generation.json

curl --fail-with-body "$BASE_URL/images/edits" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -F "model=gpt-image-2" \
  -F "prompt=使用提供的图片作为源图。保留主体、构图和文字，只把杯身贴纸改成红色，并给杯口加一条很细的金色边。" \
  -F "image=@source.png" \
  -o edit.json

curl --fail-with-body "$BASE_URL/images/edits" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -F "model=gpt-image-2-vip" \
  -F "prompt=使用提供的图片作为源图。保留主体、构图和文字，只把贴纸改成蓝色。" \
  -F "image=@source.png" \
  -o edit.json

curl --fail-with-body "$BASE_URL/images/edits" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -F "model=gpt-image-2" \
  -F "prompt=使用提供的图片作为源图。保留主体、构图和文字，只把贴纸改成绿色。" \
  -F "image=@source.png" \
  -F "size=1024x1024" \
  -F "quality=high" \
  -o edit.json

import base64
import io
import json
import sys
import urllib.request
from pathlib import Path
from PIL import Image

if len(sys.argv) != 3:
    raise SystemExit("用法：python save_images.py 响应文件.json 输出文件前缀")


def decode_b64(value):
    # 去掉可能存在的 data:image/...;base64, 前缀，并补齐缺失的 = 填充
    if value.startswith("data:"):
        value = value.split(",", 1)[1]
    value = "".join(value.split())
    value += "=" * (-len(value) % 4)
    return base64.b64decode(value, validate=True)


response_path = Path(sys.argv[1])
output_prefix = sys.argv[2]
payload = json.loads(response_path.read_text(encoding="utf-8"))
if payload.get("error"):
    raise RuntimeError(payload["error"])
items = payload.get("data") or []
if not items:
    raise RuntimeError("响应中没有图片结果，请检查错误信息和调用日志")

for index, item in enumerate(items, start=1):
    if item.get("b64_json"):
        raw = decode_b64(item["b64_json"])
    elif item.get("url"):
        with urllib.request.urlopen(item["url"], timeout=60) as download:
            raw = download.read()
    else:
        raise RuntimeError(f"第 {index} 项缺少图片内容或下载地址")

    with Image.open(io.BytesIO(raw)) as picture:
        picture.load()
        extension = {"PNG": "png", "JPEG": "jpg", "WEBP": "webp"}.get(
            picture.format
        )
        if extension is None:
            raise RuntimeError(f"无法保存图片格式：{picture.format}")
        output = Path(f"{output_prefix}-{index}.{extension}")
        output.write_bytes(raw)
        print(f"已保存 {output}，尺寸 {picture.width} × {picture.height}")

python save_images.py generation.json generated
python save_images.py edit.json edited