Ethanfel
340c0aa402
Chinese characters vs English words are self-identifying to the model. No need for a separate language signal on either node. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
75 lines
3.3 KiB
Python
75 lines
3.3 KiB
Python
class OmniVoiceVoiceDesign:
|
||
"""Compose a voice design instruct string from structured dropdowns."""
|
||
|
||
GENDERS = ["none", "male", "female"]
|
||
|
||
AGES = ["none", "child", "teenager", "young adult", "middle-aged", "elderly"]
|
||
|
||
PITCHES = [
|
||
"none",
|
||
"very low pitch", "low pitch", "moderate pitch",
|
||
"high pitch", "very high pitch", "whisper",
|
||
]
|
||
|
||
# Exactly the accents validated by the model's _resolve_instruct() for English
|
||
ACCENTS = [
|
||
"none",
|
||
"american accent", "australian accent", "british accent",
|
||
"canadian accent", "chinese accent", "indian accent",
|
||
"japanese accent", "korean accent", "portuguese accent",
|
||
"russian accent",
|
||
]
|
||
|
||
# Chinese dialect items validated by the model's _resolve_instruct()
|
||
ZH_GENDERS = ["none", "男", "女"]
|
||
ZH_AGES = ["none", "儿童", "少年", "青年", "中年", "老年"]
|
||
ZH_PITCHES = ["none", "极低音调", "低音调", "中音调", "高音调", "极高音调", "耳语"]
|
||
ZH_DIALECTS = [
|
||
"none",
|
||
"东北话", "云南话", "四川话", "宁夏话", "桂林话",
|
||
"河南话", "济南话", "甘肃话", "石家庄话", "贵州话",
|
||
"陕西话", "青岛话",
|
||
]
|
||
|
||
@classmethod
|
||
def INPUT_TYPES(cls):
|
||
return {
|
||
"required": {
|
||
"language": (
|
||
["English", "Chinese"],
|
||
{
|
||
"default": "English",
|
||
"tooltip": "Selects the instruct vocabulary. The language output wires directly into Generate — no need to set it there too.",
|
||
},
|
||
),
|
||
"gender": (cls.GENDERS, {"default": "female",
|
||
"tooltip": "Voice gender (English). Ignored when language is Chinese — use zh_gender."}),
|
||
"age": (cls.AGES, {"default": "none",
|
||
"tooltip": "Age of the speaker (English). Ignored when language is Chinese — use zh_age."}),
|
||
"pitch": (cls.PITCHES, {"default": "none",
|
||
"tooltip": "Pitch (English). Ignored when language is Chinese — use zh_pitch."}),
|
||
"accent": (cls.ACCENTS, {"default": "none",
|
||
"tooltip": "Accent (English only, 10 supported values)."}),
|
||
},
|
||
"optional": {
|
||
"zh_gender": (cls.ZH_GENDERS, {"default": "none", "tooltip": "声线性别 (Chinese mode)"}),
|
||
"zh_age": (cls.ZH_AGES, {"default": "none", "tooltip": "年龄段 (Chinese mode)"}),
|
||
"zh_pitch": (cls.ZH_PITCHES, {"default": "none", "tooltip": "音调 (Chinese mode)"}),
|
||
"zh_dialect": (cls.ZH_DIALECTS, {"default": "none", "tooltip": "方言/口音 (Chinese mode)"}),
|
||
},
|
||
}
|
||
|
||
RETURN_TYPES = ("STRING",)
|
||
RETURN_NAMES = ("instruct",)
|
||
FUNCTION = "compose"
|
||
CATEGORY = "OmniVoice"
|
||
|
||
def compose(self, language, gender, age, pitch, accent,
|
||
zh_gender="none", zh_age="none", zh_pitch="none", zh_dialect="none"):
|
||
if language == "Chinese":
|
||
parts = [v for v in [zh_gender, zh_age, zh_pitch, zh_dialect] if v != "none"]
|
||
return ("，".join(parts),)
|
||
else:
|
||
parts = [v for v in [gender, age, pitch, accent] if v != "none"]
|
||
return (", ".join(parts),)
|