47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from .utils import runwareUtils as rwUtils
|
|
from typing import Dict, Any
|
|
|
|
|
|
class audioInferenceInputs:
|
|
"""Audio Inference Inputs node for configuring audio generation inputs"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"Video": ("STRING", {
|
|
"tooltip": "Video URL or mediaUUID for audio generation. Can be a direct URL or a mediaUUID from Runware Media Upload node.",
|
|
"default": "",
|
|
}),
|
|
}
|
|
}
|
|
|
|
DESCRIPTION = "Configure custom inputs for Runware Audio Inference, including video input for audio extraction or generation."
|
|
FUNCTION = "createInputs"
|
|
RETURN_TYPES = ("RUNWAREAUDIOINFERENCEINPUTS",)
|
|
RETURN_NAMES = ("Audio Inference Inputs",)
|
|
CATEGORY = "Runware"
|
|
|
|
def createInputs(self, **kwargs) -> tuple[Dict[str, Any]]:
|
|
"""Create audio inference inputs from provided parameters"""
|
|
video = kwargs.get("Video", None)
|
|
|
|
inputs = {}
|
|
|
|
if video is not None and video.strip() != "":
|
|
inputs["video"] = video.strip()
|
|
|
|
return (inputs,)
|
|
|
|
|
|
# Node class mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RunwareAudioInferenceInputs": audioInferenceInputs,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RunwareAudioInferenceInputs": "Runware Audio Inference Inputs",
|
|
}
|
|
|