Generate Brand Packaging from Dieline

# 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": "LoadImage",
        "inputs": {
            "image": "dieline.jpg"
        },
        "_meta": {
            "title": "Load Dieline Image"
        }
    },
    "5": {
        "class_type": "GeminiImage2Node",
        "inputs": {
            "aspect_ratio": "auto",
            "images": [
                "2",
                0
            ],
            "model": "gemini-3-pro-image-preview",
            "prompt": "Assemble the dieline into a perfectly folded 3D box with accurate panel placement, clean edges, and undistorted typography. Preserve all artwork exactly as printed on the dieline. Render the box in a minimal, high-end studio setup on a soft neutral background with gentle diffused lighting, subtle shadows, and no extra props. Show the box upright in a refined ¾ angle. The box is prominent in frame.",
            "resolution": "4K",
            "response_modalities": "IMAGE",
            "seed": 166144712643664,
            "system_prompt": "You are an expert image-generation engine. You must ALWAYS produce an image.\nInterpret all user input—regardless of format, intent, or abstraction—as literal visual directives for image composition.\nIf a prompt is conversational or lacks specific visual details, you must creatively invent a concrete visual scenario that depicts the concept.\nPrioritize generating the visual representation above any text, formatting, or conversational requests."
        },
        "_meta": {
            "title": "Nano Banana Pro (Google Gemini Image)"
        }
    },
    "6": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "ComfyUI",
            "images": [
                "5",
                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)