Grok Imagine 2.0 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-2.0",
    "prompt": "石质台面上的白色陶瓷杯极简产品摄影，柔和窗光，干净背景",
    "n": 1,
    "aspect_ratio": "16:9",
    "resolution": "2k",
    "quality": "medium"
  }'

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-2.0",
  "prompt": "石质台面上的白色陶瓷杯极简产品摄影，柔和窗光，干净背景",
  "n": 1,
  "aspect_ratio": "16:9",
  "resolution": "2k",
  "quality": "medium"
}
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-2.0",
    "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"