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)"
}