Kling O1: Video to Video

# 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": "SaveVideo",
        "inputs": {
            "filename_prefix": "video/kling-omni-t2v",
            "format": "auto",
            "format.codec": "auto",
            "video": [
                "6",
                0
            ]
        },
        "_meta": {
            "title": "Save Video"
        }
    },
    "3": {
        "class_type": "LoadImage",
        "inputs": {
            "image": "blue_shirt_rabbit.jpg"
        },
        "_meta": {
            "title": "Load Image"
        }
    },
    "6": {
        "class_type": "KlingOmniProVideoToVideoNode",
        "inputs": {
            "aspect_ratio": "1:1",
            "duration": 3,
            "keep_original_sound": True,
            "model_name": "kling-video-o1",
            "prompt": "Transform the dancing person into the anthropomorphic rabbit character from the reference image. The rabbit character, dressed in a light blue knitted cardigan over a white shirt and light brown pants, should replicate all the dancing movements and poses from the original video. Keep the same background, camera movement, and dance sequence, but replace the human character with the rabbit character while maintaining the same energetic and dynamic dancing style.",
            "reference_images": [
                "3",
                0
            ],
            "reference_video": [
                "7",
                0
            ],
            "resolution": "1080p",
            "seed": 0
        },
        "_meta": {
            "title": "Kling 3.0 Omni Video to Video"
        }
    },
    "7": {
        "class_type": "LoadVideo",
        "inputs": {
            "file": "dancing_women.mp4"
        },
        "_meta": {
            "title": "Load Video"
        }
    }
})

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