Grok Imagine Image 2.0: Text to Image

# 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({
    "1": {
        "class_type": "GrokImageNode",
        "inputs": {
            "aspect_ratio": "1:1",
            "model": "grok-imagine-image-2.0",
            "number_of_images": 1,
            "prompt": "Editorial fashion photography poster design, square format. Whimsical surreal fantasy marketplace scene: giant oversized peaches, figs and citrus fruits, each fruit larger than the surrounding buildings, towering clusters of glossy fruit winding around and dwarfing the pastel European-style town architecture. An elevated light rail train travels along vine-covered aerial tracks. Tiny mechanical flying drone insects float through the warm golden sunset atmosphere. A fashionable modern woman stands on an enormous glossy green leaf in the center of frame, her figure between and overlapping the giant fruits. She wears a terracotta patterned oversized blazer, cream knit turtleneck, wide leg tailored trousers, chunky platform leather shoes, holding a woven terracotta round handbag, wavy dark hair flowing in the breeze, confident editorial pose with a subtle smile, high fashion model styling. Lush twisting green vines wrap all architecture, market stalls with striped canvas awnings line the street, the ground covered with huge fruit berries each the size of a car. Warm soft golden hour lighting, dreamy volumetric haze, vibrant warm pastel color palette, hyperrealistic 3D render, high fashion editorial aesthetic, playful uncanny scale distortion, intricate details, cinematic depth of field.Bold oversized display typography spelling the exact text \"GROK IMAGINE IMAGE 2.0\" in a playful rounded condensed sans-serif font, deep crimson red, the large letters weaving through the composition across the upper half of the frame, some letters passing behind the woman's figure and some in front of her, letters slightly overlapping the giant fruits and rooftops, creating a layered interweaving between text and subject, the only text in the scene, no other words. Small scattered graphic elements around the composition: a colorful sticker-style circular badge with orbiting dots in the upper right, a cute round rubber-stamp icon in the lower left, playful thin wavy line doodles, tiny star icons and a small crescent shape near the bottom, a small vertical stack of colorful abstract symbols on the right edge, and a small sun-shaped emblem in the bottom right corner. A short block of small justified caption-style text sits in the lower right area, paired with a subtitle line above it in thin tracked-out caps. Overall layout uses asymmetric grid placement, generous negative space, vibrant warm pastel palette of cream, terracotta, peach, sage green and deep crimson accents, soft grainy film texture, playful surrealist editorial poster aesthetic, ultra-detailed, 8k print quality.\n\n",
            "quality": "medium",
            "resolution": "1K",
            "seed": 156395678
        },
        "_meta": {
            "title": "Grok Image"
        }
    },
    "2": {
        "class_type": "SaveImageAdvanced",
        "inputs": {
            "filename_prefix": "Grok_imagine_image_2.0",
            "format": "png",
            "format.bit_depth": "8-bit",
            "format.input_color_space": "sRGB",
            "images": [
                "1",
                0
            ]
        },
        "_meta": {
            "title": "Save Image (Advanced)"
        }
    }
})

# run() also passes api_key to the partner nodes in this workflow.
job = client.run(workflow, api_key=api_key)