diff --git a/nodes/nodes.py b/nodes/nodes.py index 748301e..75e749a 100644 --- a/nodes/nodes.py +++ b/nodes/nodes.py @@ -450,10 +450,25 @@ class AnyInput: return True +class OutputAudio(SaveAudio): + CPACK_NODE = True + CATEGORY = "ComfyPack/output" + + @classmethod + def INPUT_TYPES(s): + return { + "required": { + "audio": ("AUDIO",), + "filename_prefix": ("STRING", {"default": "cpack_output_"}), + }, + "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"}, + } + + NODE_CLASS_MAPPINGS = { "CPackOutputFile": OutputFile, "CPackOutputImage": OutputImage, - "CPackOutputAudio": SaveAudio, + "CPackOutputAudio": OutputAudio, "CPackOutputZip": OutputImageWithStringTxt, "CPackInputImage": ImageInput, "CPackInputString": StringInput,