Swap Product Into a Reference AD

# 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": "reference_ad.png"
        },
        "_meta": {
            "title": "Load Ad Image"
        }
    },
    "2": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "comfy_gpu.png"
        },
        "_meta": {
            "title": "Load Product Image"
        }
    },
    "3": {
        "class_type": "GeminiImage2Node",
        "inputs": {
            "aspect_ratio": "auto",
            "images": [
                "18",
                0
            ],
            "model": "gemini-3-pro-image-preview",
            "prompt": [
                "14",
                0
            ],
            "resolution": "4K",
            "response_modalities": "IMAGE+TEXT",
            "seed": 611482562925396,
            "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)"
        }
    },
    "9": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "ComfyUI",
            "images": [
                "3",
                0
            ]
        },
        "_meta": {
            "title": "Save Image"
        }
    },
    "11": {
        "class_type": "PrimitiveStringMultiline",
        "inputs": {
            "value": "Objective: Create a composite commercial image by synthesizing layout inputs with subject inputs. Target Product: {user_input}\n\n1. Inputs\nReference Input 1 (Style & Layout): Acts as the strict structural blueprint. This dictates composition, graphic elements, lighting direction, and text layout.\n\nReference Input 2 (Subject): Provides the primary subject matter (product shape, texture, branding).\n\n2. Execution Instructions\nPhase A: Scene & Design Analysis\nEnvironmental Scan: Analyze the perspective, depth, and lighting environment of [Input 1].\n\nDNA Deconstruction: Identify font styles, graphic shapes, and specific noise/grain levels.\n\nPhase B: Object Replacement\nRemoval: Remove the main product currently featured in [Input 1].\n\nInsertion: Insert the product isolated from [Input 2] into the exact focal point and scale defined by the original layout.\n\nPhase C: Text Modification (Content & Style)\nCondition Check: Analyze [Input 1] for existing Copy text elements (excluding text physically on the product).\n\nIf Copy Text Exists: Replace the existing copy with text relevant to the target product. You must replicate the exact font, weight, kerning, and drop-shadows of the original.\n\nIf No Copy Text Exists: Strictly do not generate any text. Preserve the negative space exactly as it appears in [Input 1].\n\nPhase D: Harmonize & Finish\nLighting: Adjust lighting, cast shadows, and reflections on the inserted product to match the environment.\n\nTexture: Apply the original image's texture (film grain, halftone, print artifacts) to the new elements to ensure a cohesive \"print ad\" look.\n\n3. Final Output\nAspect Ratio: Must match Input 1.\n\nQuality: A seamless, professional advertisement where the design layout is preserved, but the subject and color story are synthesized according to the user's preference."
        },
        "_meta": {
            "title": "Prompt Template"
        }
    },
    "14": {
        "class_type": "RegexReplace",
        "inputs": {
            "case_insensitive": True,
            "count": 0,
            "dotall": False,
            "multiline": False,
            "regex_pattern": "{user_input}",
            "replace": [
                "15",
                0
            ],
            "string": [
                "11",
                0
            ]
        },
        "_meta": {
            "title": "Replace Text (Regex)"
        }
    },
    "15": {
        "class_type": "PrimitiveStringMultiline",
        "inputs": {
            "value": "A ComfyUI branded RTX PRO 6000. This ad is promoting Comfy Cloud. The header 1 says \"The Comfy Cloud\". The smaller copy should inform that Comfy Cloud switched to Blackwell RTX 6000 Pros, which are approimately twice as fast as A100s. They also have 96GB of VRAM and 180GB of RAM, which will unlock much heavier workflows such as video upscaling."
        },
        "_meta": {
            "title": "Describe the Product and Ad Copy"
        }
    },
    "16": {
        "class_type": "ResizeAndPadImage",
        "inputs": {
            "image": [
                "2",
                0
            ],
            "interpolation": "area",
            "padding_color": "white",
            "target_height": [
                "17",
                1
            ],
            "target_width": [
                "17",
                0
            ]
        },
        "_meta": {
            "title": "Resize And Pad Image"
        }
    },
    "17": {
        "class_type": "GetImageSize",
        "inputs": {
            "image": [
                "1",
                0
            ]
        },
        "_meta": {
            "title": "Get Image Size"
        }
    },
    "18": {
        "class_type": "ImageBatch",
        "inputs": {
            "image1": [
                "1",
                0
            ],
            "image2": [
                "16",
                0
            ]
        },
        "_meta": {
            "title": "Batch Images (DEPRECATED)"
        }
    }
})

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