Nano Banana Pro: AI Image Enhancement

# 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": "hovering_figure.png"
        },
        "_meta": {
            "title": "Load Image"
        }
    },
    "4": {
        "class_type": "GeminiImage2Node",
        "inputs": {
            "aspect_ratio": "auto",
            "images": [
                "2",
                0
            ],
            "model": "gemini-3-pro-image-preview",
            "prompt": "upscale this image. refine details. preserve text. retain composition. refine textures, skin and materials.",
            "resolution": "4K",
            "response_modalities": "IMAGE+TEXT",
            "seed": 902254240177735,
            "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)"
        }
    },
    "5": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "ComfyUI",
            "images": [
                "4",
                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)