comfy_thai_translate / ThaiToEnglishPrompt.py
arxe's picture
Create ThaiToEnglishPrompt.py
6d169cc verified
from googletrans import Translator
class ThaiToEnglishPrompt:
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"thai_prompt": ("STRING", {"multiline": True}),
}
}
RETURN_TYPES = ("STRING",)
FUNCTION = "translate"
CATEGORY = "Prompt/Translate"
def translate(self, thai_prompt):
translator = Translator()
result = translator.translate(thai_prompt, src='th', dest='en')
return (result.text,)
NODE_CLASS_MAPPINGS = {
"ThaiToEnglishPrompt": ThaiToEnglishPrompt
}
NODE_DISPLAY_NAME_MAPPINGS = {
"ThaiToEnglishPrompt": "แปลไทย ➜ อังกฤษ (Google)"
}