Grok Imagine Image API

curl -X POST https://api.apimart.ai/v1/images/generations \
  -H "Authorization: Bearer $APIMART_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "grok-imagine-image",
    "prompt": "돌 표면 위 흰색 도자기 컵의 미니멀한 제품 사진, 부드러운 창가 조명, 깔끔한 배경",
    "n": 1,
    "aspect_ratio": "1:1",
    "resolution": "2k"
  }'

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": "grok-imagine-image",
  "prompt": "돌 표면 위 흰색 도자기 컵의 미니멀한 제품 사진, 부드러운 창가 조명, 깔끔한 배경",
  "n": 1,
  "aspect_ratio": "1:1",
  "resolution": "2k"
}
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": "grok-imagine-image",
    "prompt": "아침 햇살이 드는 대리석 주방 조리대 위에 제품을 놓고 제품은 그대로 유지",
    "n": 1,
    "image_urls": [
      "https://example.com/product.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"