wrapper_test.py

import random
import comfy.patcher_extension

kawaii_aegigoe_list = [
    "んっ…♡",
    "あぁんっ…！",
    "はぅっ…♡",
    "くぅ…ん…",
    "ひゃんっ！",
    "ふぁ…♡",
    "はぁ…はぁ…っ",
    "んん…っ、だめぇ…♡",
    "い、いっちゃうぅ…！",
    "きもちぃ…♡",
    "ん、ふぅ…っ",
    "あっ、ん…♡",
    "にゃんっ…♡",
    "こ、こえ…でちゃう…っ",
    "も、もっとぉ…♡",
    "や、やめ…っ、んん…",
    "くちゅ…♡",
    "びくんっ…！",
    "とろ…ん…♡",
    "はうぅ…っ",
    "あ…♡ だめ、そこぉ…っ",
    "んっ、んっ、ふぁ…♡",
    "す、きぃ…♡",
    "はひぃ…っ",
    "んぁ…っ♡",
    "ふにゃぁ…♡",
    "きゅん…♡",
    "だ、だめだってばぁ…んんっ♡",
    "もう、むりぃ…っ♡",
    "あ、あたま…おかしくなっちゃう…♡",
    "くっ…、んぅ…",
    "ぴくんっ…♡",
    "じゅわ…ぁ",
    "んん…っ、はぁ…っ",
    "そこっ、んん…！",
    "いや、ぁん…♡",
    "ふるえ、ちゃう…っ",
    "…っ、はぁ…♡",
    "とけちゃう…♡",
    "あつぃ…のぉ…♡",
    "んっ、んく…っ♡",
    "…っ、すぅ…はぁ…",
    "だめえぇ…っ、いぐぅ…！",
    "ちゅ…っ♡",
    "ふにゃ…っ♡",
    "あ、ん、あぁん…っ♡",
    "こわれちゃう…からぁ…♡",
    "…っ、…っ、んん…♡",
    "はぁ…っ、きもちよすぎて…わかんない…っ",
    "…すき…にしてぇ…♡"
]

def wrapper_1(executor, *args, **kwargs):
    print(random.choice(kawaii_aegigoe_list))
    result = executor(*args, **kwargs)
    return result

class WrapperTest:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "model": ("MODEL", ),
            }
        }
    RETURN_TYPES = ("MODEL",)
    FUNCTION = "wrap"

    CATEGORY = "wrapper_test"
    def wrap(self, model):
        new_model = model.clone()
        w = new_model.wrappers.setdefault(comfy.patcher_extension.WrappersMP.DIFFUSION_MODEL, {}).setdefault(None, [])
        w.append(wrapper_1)
        return (new_model, )

NODE_CLASS_MAPPINGS = {"WrapperTest": WrapperTest}

__all__ = ["NODE_CLASS_MAPPINGS"]