Upload 2 files

app.py

@@ -52,7 +52,7 @@ demo = gr.Interface(fn=predict,
                             gr.Dropdown([("Google", "google"),
                                          ("Helsinki-NLP's opus-mt-ja-en model",
                                           "hf"),
-                                         ("Baidu", "baidu"),
+                                         ("Sogou", "sogou"),
                                          ("Bing", "bing")],
                                         label="Translation Method",
                                         value="google"),

translator.py

@@ -1,6 +1,7 @@
 from deep_translator import GoogleTranslator
 from transformers import pipeline
 import translators as ts
+import random
 import time
 
 
@@ -11,7 +12,7 @@ class MangaTranslator:
         self.translators = {
             "google": self._translate_with_google,
             "hf": self._translate_with_hf,
-            "baidu": self._translate_with_baidu,
+            "sogou": self._translate_with_sogou,
             "bing": self._translate_with_bing
         }
 
@@ -23,20 +24,21 @@ class MangaTranslator:
             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)
-                         "baidu" for Baidu Translate
+                         "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
@@ -46,14 +48,15 @@ class MangaTranslator:
         translated_text = pipe(text)[0]["translation_text"]
         return translated_text if translated_text is not None else text
 
-    def _translate_with_baidu(self, text):
-        translated_text = ts.translate_text(text, translator="baidu",
-                                            from_language="jp", 
+    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):
-        time.sleep(4)
+        self._delay()
         translated_text = ts.translate_text(text, translator="bing",
                                             from_language=self.source, 
                                             to_language=self.target)
@@ -61,4 +64,6 @@ class MangaTranslator:
 
     def _preprocess_text(self, text):
         preprocessed_text = text.replace("．", ".")
-        return preprocessed_text
+
+    def _delay(self):
+        time.sleep(random.randint(3, 5))