FashGate's picture
Upload 204 files
8a469fd
from scripts.physton_prompt.translator.base_tanslator import BaseTranslator
import os
class TranslatorsTranslator(BaseTranslator):
translator = None
def set_translator(self, translator):
self.translator = translator
return self
def translate(self, text):
region = self.api_config.get('region', 'CN')
host = self.api_config.get('host', '')
os.environ['translators_default_region'] = region
from scripts.physton_prompt.translators.server import translate_text, tss, AlibabaV1
tss.server_region = region
tss._bing.server_region = region
tss._google.server_region = region
params = {
'from_language': self.from_lang,
'to_language': self.to_lang,
'translator': self.translator,
'reset_host_url': host,
'if_check_reset_host_url': False,
'timeout': 30
}
return translate_text(text, **params)
# return translate_text(text, from_language=self.from_lang, to_language=self.to_lang, translator=self.translator, timeout=30)