Muse Image API

IDEMPOTENCY_KEY="${IDEMPOTENCY_KEY:-$(uuidgen)}"
curl -X POST https://api.apimart.ai/v1/images/generations \
  -H "Authorization: Bearer $APIMART_API_KEY" \
  -H "Content-Type: application/json" \
  -H "Idempotency-Key: $IDEMPOTENCY_KEY" \
  -d '{
    "model": "muse-image-ext",
    "prompt": "一只红色陶瓷杯放在白色桌面上，柔和棚拍灯光，产品摄影，无文字",
    "n": 1,
    "aspect_ratio": "1:1"
  }'

import os
import time
import requests
import uuid

API_BASE = "https://api.apimart.ai"
headers = {"Authorization": f"Bearer {os.environ['APIMART_API_KEY']}"}
idempotency_key = os.environ.get("IDEMPOTENCY_KEY") or str(uuid.uuid4())
headers["Idempotency-Key"] = idempotency_key
print("Idempotency-Key:", idempotency_key)

# 1. Submit the generation task
payload = {
  "model": "muse-image-ext",
  "prompt": "一只红色陶瓷杯放在白色桌面上，柔和棚拍灯光，产品摄影，无文字",
  "n": 1,
  "aspect_ratio": "1:1"
}
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"]
print("Task ID:", task_id)

# 2. Poll the task until it finishes
for _ in range(60):
    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(5)
else:
    raise TimeoutError(f"task {task_id} did not finish in time; continue GET polling with this task ID")

IDEMPOTENCY_KEY="${IDEMPOTENCY_KEY:-$(uuidgen)}"
curl -X POST https://api.apimart.ai/v1/images/generations \
  -H "Authorization: Bearer $APIMART_API_KEY" \
  -H "Content-Type: application/json" \
  -H "Idempotency-Key: $IDEMPOTENCY_KEY" \
  -d '{
    "model": "muse-image-ext",
    "prompt": "保留参考图中杯子的外观，将背景改为木质咖啡桌，温暖自然光",
    "n": 1,
    "image_urls": [
      "https://example.com/reference.png"
    ]
  }'

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"