name change to ChatterBox_Voice

This commit is contained in:
Shmuel Ronen
2025-06-04 09:37:09 +03:00
committed by GitHub
parent 07b18fbd02
commit 320e19056c
2 changed files with 26 additions and 25 deletions
+14 -14
View File
@@ -1,7 +1,7 @@
"""
ComfyUI ChatterBox Extension
ComfyUI ChatterBox Voice Extension
High-quality Text-to-Speech and Voice Conversion nodes using ResembleAI's ChatterboxTTS
Now with integrated audio recording and volume control!
Enhanced with integrated audio recording and smart voice capture!
"""
# Import ChatterBox TTS and VC nodes
@@ -17,30 +17,30 @@ except ImportError as e:
print("📋 Install sounddevice: pip install sounddevice")
AUDIO_RECORDER_AVAILABLE = False
# Node class mappings for ComfyUI
# Node class mappings for ComfyUI - UNIQUE NAMES TO AVOID CONFLICTS
NODE_CLASS_MAPPINGS = {
"ChatterboxTTS": ChatterboxTTSNode,
"ChatterboxVC": ChatterboxVCNode,
"ChatterBoxVoiceTTS": ChatterboxTTSNode,
"ChatterBoxVoiceVC": ChatterboxVCNode,
}
# Display names for the UI
# Display names for the UI - CLEAR BRANDING
NODE_DISPLAY_NAME_MAPPINGS = {
"ChatterboxTTS": "ChatterBox Text-to-Speech",
"ChatterboxVC": "ChatterBox Voice Conversion",
"ChatterBoxVoiceTTS": "🎤 ChatterBox Voice TTS",
"ChatterBoxVoiceVC": "🔄 ChatterBox Voice Conversion",
}
# Add Audio Recorder if available
# Add Audio Recorder if available - UNIQUE NAME
if AUDIO_RECORDER_AVAILABLE:
NODE_CLASS_MAPPINGS["ChatterBoxVoiceCapture"] = ChatterBoxVoiceCapture
NODE_DISPLAY_NAME_MAPPINGS["ChatterBoxVoiceCapture"] = "🎙️ ChatterBox Voice Capture"
# Extension info
# Extension info - UPDATED BRANDING
__version__ = "1.1.0"
__author__ = "ComfyUI ChatterBox Extension"
__description__ = "High-quality TTS and VC nodes using ResembleAI's ChatterboxTTS with integrated audio recording"
__author__ = "ComfyUI ChatterBox Voice Extension"
__description__ = "Enhanced ChatterBox TTS/VC with integrated voice recording and smart audio capture"
# Check for required packages
print("🎯 Loading ChatterBox Extension...")
print("🎯 Loading ChatterBox Voice Extension...")
try:
import chatterbox
@@ -59,7 +59,7 @@ except ImportError:
# Summary
total_nodes = len(NODE_CLASS_MAPPINGS)
print(f"🚀 ChatterBox Extension v{__version__} loaded with {total_nodes} nodes:")
print(f"🚀 ChatterBox Voice Extension v{__version__} loaded with {total_nodes} nodes:")
for node_class, display_name in NODE_DISPLAY_NAME_MAPPINGS.items():
print(f" • {display_name}")
+12 -11
View File
@@ -1,5 +1,6 @@
"""
ComfyUI Custom Nodes for ChatterboxTTS
ComfyUI Custom Nodes for ChatterboxTTS - Voice Edition
Enhanced with unique naming to avoid conflicts
"""
import torch
@@ -31,7 +32,7 @@ except ImportError as e:
class ChatterboxTTSNode:
"""
Text-to-Speech node using ChatterboxTTS
Text-to-Speech node using ChatterboxTTS - Voice Edition
"""
@classmethod
@@ -40,7 +41,7 @@ class ChatterboxTTSNode:
"required": {
"text": ("STRING", {
"multiline": True,
"default": "Hello world! This is ChatterboxTTS in ComfyUI."
"default": "Hello world! This is ChatterboxTTS Voice Edition in ComfyUI."
}),
"device": (["auto", "cuda", "cpu"], {"default": "auto"}),
"exaggeration": ("FLOAT", {
@@ -72,7 +73,7 @@ class ChatterboxTTSNode:
RETURN_TYPES = ("AUDIO",)
RETURN_NAMES = ("audio",)
FUNCTION = "generate_speech"
CATEGORY = "ChatterBox"
CATEGORY = "ChatterBox Voice" # UPDATED: Unique category
def __init__(self):
self.model = None
@@ -145,7 +146,7 @@ class ChatterboxTTSNode:
class ChatterboxVCNode:
"""
Voice Conversion node using ChatterboxVC
Voice Conversion node using ChatterboxVC - Voice Edition
"""
@classmethod
@@ -161,7 +162,7 @@ class ChatterboxVCNode:
RETURN_TYPES = ("AUDIO",)
RETURN_NAMES = ("converted_audio",)
FUNCTION = "convert_voice"
CATEGORY = "ChatterBox"
CATEGORY = "ChatterBox Voice" # UPDATED: Unique category
def __init__(self):
self.model = None
@@ -231,13 +232,13 @@ class ChatterboxVCNode:
raise e
# Node mappings for ComfyUI
# Node mappings for ComfyUI - UPDATED: Unique names to avoid conflicts
NODE_CLASS_MAPPINGS = {
"ChatterboxTTS": ChatterboxTTSNode,
"ChatterboxVC": ChatterboxVCNode,
"ChatterBoxVoiceTTS": ChatterboxTTSNode,
"ChatterBoxVoiceVC": ChatterboxVCNode,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"ChatterboxTTS": "ChatterBox Text-to-Speech",
"ChatterboxVC": "ChatterBox Voice Conversion",
"ChatterBoxVoiceTTS": "🎤 ChatterBox Voice TTS",
"ChatterBoxVoiceVC": "🔄 ChatterBox Voice Conversion",
}