FLUX.2 图片编辑 API：单图修改与多图合成

"""FLUX.2 图片编辑：python flux_edit.py "编辑指令" 图片1 [图片2 ...]"""
import base64
import mimetypes
import os
import sys
from pathlib import Path

import requests

API_URL = "https://api.laozhang.ai/v1/images/generations"
HEADERS = {"Authorization": f"Bearer {os.environ['LAOZHANG_API_KEY']}"}
MODEL = "flux-2-pro"         # 也可以用 flux-2-flex 或 flux-2-max
WIDTH, HEIGHT = 1024, 1024   # 输出尺寸，宽高须为 16 的倍数


def to_reference(source):
    """公网 URL 和 data URL 原样使用，本地文件转成 data URL。"""
    if source.startswith(("http://", "https://", "data:")):
        return source
    path = Path(source)
    mime_type = mimetypes.guess_type(path.name)[0] or "image/png"
    return f"data:{mime_type};base64,{base64.b64encode(path.read_bytes()).decode('ascii')}"


def save_result(item, output):
    """下载结果图。FLUX 只返回 url，链接约 10 分钟后失效。"""
    if not item.get("url"):
        raise SystemExit(f"返回结果中没有图片：{item}")
    image = requests.get(item["url"], timeout=60)
    image.raise_for_status()
    Path(output).write_bytes(image.content)
    print("已保存", output)


if not 1 <= len(sys.argv) - 2 <= 8:
    raise SystemExit('用法：python flux_edit.py "编辑指令" 图片1 [图片2 ...]，最多 8 张')

payload = {
    "model": MODEL,
    "prompt": sys.argv[1],
    "width": WIDTH,
    "height": HEIGHT,
    "output_format": "png",
}
for index, source in enumerate(sys.argv[2:], start=1):
    payload["input_image" if index == 1 else f"input_image_{index}"] = to_reference(source)

response = requests.post(API_URL, headers=HEADERS, json=payload, timeout=240)
if response.status_code != 200:
    raise SystemExit(f"HTTP {response.status_code}：{response.text[:1000]}")
save_result(response.json()["data"][0], "flux-edit.png")