| from deep_translator import GoogleTranslator |
| from transformers import pipeline |
| import translators as ts |
| import random |
| import time |
|
|
|
|
| class MangaTranslator: |
| def __init__(self): |
| self.target = "en" |
| self.source = "ja" |
| self.translators = { |
| "google": self._translate_with_google, |
| "hf": self._translate_with_hf, |
| "sogou": self._translate_with_sogou, |
| "bing": self._translate_with_bing |
| } |
|
|
| def translate(self, text, method="google"): |
| """ |
| Translates the given text to the target language using the specified method. |
| |
| Args: |
| text (str): The text to be translated. |
| method (str):"google" for Google Translator, |
| "hf" for Helsinki-NLP's opus-mt-ja-en model (HF pipeline) |
| "sogou" for Sogou Translate |
| "bing" for Microsoft Bing Translator |
| |
| Returns: |
| str: The translated text. |
| """ |
| translator_func = self.translators.get(method) |
|
|
| if translator_func: |
| return translator_func(self._preprocess_text(text)) |
| else: |
| raise ValueError("Invalid translation method.") |
| |
| def _translate_with_google(self, text): |
| self._delay() |
| translator = GoogleTranslator(source=self.source, target=self.target) |
| translated_text = translator.translate(text) |
| return translated_text if translated_text is not None else text |
|
|
| def _translate_with_hf(self, text): |
| pipe = pipeline("translation", model=f"Helsinki-NLP/opus-mt-ja-en") |
| translated_text = pipe(text)[0]["translation_text"] |
| return translated_text if translated_text is not None else text |
|
|
| def _translate_with_sogou(self, text): |
| self._delay() |
| translated_text = ts.translate_text(text, translator="sogou", |
| from_language=self.source, |
| to_language=self.target) |
| return translated_text if translated_text is not None else text |
|
|
| def _translate_with_bing(self, text): |
| self._delay() |
| translated_text = ts.translate_text(text, translator="bing", |
| from_language=self.source, |
| to_language=self.target) |
| return translated_text if translated_text is not None else text |
|
|
| def _preprocess_text(self, text): |
| preprocessed_text = text.replace("．", ".") |
| return preprocessed_text |
|
|
| def _delay(self): |
| time.sleep(random.randint(3, 5)) |
|
|
