WaveSpeed: AI Image Restoration with SeedVR2

# 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({
    "3": {
        "class_type": "SaveImage",
        "inputs": {
            "filename_prefix": "wavespeed-seedvr2",
            "images": [
                "18",
                0
            ]
        },
        "_meta": {
            "title": "Save Image"
        }
    },
    "9": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "UPLOADED_image_FILENAME"
        },
        "_meta": {
            "title": "Load Image"
        }
    },
    "17": {
        "class_type": "ImageCompare",
        "inputs": {
            "compare_view": None,
            "image_a": [
                "9",
                0
            ],
            "image_b": [
                "18",
                0
            ]
        },
        "_meta": {
            "title": "Compare Images"
        }
    },
    "18": {
        "class_type": "WavespeedImageUpscaleNode",
        "inputs": {
            "image": [
                "9",
                0
            ],
            "model": "SeedVR2",
            "target_resolution": "4K"
        },
        "_meta": {
            "title": "WaveSpeed Image Upscale"
        }
    }
})
workflow.set_input("9", "image", client.assets.from_url("https://comfy.org/catalog-media/8e5499d9-d177-42d2-aa51-a0bc678bbd44/b3108314-97fe-4392-83f5-de620b820dba"))

# run() also passes api_key to the partner nodes in this workflow.
job = client.run(workflow, api_key=api_key)
for output in job.get_outputs("3"):
    output.to_file(output.name)