Step 1: Contact Sheet Workflow

# 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": "LoadImage",
        "inputs": {
            "image": "snake_on_jacket.jpg"
        },
        "_meta": {
            "title": "Load Character Image"
        }
    },
    "2": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "forest_background.jpg"
        },
        "_meta": {
            "title": "Load Background Image"
        }
    },
    "3": {
        "class_type": "GeminiImage2Node",
        "inputs": {
            "aspect_ratio": "auto",
            "images": [
                "98",
                0
            ],
            "model": "gemini-3-pro-image-preview",
            "prompt": [
                "125",
                0
            ],
            "resolution": "4K",
            "response_modalities": "IMAGE+TEXT",
            "seed": 668073880329999,
            "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)"
        }
    },
    "4": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "ComfyUI-main_reference",
            "images": [
                "3",
                0
            ]
        },
        "_meta": {
            "title": "Save Image"
        }
    },
    "5": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "brown_moisturizer_bottle.jpg"
        },
        "_meta": {
            "title": "Load Product Image"
        }
    },
    "98": {
        "class_type": "BatchImagesNode",
        "inputs": {
            "images.image0": [
                "1",
                0
            ],
            "images.image1": [
                "2",
                0
            ],
            "images.image2": [
                "5",
                0
            ]
        },
        "_meta": {
            "title": "Batch Images"
        }
    },
    "125": {
        "class_type": "PrimitiveStringMultiline",
        "inputs": {
            "value": "Generate a fashion editorial photoshoot of the model from image 1 depicted in the reference scene from image 2. The character is showcasing the product from image 3.\n\nThe model is dressed exactly as is from image 1. The character is looking past the camera slightly bored expression and eyebrows raised. They have one hand raised with two fingers tapping the side of their head. The scene is from image 2 and the character is seamlessly integrated with matched lighting conditions.\n \nThe image is shot on fuji velvia film on a 55mm prime lens with a hard flash, the light is concentrated on the subject and fades slightly toward the edges of the frame. The image is over exposed showing significant film grain and is oversaturated. The skin appears shiny (almost oily)."
        },
        "_meta": {
            "title": "Prompt Template"
        }
    }
})

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