Grok Imagine Image Quality: Generation

# Python 3.10+: pip install comfy-sdk==0.4.0
import os

from comfy_sdk import Comfy

api_key = os.environ.get("COMFY_API_KEY", "").strip()
if not api_key: raise RuntimeError("Set COMFY_API_KEY")
os.environ.setdefault("COMFY_BASE_URL", "https://cloud.comfy.org")
client = Comfy(api_key=api_key)
workflow = client.workflows.from_json({
    "2": {
        "class_type": "GrokImageNode",
        "inputs": {
            "aspect_ratio": "16:9",
            "model": "grok-imagine-image-quality",
            "number_of_images": 1,
            "prompt": "A detailed digital concept art piece depicting the colossal undercarriage of a towering industrial bipedal machine, viewed from directly below against a turbulent twilight sky. Massive scorched bronze and heat-tempered cast iron struts stretch outward across the expanse, flanked by gargantuan oil-streaked hydraulic cylinders and layered heat-resistant plates. Every heavy armor plate is completely plain and unmarked, showing rugged cast metal textures, pitted slag, and scorch marks without any stencils or insignias. Dramatic, sweltering underlighting erupts from deep amber thermal exhaust ports and glowing heat sinks, casting fiery copper rims and deep, obsidian shadows along the immense mechanical joints. Thick plumes of black diesel smoke and shimmering heat distortion rise through the oppressive, soot-choked air, emphasizing the overwhelming mass and suffocating scale of the titan looming overhead.",
            "quality": "medium",
            "resolution": "1K",
            "seed": 51209668
        },
        "_meta": {
            "title": "Grok Image"
        }
    },
    "3": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "Grok_imagine_image_quality",
            "images": [
                "2",
                0
            ]
        },
        "_meta": {
            "title": "Save Image"
        }
    }
})

# run() also passes api_key to the partner nodes in this workflow.
job = client.run(workflow, api_key=api_key)
for output in job.get_outputs("3"):
    output.to_file(output.name)