name change to ChatterBox_Voice
This commit is contained in:
+14
-14
@@ -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}")
|
||||
|
||||
|
||||
@@ -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",
|
||||
}
|
||||
Reference in New Issue
Block a user