GPT Image 2.5 图像编辑

import base64
import os
from pathlib import Path
import requests

encoded = base64.b64encode(Path("reference.png").read_bytes()).decode("ascii")
response = requests.post(
    "https://code.heihuzi.ai/v1/images/edits",
    headers={"Authorization": f"Bearer {os.environ['HEIHUZI_API_KEY']}"},
    json={
        "model": "gpt-image-2.5-flare",
        "prompt": "Change the blue cup to a bright red cup. Keep the cup shape and the white background.",
        "images": [{"image_url": f"data:image/png;base64,{encoded}"}],
        "n": 1,
        "size": "1024x1024",
        "quality": "low",
        "output_format": "png",
    },
    timeout=(15, 600),
)
response.raise_for_status()
payload = response.json()
if payload.get("error") or not payload.get("data"):
    raise RuntimeError(payload.get("error", "No image returned"))
for index, item in enumerate(payload["data"], 1):
    Path(f"edited-json-{index}.png").write_bytes(base64.b64decode(item["b64_json"]))
print("Saved", len(payload["data"]), "image(s)")

curl --fail-with-body --silent --show-error --max-time 600 \
  --request POST \
  --url https://code.heihuzi.ai/v1/images/edits \
  --header "Authorization: Bearer $HEIHUZI_API_KEY" \
  --header 'Content-Type: application/json' \
  --data '{
    "model": "gpt-image-2.5-flare",
    "prompt": "Change the blue cup to a bright red cup. Keep the cup shape and the white background.",
    "images": [{"image_url": "https://docs.heihuzi.ai/images/reference.png"}],
    "n": 1,
    "size": "1024x1024",
    "quality": "low",
    "output_format": "png"
  }' \
  --output edited-url.json \
  --write-out '%{http_code}\n'

curl --fail-with-body --silent --show-error --max-time 600 \
  --request POST \
  --url https://code.heihuzi.ai/v1/images/edits \
  --header "Authorization: Bearer $HEIHUZI_API_KEY" \
  --form 'model=gpt-image-2.5-flare' \
  --form 'prompt=Use the shape of the cup from the first image and the red color from the second image. White background.' \
  --form 'image[]=@reference.png' \
  --form 'image[]=@palette.png' \
  --form 'n=1' \
  --form 'size=1024x1024' \
  --form 'quality=low' \
  --form 'output_format=png' \
  --output edited-multi.json \
  --write-out '%{http_code}\n'

curl --fail-with-body --silent --show-error --max-time 600 \
  --request POST \
  --url https://code.heihuzi.ai/v1/images/edits \
  --header "Authorization: Bearer $HEIHUZI_API_KEY" \
  --form 'model=gpt-image-2.5-sunburst' \
  --form 'prompt=Add a small red apple on the right side of the blue cup, inside the transparent masked region. Keep the blue cup.' \
  --form 'image=@reference.png' \
  --form 'mask=@mask.png' \
  --form 'n=1' \
  --form 'size=1024x1024' \
  --form 'quality=low' \
  --form 'output_format=png' \
  --output edited-mask.json \
  --write-out '%{http_code}\n'

import base64
from pathlib import Path
import os
from openai import OpenAI

client = OpenAI(
    api_key=os.environ["HEIHUZI_API_KEY"],
    base_url="https://code.heihuzi.ai/v1",
    timeout=600.0,
    max_retries=0,
)

completed = 0
with open("reference.png", "rb") as image:
    events = client.images.edit(
        model="gpt-image-2.5-sunburst",
        image=image,
        prompt="Change the blue cup to a bright red cup. Keep the cup shape and the white background.",
        n=1,
        size="1024x1024",
        quality="low",
        output_format="png",
        stream=True,
        partial_images=1,
    )
    for event in events:
        print(event.type, flush=True)
        if event.type == "image_edit.partial_image":
            Path(f"edited-preview-{event.partial_image_index}.png").write_bytes(
                base64.b64decode(event.b64_json)
            )
        elif event.type == "image_edit.completed":
            completed += 1
            Path(f"edited-final-{completed}.png").write_bytes(base64.b64decode(event.b64_json))
if completed != 1:
    raise RuntimeError(f"Expected 1 final image, got {completed}")