Class PD_StringToCaptionJSON

py/string_to_caption_json.py:5–144

PD字符串转Caption JSON 节点 功能：将输入的字符串转换为包含caption和lang字段的JSON格式字符串 自动检测语言（中文/英文）

class PD_StringToCaptionJSON:
    """
    PD字符串转Caption JSON 节点
    功能：将输入的字符串转换为包含caption和lang字段的JSON格式字符串
    自动检测语言（中文/英文）
    """
    
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "text": ("STRING", {
                    "default": "",
                    "multiline": True,
                    "placeholder": "输入文本内容...",
                    "forceInput": True
                }),
            },
            "optional": {
                "lang": ("STRING", {
                    "default": "en",
                    "multiline": False,
                    "placeholder": "语言代码(en/zh/ja等)，留空为en"
                }),
            }
        }
    
    RETURN_TYPES = ("STRING",)
    RETURN_NAMES = ("json_string",)
    FUNCTION = "convert_to_json"
    CATEGORY = "PDuse/Text"
    
    # 支持列表输入和输出
    INPUT_IS_LIST = {"text": True}
    OUTPUT_IS_LIST = (True,)
    
    def detect_language(self, text):
        """
        检测文本语言
        返回语言代码：en(英文), zh(中文), ja(日文) 等
        """
        if not text or not text.strip():
            return "en"
        
        # 检测中文字符
        chinese_pattern = re.compile(r'[\u4e00-\u9fff]')
        chinese_count = len(chinese_pattern.findall(text))
        
        # 检测日文字符（平假名、片假名）
        japanese_pattern = re.compile(r'[\u3040-\u309f\u30a0-\u30ff]')
        japanese_count = len(japanese_pattern.findall(text))
        
        # 检测韩文字符
        korean_pattern = re.compile(r'[\uac00-\ud7a3]')
        korean_count = len(korean_pattern.findall(text))
        
        # 检测英文字符
        english_pattern = re.compile(r'[a-zA-Z]')