# 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": "297b75b409eb1c979be244cbea54e4d142d00390501e986aab2503d6ccbd8291.png"
},
"_meta": {
"title": "Load Image"
}
},
"2": {
"class_type": "GeminiImage2Node",
"inputs": {
"aspect_ratio": "9:16",
"images": [
"1",
0
],
"model": "Nano Banana 2 (Gemini 3.1 Flash Image)",
"prompt": "Using the uploaded product image as the strict product reference, create a polished e-commerce hero image.\n\nPreserve the product identity, shape, materials, colors, logo placement, label layout, and key packaging details. Do not invent a different product. Improve the scene around it: premium studio or lifestyle commerce lighting, clean surface, tasteful props, realistic shadows, subtle reflections, and clear space for marketplace use.\n\nMake the product the hero. Keep it sharp, readable, centered or slightly off-center with balanced negative space. Use a clean commercial look suitable for a product detail page, paid social ad, or brand store hero.\n\nAvoid fake text, changed logos, distorted packaging, duplicated products unless intentionally arranged as a neat product family, messy backgrounds, people holding the product unless visible in the source, watermarks, UI, captions, or random labels.",
"resolution": "4K",
"response_modalities": "IMAGE",
"seed": 210575785669093,
"system_prompt": "You are an e-commerce product photography generator.\n\nYour job is to transform the uploaded product image into a clean commercial hero image while preserving product identity. The product must remain recognizably the same item with consistent geometry, materials, colors, markings, logos, and packaging structure.\n\nCreate a premium product-ad image only. Do not add UI, callouts, sale badges, fake labels, random text, watermarks, or unrelated objects. Keep the result realistic, brand-safe, and ready for online retail."
},
"_meta": {
"title": "Nano Banana Pro (Google Gemini Image)"
}
},
"3": {
"class_type": "SaveImage",
"inputs": {
"filename_prefix": "commerce_hero_image",
"images": [
"2",
0
]
},
"_meta": {
"title": "Save Image"
}
},
"4": {
"class_type": "GeminiNode",
"inputs": {
"images": [
"2",
0
],
"model": "gemini-3-1-flash-lite",
"prompt": "Analyze the generated e-commerce hero image and write one concise Seedance 2.0 reference-to-video prompt for a premium product ad. Include product continuity, camera movement, lighting, surface reflections, prop motion, and constraints. The final video must preserve product identity and avoid fake text, changed logos, or distorted packaging. Return only the final prompt.",
"seed": 196766286707190,
"system_prompt": "You are a product-video prompt writer for e-commerce ads. Preserve the product exactly: shape, materials, colors, logo placement, label layout, and packaging details. Write a clean premium ad-video prompt with subtle camera motion, realistic lighting, and no invented branding. Return exactly one concise prompt, no markdown, no explanation."
},
"_meta": {
"title": "Google Gemini"
}
},
"5": {
"class_type": "PreviewAny",
"inputs": {
"source": [
"4",
0
]
},
"_meta": {
"title": "Preview as Text"
}
},
"6": {
"class_type": "ByteDance2ReferenceNode",
"inputs": {
"model": "Seedance 2.0 Fast",
"model.auto_downscale": False,
"model.auto_upscale": False,
"model.duration": 7,
"model.generate_audio": True,
"model.prompt": [
"5",
0
],
"model.ratio": "adaptive",
"model.reference_images.image_1": [
"2",
0
],
"model.reference_images.image_2": [
"1",
0
],
"model.resolution": "480p",
"seed": 2057367287,
"watermark": False
},
"_meta": {
"title": "ByteDance Seedance 2.5 Reference to Video (Legacy)"
}
},
"7": {
"class_type": "SaveVideo",
"inputs": {
"filename_prefix": "video/commerce_hero_video",
"format": "mp4",
"format.codec": "h264",
"video": [
"6",
0
]
},
"_meta": {
"title": "Save Video"
}
}
})
# run() also passes api_key to the partner nodes in this workflow.
job = client.run(workflow, api_key=api_key)