IndicTTS-Malayalam / src /utils /translator.py
trysem's picture
Upload 14 files
a920b41 verified
raw
history blame contribute delete
824 Bytes
class GoogleTranslator:
def __init__(self):
from translators.server import google, _google
self._translate = google
google("Testing...")
self.supported_languages = set(_google.language_map['en'])
self.custom_lang_map = {
"mni": "mni-Mtei",
"raj": "hi",
}
def translate(self, text, from_lang, to_lang):
if from_lang in self.custom_lang_map:
from_lang = self.custom_lang_map[from_lang]
elif from_lang not in self.supported_languages:
return text
if to_lang in self.custom_lang_map:
to_lang = self.custom_lang_map[to_lang]
elif to_lang not in self.supported_languages:
return text
return self._translate(text, from_language=from_lang, to_language=to_lang)
def __call__(self, **kwargs):
return self.translate(**kwargs)