113 lines
4.7 KiB
Python
113 lines
4.7 KiB
Python
"""
|
|
Runware Video Inference Speech Input Node.
|
|
Provides voice and text for speech synthesis; connect only to Runware Video Inference → speech (not Video Inference Inputs → Speech Inputs; that socket expects RUNWARESPEECHINPUT).
|
|
"""
|
|
|
|
from typing import Any, Dict, Tuple
|
|
|
|
|
|
class RunwareVideoInferenceSpeechInput:
|
|
"""Runware Video Inference Speech Input Node"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"useVoice": ("BOOLEAN", {
|
|
"tooltip": "Enable to include voice in speech synthesis.",
|
|
"default": False,
|
|
}),
|
|
"voice": ("STRING", {
|
|
"tooltip": "Voice ID for text-to-speech or speech-to-speech. Required when text script is provided. Only used when 'Use Voice' is enabled.",
|
|
"default": "",
|
|
}),
|
|
"useText": ("BOOLEAN", {
|
|
"tooltip": "Enable to include text in speech synthesis.",
|
|
"default": False,
|
|
}),
|
|
"text": ("STRING", {
|
|
"multiline": True,
|
|
"tooltip": "Text script for the avatar to speak. Use with voice, or text-only when voice is described via settings.voiceDescription on Runware Video Inference Settings. Only used when 'Use Text' is enabled.",
|
|
"default": "",
|
|
}),
|
|
"useSpeed": ("BOOLEAN", {
|
|
"tooltip": "Enable to set voice playback speed multiplier.",
|
|
"default": False,
|
|
}),
|
|
"speed": ("FLOAT", {
|
|
"tooltip": "Voice playback speed multiplier (0.5 - 1.5). Only used when 'Use Speed' is enabled.",
|
|
"default": 1.0,
|
|
"min": 0.5,
|
|
"max": 1.5,
|
|
"step": 0.05,
|
|
}),
|
|
"usePitch": ("BOOLEAN", {
|
|
"tooltip": "Enable to set voice pitch adjustment in semitones.",
|
|
"default": False,
|
|
}),
|
|
"pitch": ("FLOAT", {
|
|
"tooltip": "Voice pitch adjustment in semitones (-50 to 50). Only used when 'Use Pitch' is enabled.",
|
|
"default": 0.0,
|
|
"min": -50.0,
|
|
"max": 50.0,
|
|
"step": 0.5,
|
|
}),
|
|
"useLanguage": ("BOOLEAN", {
|
|
"tooltip": "Enable to set locale or accent hint for multilingual voices.",
|
|
"default": False,
|
|
}),
|
|
"language": ("STRING", {
|
|
"tooltip": "Locale or accent hint for multilingual voices, e.g. en-US. Only used when 'Use Language' is enabled.",
|
|
"default": "",
|
|
}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES: Tuple[str] = ("RUNWAREVIDEOINFERENCESPEECHINPUT",)
|
|
RETURN_NAMES = ("speech",)
|
|
FUNCTION = "createSpeech"
|
|
CATEGORY = "Runware"
|
|
DESCRIPTION = (
|
|
"Configure speech synthesis for Runware Video Inference. Connect the output to Runware Video Inference → speech "
|
|
"(top-level task speech). For speech inside inputs (e.g. inputs.speech), use Runware Video Speech Input "
|
|
"(RUNWARESPEECHINPUT) into Video Inference Inputs → Speech Inputs. "
|
|
"Outputs speech when at least one of voice or text is included (non-empty with its toggle); "
|
|
"text-only is valid when voice is supplied via settings.voiceDescription. "
|
|
"Optional fields (speed, pitch, language) are added only when their toggles are on."
|
|
)
|
|
|
|
def createSpeech(self, **kwargs) -> tuple:
|
|
"""Build speech dict for the Runware Video Inference speech connection (top-level task speech). Returns None if neither voice nor text is set."""
|
|
speech: Dict[str, Any] = {}
|
|
|
|
voice = (kwargs.get("voice") or "").strip()
|
|
if kwargs.get("useVoice") and voice:
|
|
speech["voice"] = voice
|
|
|
|
text = (kwargs.get("text") or "").strip()
|
|
if kwargs.get("useText") and text:
|
|
speech["text"] = text
|
|
|
|
if kwargs.get("useSpeed"):
|
|
speech["speed"] = float(kwargs.get("speed", 1.0))
|
|
if kwargs.get("usePitch"):
|
|
speech["pitch"] = float(kwargs.get("pitch", 0.0))
|
|
if kwargs.get("useLanguage"):
|
|
lang = (kwargs.get("language") or "").strip()
|
|
if lang:
|
|
speech["language"] = lang
|
|
|
|
if "voice" not in speech and "text" not in speech:
|
|
return (None,)
|
|
return (speech,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RunwareVideoInferenceSpeechInput": RunwareVideoInferenceSpeechInput,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RunwareVideoInferenceSpeechInput": "Runware Video Inference Speech Input",
|
|
}
|