更新版本V0.0.66

更新版本V0.0.66

Author: weilin9999

README.md

@@ -28,20 +28,31 @@
 
 # 版本更新介绍
 
-> 最新更新：2025-09-23
+> 最新更新：2025-10-12
 
-> 0.0.65 公测版本介绍 如果你要使用本插件请务必进我们的官方 QQ 群（1018231382）！
+> 0.0.66 公测版本介绍 如果你要使用本插件请务必进我们的官方 QQ 群（1018231382）！
 >
 > 1. 修复了一些已知的BUG
 >
-> 2. 合并PR [#40](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/40) 感谢各位小伙伴的帮助
+> 2. 合并PR [#44](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/44) 感谢各位小伙伴的帮助
 >
-> 3. 合并PR [#43](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/43) 感谢各位小伙伴的帮助
+> 3. 新增功能：在Tag的控制栏中可以一键收藏Tag，还添加了添加换行符的功能
 > 
+> 4. 翻译大修改！本次更新将翻译功能全部重改了，使用AI进行翻译，对接硅基AI进行调用API翻译，如果你想要对接其它的AI平台你可以进群提交建议，后续更新会添加新的平台
+>
 
 <details>
 <summary>点击查看往期更多更新内容</summary>
 
+> 0.0.65 公测版本介绍 2025-09-23
+>
+> 1. 修复了一些已知的BUG
+>
+> 2. 合并PR [#40](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/40) 感谢各位小伙伴的帮助
+>
+> 3. 合并PR [#43](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/43) 感谢各位小伙伴的帮助
+> 
+
 > 0.0.64 公测版本介绍 2025-09-14
 >
 > 1. 修复了已知BUG（Comfyui中执行栏拖拽图标错位问题）

README_EN.md

@@ -24,20 +24,31 @@ Due to limited personal time, the frequency of updating plug-ins will not be ver
 
 # Version update introduction
 
-> Last updated: 2025-09-23
+> Last updated: 2025-10-12
 
-> 0.0.65 Public Beta Version Introduction: If you want to use this plugin, please be sure to join our official QQ group (1018231382)! 
+> 0.0.66 Public Beta Version Introduction: If you want to use this plugin, please be sure to join our official QQ group (1018231382)! 
 >
-> 1. Fixed a known BUG 
+> 1. Fixed some known bugs
 >
-> 2. Merge PR [#40](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/40) thank you for your friend's help
+> 2. Merger PR # [44] (https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/44) thank you for your friend's help
 >
-> 3. Merge PR [#43](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/43) thank you for your friend's help
+> 3. New features: Tags can be bookmarked with one click in the Tag control bar. The ability to add a newline character has also been added
+>
+> 4. Translation overhaul! This update has changed all the translation functions, using AI for translation, and docking silicon based AI for API translation. If you want to connect with other AI platforms, you can submit suggestions in the group, and a new platform will be added in the subsequent update
 >
 
 <details>
 <summary>Click here for more updates from the past</summary>
 
+> 0.0.65 Public Beta Version 2025-09-23
+>
+> 1. Fixed a known BUG 
+>
+> 2. Merge PR [#40](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/40) thank you for your friend's help
+>
+> 3. Merge PR [#43](https://github.com/weilin9999/WeiLin-Comfyui-Tools/pull/43) thank you for your friend's help
+>
+
 > 0.0.64 Public Beta Version 2025-09-14
 >
 > 1. Fixed a known BUG (the issue of misaligned drag ICONS in the execution bar in Comfyui)

__init__.py

@@ -257,7 +257,7 @@ def INPUT_TYPES(self):
                 "lora_str": ("STRING", {
                     "multiline": True,
                     "default": "",
-                    "placeholder": placeholder_lora_text,
+                    "placeholder": "lora info box",
                 }),
                 "temp_lora_str": ("STRING", {
                     "multiline": True,
@@ -318,6 +318,8 @@ def load_lora_ing(self, clip=None, model=None, lora_str="", temp_lora_str=""):
 
                 model_lora_secondA, clip_lora_secondA = load_lora_for_models(
                     model_lora_secondA, clip_lora_secondA, lora, strength_model, strength_clip)
+            else:
+                print("Lora堆没有可用的Lora信息")
 
         return (clip_lora_secondA, model_lora_secondA)
         # return (model_lora_second, clip_lora_second)

app/server/ai_server/__init__.py

@@ -0,0 +1,82 @@
+# 硅基AI对接
+import requests
+from ..user_init.user_init import get_ai_info_setting
+
+_LANG_NAME = {
+    "zh": "Chinese ", "zh_CN": " Chinese ", "zh_TW": "Chinese (traditional) ",
+    "en": "English ", "ja": " Japanese ", "ko": "Korean ",
+    "fr": "French ", "de": " German ", "es": "Spanish ",
+    "ru": "Russian ", "it": " Italian ", "pt": "Portuguese"
+}
+
+
+def _lang_to_name(code: str) -> str:
+    if not code:
+        return "Chinese"
+    return _LANG_NAME.get(code, code)
+
+
+# 硅基AI翻译接口
+async def translateObject(objectData: str, target_lang_code: str = "zh") -> str:
+    ai_info_setting = get_ai_info_setting()
+    api_key = ai_info_setting.get("api_key", "")
+    base_url = ai_info_setting.get(
+        "base_url", "https://api.siliconflow.cn/v1").rstrip("/")
+    model = ai_info_setting.get("model", "THUDM/glm-4-9b-chat")
+
+    print("💡[WeiLin-Comfyui-Tools] 硅基AI翻译接口-使用模型：",  model, " 目标语种：", target_lang_code, " 正在翻译中...")
+
+    url = f"{base_url}/chat/completions"
+
+    payload = {
+        "model": model,
+        "messages": [
+            {
+                "role": "system",
+                "content": f"You are an expert in data translation processing. translate the text field of the JSON string data passed by the user into {_lang_to_name(target_lang_code)} and fill it in the corresponding translate field. You only need to return the corresponding JSON string data and do not modify any other data or parameters"
+            },
+            {
+                "content": objectData,
+                "role": "user"
+            }
+        ],
+        "stream": False,
+        "max_tokens": 4096,
+        "enable_thinking": False,
+        "thinking_budget": 4096,
+        "min_p": 0.05,
+        "stop": None,
+        "temperature": 0,
+        "top_p": 0.7,
+        "top_k": 50,
+        "frequency_penalty": 0.5,
+        "n": 1,
+        "response_format": {"type": "json_object"},
+        "tools": []
+    }
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+    dataResponse = response.json()
+    print(f"💡[WeiLin-Comfyui-Tools] 硅基AI翻译接口-翻译完成，以下是使用信息：\n - PromptToken：{dataResponse['usage']['total_tokens']}，\n - CompletionTokens：{dataResponse['usage']['completion_tokens']}，\n - TotalTokens：{dataResponse['usage']['total_tokens']}")
+    return dataResponse["choices"][0]["message"]["content"]
+
+
+# 硅基AI模型列表接口
+def getModelList() -> dict:
+    ai_info_setting = get_ai_info_setting()
+    api_key = ai_info_setting["api_key"]
+    base_url = ai_info_setting["base_url"]
+
+    url = f"{base_url}/models"
+
+    headers = {"Authorization": f"Bearer {api_key}"}
+
+    querystring = {"type": "text"}
+
+    response = requests.get(url, headers=headers, params=querystring)
+
+    return response.json()

app/server/ai_server/siliconflow.py

@@ -23,6 +23,7 @@
 )
 from .prompt_api.random_tag_template import * 
 from .prompt_api.danbooru import *
+from .ai_server.siliconflow import translateObject,getModelList
 
 
 static_path = os.path.join(os.path.dirname(__file__), "../../dist/")
@@ -790,9 +791,14 @@ async def _tanslater_text(request):
         data_setting = get_translate_settings()     # service/source_lang/target_lang
 
         text = data['text']
+        strObjectData = data['str_object']
         if current_setting == "openai":
             # OpenAI 只需要目标语种来写 prompt
-            result = await openai_translate(text, data_setting['translate_target_lang'])
+            result = await openai_translate(strObjectData, data_setting['translate_target_lang'])
+        elif current_setting == "other_ai_plate":
+            aiInfoData = get_ai_info_setting()
+            if aiInfoData.get("base_url") == "https://api.siliconflow.cn/v1":
+                result = await translateObject(strObjectData, data_setting['translate_target_lang'])
         else:
             # 仍然走 translators（含 'translater' 与历史的 'network'）
             result = translateText(
@@ -801,7 +807,7 @@ async def _tanslater_text(request):
                 data_setting['translate_source_lang'],
                 data_setting['translate_target_lang']
             )
-        return web.json_response({"text": result})
+        return web.json_response({"data": result})
     except Exception as e:
         print(f"Error: {e}")
         return web.Response(status=500)
@@ -814,17 +820,22 @@ async def _tanslater_input_text(request):
         current_setting = get_translate_setting()
         data_setting = get_translate_settings()
         text = data['text']
+        strObjectData = data['str_object']
         if current_setting == "openai":
             # 反向时把“源语种代码”当成目标，复用同一 prompt 模板
-            result = await openai_translate(text, data_setting['translate_source_lang'])
+            result = await openai_translate(strObjectData, data_setting['translate_source_lang'])
+        elif current_setting == "other_ai_plate":
+            aiInfoData = get_ai_info_setting()
+            if aiInfoData.get("base_url") == "https://api.siliconflow.cn/v1":
+                result = await translateObject(strObjectData, data_setting['translate_source_lang'])
         else:
             result = translateText(
                 text,
                 data_setting['translate_service'],
                 data_setting['translate_target_lang'],
                 data_setting['translate_source_lang']
             )
-        return web.json_response({"text": result})
+        return web.json_response({"data": result})
     except Exception as e:
         print(f"Error: {e}")
         return web.Response(status=500)
@@ -1036,6 +1047,35 @@ async def _run_danbooru_sql_text(request):
     return web.json_response(result)
 # ===================================================================================
 
+# ============================================ AI平台对接 ============================================
+
+@PromptServer.instance.routes.post(baseUrl+"ai_server/get_settings")
+async def _get_ai_server_setting(request):
+    return  web.json_response({"data":get_ai_info_setting()})
+
+
+@PromptServer.instance.routes.post(baseUrl+"ai_server/update_settings")
+async def _update_ai_server_settings(request):
+    data = await request.json()
+    try:
+        # 如果前端传递 {"ai_info_setting": {...}}，则取内部 dict
+        if "ai_info_setting" in data:
+            update_ai_info_setting(data["ai_info_setting"])
+        else:
+            update_ai_info_setting(data)
+    except Exception as e:
+        print(f"Error: {e}")
+        return web.Response(status=500)
+    return web.json_response({"info": 'ok'})
+
+@PromptServer.instance.routes.post(baseUrl+"ai_server/get_ai_models")
+def _get_ai_server_get_ai_models(request):
+    return  web.json_response({"data": getModelList()})
+
+
+
+# ============================================ AI平台对接 End ============================================
+
 print("======== WeiLin插件服务已启动 ========")
 print("======== WeiLin Server Init ========")
 

app/server/prompt_server.py

@@ -9,11 +9,13 @@
     "ru": "俄语", "it": "意大利语", "pt": "葡萄牙语"
 }
 
+
 def _lang_to_name(code: str) -> str:
     if not code:
         return "中文"
     return _LANG_NAME.get(code, code)
 
+
 async def openai_translate(text: str, target_lang_code: str) -> str:
     cfg = initialize_config()
     api_key = cfg.get("api_key", "")
@@ -23,21 +25,26 @@ async def openai_translate(text: str, target_lang_code: str) -> str:
     if not api_key:
         raise RuntimeError("OpenAI api_key 未配置")
 
-    target_lang_name = _lang_to_name(target_lang_code)
-    prompt = f"将以下AI绘画提示词翻译成{target_lang_name}，只输出翻译结果：{text}"
-
     url = f"{base_url}/chat/completions"
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
     }
     payload = {
         "model": model,
-        "messages": [{"role": "user", "content": prompt}],
+        "messages": [
+            {
+                "role": "system",
+                "content": f"You are an expert in data translation processing. translate the text field of the JSON string data passed by the user into {_lang_to_name(target_lang_code)} and fill it in the corresponding translate field. You only need to return the corresponding JSON string data and do not modify any other data or parameters"
+            },
+            {"role": "user", "content": text}
+        ],
+        "response_format": {"type": "json_object"},
         "temperature": 0
     }
 
-    async with aiohttp.ClientSession() as session:
+    # 配置会话以使用系统环境变量中的代理设置
+    async with aiohttp.ClientSession(trust_env=True) as session:
         async with session.post(url, headers=headers, json=payload, timeout=60) as resp:
             data = await resp.json()
             if resp.status != 200:
@@ -47,11 +54,14 @@ async def openai_translate(text: str, target_lang_code: str) -> str:
             usage = data.get("usage", {})
             prompt_tokens = usage.get("prompt_tokens", 0)
             completion_tokens = usage.get("completion_tokens", 0)
-            total_tokens = usage.get("total_tokens", prompt_tokens + completion_tokens)
+            total_tokens = usage.get(
+                "total_tokens", prompt_tokens + completion_tokens)
 
             # 控制台日志
-            print(f"🤖 正在使用OpenAI翻译: {text}")
-            print(f"📊 OpenAI翻译tokens使用: {prompt_tokens}+{completion_tokens}={total_tokens}")
-            print(f"✅ OpenAI翻译成功: {text} -> {result}")
+            # print(f"🤖 正在使用OpenAI翻译: {text}")
+            print(f"🤖 正在使用OpenAI翻译")
+            print(
+                f"📊 OpenAI翻译tokens使用: {prompt_tokens}+{completion_tokens}={total_tokens}")
+            # print(f"✅ OpenAI翻译成功: {text} -> {result}")
 
             return result

app/server/translate/openai_translate.py

@@ -52,6 +52,14 @@
     "translate_target_lang": "zh",
     "show_auto_limit": 25,
     "random_template": "",
+    "ai_info_setting": {
+        "api_key": "",
+        "base_url": "https://api.siliconflow.cn/v1",
+        "model": "THUDM/glm-4-9b-chat",
+        "temperature": 0,
+        "top_p": 0.7,
+        "max_tokens": 4096
+    }
 }
 
 def read_init_file():
@@ -176,6 +184,30 @@ def update_random_template_setting(new_setting):
     """更新random_template参数"""
     data = read_init_file() or {}
     data['random_template'] = new_setting
+    with open(init_file_path, 'w', encoding='utf-8') as f:
+        json.dump(data, f, ensure_ascii=False, indent=4)
+    return True
+
+def get_ai_info_setting():
+    """获取ai_info_setting参数，如果不存在则添加默认值"""
+    data = read_init_file() or {}
+    if 'ai_info_setting' not in data:
+        data['ai_info_setting'] = {
+            "api_key": "",
+            "base_url": "https://api.siliconflow.cn/v1",
+            "model": "THUDM/glm-4-9b-chat",
+            "temperature": 0,
+            "top_p": 0.7,
+            "max_tokens": 4096
+        }
+        with open(init_file_path, 'w', encoding='utf-8') as f:
+            json.dump(data, f, ensure_ascii=False, indent=4)
+    return data['ai_info_setting']
+
+def update_ai_info_setting(new_setting: dict):
+    """更新ai_info_setting参数"""
+    data = read_init_file() or {}
+    data['ai_info_setting'] = new_setting
     with open(init_file_path, 'w', encoding='utf-8') as f:
         json.dump(data, f, ensure_ascii=False, indent=4)
     return True