sd-webui-prompt-all-in-one/scripts/physton_prompt/translator/translators_translator.py

21 lines
738 B
Python

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')
os.environ['translators_default_region'] = region
from translators.server import translate_text, tss
tss.server_region = region
tss._bing.server_region = region
tss._google.server_region = region
return translate_text(text, from_language=self.from_lang, to_language=self.to_lang, translator=self.translator,
timeout=30)