# ComfyUI-EdgeTTS

# ComfyUI-EdgeTTS is a powerful text-to-speech node for ComfyUI, leveraging Microsoft's Edge TTS capabilities. 
# It enables seamless conversion of text into natural-sounding speech, supporting multiple languages and voices. 
# Ideal for enhancing user interactions, this node is easy to integrate and customize, making it perfect for various applications.

# This integration script follows GPL-3.0 License.
# When using or modifying this code, please respect both the original model licenses
# and this integration's license terms.
#
# Source: https://github.com/1038lab/ComfyUI-EdgeTTS

import importlib.util
import os
import sys

current_dir = os.path.dirname(__file__)
sys.path.insert(0, current_dir)

NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}
WEB_DIRECTORY = "./web"

def load_modules_from_directory(directory):
    if not os.path.exists(directory):
        return

    for file in os.listdir(directory):
        if not file.endswith(".py"):
            continue
            
        module_name = os.path.basename(file)[:-3]
        if module_name == os.path.basename(__file__)[:-3]:
            continue

        file_path = os.path.join(directory, file)
        try:
            spec = importlib.util.spec_from_file_location(module_name, file_path)
            if spec is None or spec.loader is None:
                continue
                
            module = importlib.util.module_from_spec(spec)
            sys.modules[module_name] = module
            spec.loader.exec_module(module)

            if hasattr(module, "NODE_CLASS_MAPPINGS"):
                NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
            if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS"):
                NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
                
        except Exception as e:
            print(f"Error loading module {module_name}: {e}")

def load_javascript(web_directory):
    return []

load_modules_from_directory(current_dir)
load_modules_from_directory(os.path.join(current_dir, "py"))

NODE_CLASS_MAPPINGS = dict(sorted(
    NODE_CLASS_MAPPINGS.items(),
    key=lambda x: NODE_DISPLAY_NAME_MAPPINGS.get(x[0], x[0])
))
NODE_DISPLAY_NAME_MAPPINGS = dict(sorted(
    NODE_DISPLAY_NAME_MAPPINGS.items(),
    key=lambda x: x[1]
))

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY", "load_javascript"]