arxe commited on
Commit
6d169cc
·
verified ·
1 Parent(s): 62ae93d

Create ThaiToEnglishPrompt.py

Browse files
Files changed (1) hide show
  1. ThaiToEnglishPrompt.py +27 -0
ThaiToEnglishPrompt.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from googletrans import Translator
2
+
3
+ class ThaiToEnglishPrompt:
4
+ @classmethod
5
+ def INPUT_TYPES(cls):
6
+ return {
7
+ "required": {
8
+ "thai_prompt": ("STRING", {"multiline": True}),
9
+ }
10
+ }
11
+
12
+ RETURN_TYPES = ("STRING",)
13
+ FUNCTION = "translate"
14
+ CATEGORY = "Prompt/Translate"
15
+
16
+ def translate(self, thai_prompt):
17
+ translator = Translator()
18
+ result = translator.translate(thai_prompt, src='th', dest='en')
19
+ return (result.text,)
20
+
21
+ NODE_CLASS_MAPPINGS = {
22
+ "ThaiToEnglishPrompt": ThaiToEnglishPrompt
23
+ }
24
+
25
+ NODE_DISPLAY_NAME_MAPPINGS = {
26
+ "ThaiToEnglishPrompt": "แปลไทย ➜ อังกฤษ (Google)"
27
+ }