Grok Imagine Image 2 Serverless API

 1# pip install "segmind>=1.1.0"
 2# export SEGMIND_API_KEY="YOUR_API_KEY"
 3import segmind
 4
 5# Async (v2): submit to the queue and block until COMPLETED.
 6# run() returns the final result dict (600s deadline, 1.0s poll by default).
 7result = segmind.run(
 8    "grok-imagine-image-2",
 9    prompt="A cozy ramen shop on a narrow Tokyo alley at night in the rain, glowing red and blue neon signs reflecting on the wet pavement, steam rising from a fresh bowl of ramen on a wooden counter, warm lantern light spilling from the doorway, a lone customer seated inside, cinematic street photography, ultra-detailed, sharp focus, shallow depth of field, 35mm",
10    quality="medium",
11    aspect_ratio="16:9",
12    resolution="1k",
13    n=1,
14    output_format="png",
15)
16print(result["status"])                      # COMPLETED
17print(result.get("output"))                  # model output (e.g. media URL)
18print(result["metrics"]["inference_time"])   # server compute seconds
19
20# --- Or submit + poll manually (track request_id, control the cadence) ---
21from segmind import SegmindClient, InferenceFailed, InferenceTimeout
22
23client = SegmindClient()                      # reads SEGMIND_API_KEY
24payload = {
25    "prompt": "A cozy ramen shop on a narrow Tokyo alley at night in the rain, glowing red and blue neon signs reflecting on the wet pavement, steam rising from a fresh bowl of ramen on a wooden counter, warm lantern light spilling from the doorway, a lone customer seated inside, cinematic street photography, ultra-detailed, sharp focus, shallow depth of field, 35mm",
26    "quality": "medium",
27    "aspect_ratio": "16:9",
28    "resolution": "1k",
29    "n": 1,
30    "output_format": "png",
31}
32job = client.submit_async("grok-imagine-image-2", **payload)
33print(job.request_id)                         # available immediately
34try:
35    result = job.wait(timeout=600, interval=1.0)
36except InferenceTimeout as e:
37    print("still running:", e.request_id)
38except InferenceFailed as e:
39    print("failed:", e.detail)