FLUX.2 API: редактирование изображений и до 8 референсов

curl --fail-with-body --max-time 300 "https://api.laozhang.ai/v1/images/generations" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "flux-2-pro",
    "prompt": "Сохрани тот же главный объект. Замени фон на чистый бледно-голубой студийный фон и сохрани исходное освещение.",
    "input_image": "https://example.com/source.png",
    "width": 1792,
    "height": 1024,
    "output_format": "png"
  }' \
  -o flux-edit.json

python3 save_images.py flux-edit.json flux-edit

IMG=$(base64 < product.png | tr -d '\n')

curl --fail-with-body --max-time 300 "https://api.laozhang.ai/v1/images/generations" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -H "Content-Type: application/json" \
  --data-binary @- \
  -o flux-local-edit.json <<EOF
{
  "model": "flux-2-pro",
  "prompt": "Поставь товар на мраморную столешницу в мягком утреннем свете. Сам товар и его этикетку не меняй.",
  "input_image": "data:image/png;base64,$IMG",
  "size": "1024x1024",
  "output_format": "png"
}
EOF

python3 save_images.py flux-local-edit.json flux-product

IMG1=$(base64 < robot.png | tr -d '\n')
IMG2=$(base64 < greenhouse.png | tr -d '\n')
IMG3=$(base64 < skateboard.png | tr -d '\n')

curl --fail-with-body --max-time 300 "https://api.laozhang.ai/v1/images/generations" \
  -H "Authorization: Bearer $LAOZHANG_API_KEY" \
  -H "Content-Type: application/json" \
  --data-binary @- \
  -o flux-multi-edit.json <<EOF
{
  "model": "flux-2-pro",
  "prompt": "Объедини три референса в одно цельное новое изображение. Изображение 1 — красный робот, изображение 2 — фон с бирюзовой стеклянной оранжереей, изображение 3 — жёлтый скейтборд. Используй детали всех трёх референсов. Не делай коллаж и не дели кадр на части.",
  "input_image": "data:image/png;base64,$IMG1",
  "input_image_2": "data:image/png;base64,$IMG2",
  "input_image_3": "data:image/png;base64,$IMG3",
  "width": 1792,
  "height": 1024,
  "output_format": "png"
}
EOF

python3 save_images.py flux-multi-edit.json flux-multi-edit

import base64
import mimetypes
import os
from pathlib import Path

import requests

API_URL = "https://api.laozhang.ai/v1/images/generations"
HEADERS = {"Authorization": f"Bearer {os.environ['LAOZHANG_API_KEY']}"}


def to_image_input(source):
    """URL возвращаем как есть, локальный файл превращаем в data URL."""
    if source.startswith(("https://", "http://", "data:")):
        return source
    path = Path(source)
    mime_type = mimetypes.guess_type(path.name)[0] or "image/png"
    encoded = base64.b64encode(path.read_bytes()).decode("ascii")
    return f"data:{mime_type};base64,{encoded}"


def flux_edit(prompt, sources, width=1024, height=1024, model="flux-2-pro"):
    if not 1 <= len(sources) <= 8:
        raise ValueError("FLUX.2 принимает от 1 до 8 референсных изображений")
    payload = {
        "model": model,
        "prompt": prompt,
        "width": width,
        "height": height,
        "output_format": "png",
    }
    for number, source in enumerate(sources, start=1):
        field = "input_image" if number == 1 else f"input_image_{number}"
        payload[field] = to_image_input(source)
    response = requests.post(API_URL, headers=HEADERS, json=payload, timeout=240)
    response.raise_for_status()
    return response.json()


def save_first_image(result, stem):
    # FLUX возвращает ссылку, которая действует 10 минут, поэтому скачиваем сразу.
    download = requests.get(result["data"][0]["url"], timeout=120)
    download.raise_for_status()
    image = download.content
    extension = "png" if image.startswith(b"\x89PNG") else "jpg"
    path = Path(f"{stem}.{extension}")
    path.write_bytes(image)
    return path


result = flux_edit(
    "Изображение 1 — красный робот, изображение 2 — фон с оранжереей, изображение 3 — "
    "жёлтый скейтборд. Создай одну цельную новую сцену, а не коллаж.",
    ["robot.png", "greenhouse.png", "skateboard.png"],
    width=1792,
    height=1024,
)
print("Сохранено:", save_first_image(result, "flux-python-edit"))

Изображение 1 — персонаж, изображение 2 — фон, изображение 3 — товар.
Создай один цельный рекламный постер.
Сохрани одежду персонажа, освещение фона и логотип товара.
Не делай коллаж и не дели кадр на части.