OpenAI: GPT-Image-1 Multi Inputs

# 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": "OpenAIGPTImage1",
        "inputs": {
            "background": "opaque",
            "custom_height": 1024,
            "custom_width": 1024,
            "image": [
                "3",
                0
            ],
            "model": "gpt-image-2",
            "n": 1,
            "prompt": "A cute cat sitting on a boat, holding a glass of champagne, with the vast ocean in the background, wearing a red hat and red - framed sunglasses, in the style of Studio Ghibli",
            "quality": "high",
            "seed": 1384371958,
            "size": "auto"
        },
        "_meta": {
            "title": "OpenAI GPT Image 2"
        }
    },
    "2": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "UPLOADED_image_FILENAME"
        },
        "_meta": {
            "title": "Load Image"
        }
    },
    "3": {
        "class_type": "ImageBatch",
        "inputs": {
            "image1": [
                "4",
                0
            ],
            "image2": [
                "2",
                0
            ]
        },
        "_meta": {
            "title": "Batch Images (DEPRECATED)"
        }
    },
    "4": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "UPLOADED_image_2_FILENAME"
        },
        "_meta": {
            "title": "Load Image"
        }
    },
    "6": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "ComfyUI",
            "images": [
                "1",
                0
            ]
        },
        "_meta": {
            "title": "Save Image"
        }
    }
})
# Replace each example.com URL with a public URL of your file.
workflow.set_input("2", "image", client.assets.from_url("https://example.com/replace-with-a-url/image"))
workflow.set_input("4", "image", client.assets.from_url("https://example.com/replace-with-a-url/image_2"))

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