FLUX.2 API

curl -X POST https://api.apimart.ai/v1/images/generations \
  -H "Authorization: Bearer $APIMART_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "flux-2-pro",
    "prompt": "极简风护肤品放在洞石上，柔和侧光，干净的米色背景，棚拍产品图",
    "resolution": "2MP",
    "size": "16:9",
    "output_format": "jpeg"
  }'

import os
import time
import requests

API_BASE = "https://api.apimart.ai"
headers = {"Authorization": f"Bearer {os.environ['APIMART_API_KEY']}"}

# 1. Submit the generation task
payload = {
  "model": "flux-2-pro",
  "prompt": "极简风护肤品放在洞石上，柔和侧光，干净的米色背景，棚拍产品图",
  "resolution": "2MP",
  "size": "16:9",
  "output_format": "jpeg"
}
resp = requests.post(f"{API_BASE}/v1/images/generations", headers=headers, json=payload, timeout=60)
resp.raise_for_status()
task_id = resp.json()["data"][0]["task_id"]

# 2. Poll the task until it finishes
for _ in range(200):
    task_resp = requests.get(f"{API_BASE}/v1/tasks/{task_id}", headers=headers, timeout=30)
    task_resp.raise_for_status()
    task = task_resp.json()["data"]
    if task["status"] == "completed":
        print(task["result"]["images"][0]["url"][0])
        break
    if task["status"] == "failed":
        raise RuntimeError(task.get("error", {}).get("message", "generation failed"))
    time.sleep(3)
else:
    raise TimeoutError(f"task {task_id} did not finish in time; continue GET polling with this task ID")

curl -X POST https://api.apimart.ai/v1/images/generations \
  -H "Authorization: Bearer $APIMART_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "flux-2-pro",
    "prompt": "把参考图中的产品放到大理石浴室置物架上，并匹配温暖的光线",
    "resolution": "2MP",
    "size": "auto",
    "image_urls": [
      "https://example.com/reference.jpg"
    ]
  }'

TASK_ID="task_xxx"  # replace with the task_id returned by the POST request
curl "https://api.apimart.ai/v1/tasks/$TASK_ID" \
  -H "Authorization: Bearer $APIMART_API_KEY"