Images APIの使い方：画像の生成と編集

export LAOZHANG_API_KEY="YOUR_LAOZHANG_API_KEY"

python -m pip install openai pillow httpx

"""OpenAI Images APIのレスポンスに含まれるすべての画像を保存する。"""
import base64
import io
import json
import sys
from pathlib import Path

import httpx
from PIL import Image

EXTENSIONS = {"PNG": "png", "JPEG": "jpg", "WEBP": "webp"}


def image_bytes(item):
    """data[] の1エントリーから画像のバイト列を返す。"""
    encoded = item.get("b64_json")
    if encoded:
        encoded = "".join(encoded.split())
        if encoded.startswith("data:"):
            encoded = encoded.split(",", 1)[1]  # data:image/...;base64, の接頭辞を取り除く
        encoded += "=" * (-len(encoded) % 4)  # 欠けたパディングを補う
        return base64.b64decode(encoded)
    if item.get("url"):
        # LaoZhang APIのキーは送らずに結果をダウンロードする
        download = httpx.get(item["url"], timeout=120, follow_redirects=True)
        download.raise_for_status()
        return download.content
    raise ValueError("data[] のエントリーに b64_json も url もありません")


def save_images(response, stem):
    """各画像を <stem>.<拡張子>、<stem>-2.<拡張子> … の順に保存する。"""
    entries = response.get("data") or []
    if not entries:
        raise RuntimeError(f"レスポンスに画像がありません: {json.dumps(response)[:500]}")
    for number, item in enumerate(entries, start=1):
        raw = image_bytes(item)
        with Image.open(io.BytesIO(raw)) as image:
            image.load()
            extension = EXTENSIONS.get(image.format, (image.format or "bin").lower())
            size = f"{image.width} × {image.height}"
        suffix = "" if number == 1 else f"-{number}"
        output = Path(f"{stem}{suffix}.{extension}")
        output.write_bytes(raw)
        print(f"保存しました: {output}（{size}）")


if __name__ == "__main__":
    save_images(json.loads(Path(sys.argv[1]).read_text(encoding="utf-8")), sys.argv[2])

POST https://api.laozhang.ai/v1/images/generations

POST https://api.laozhang.ai/v1/images/edits