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[] 항목 하나의 이미지 바이트를 반환합니다."""
    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>.<ext>, <stem>-2.<ext> 순서로 저장합니다."""
    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