Upload 3 files

app.py

@@ -22,7 +22,7 @@ def predict(img, translation_method, font):
         translation_method = "google"
     if font == None:
         font = "fonts/animeace_i.ttf"
-    
+
     results = detect_bubbles(MODEL, img)
 
     manga_translator = MangaTranslator()
@@ -39,8 +39,8 @@ def predict(img, translation_method, font):
         text = mocr(im)
 
         detected_image, cont = process_bubble(detected_image)
-        
-        text_translated = manga_translator.translate(text, 
+
+        text_translated = manga_translator.translate(text,
                                                      method=translation_method)
 
         image_with_text = add_text(detected_image, text_translated, font, cont)
@@ -49,9 +49,11 @@ def predict(img, translation_method, font):
 
 demo = gr.Interface(fn=predict,
                     inputs=["image",
-                            gr.Dropdown([("Google", "google"), 
+                            gr.Dropdown([("Google", "google"),
                                          ("Helsinki-NLP's opus-mt-ja-en model",
-                                          "hf")], 
+                                          "hf"),
+                                         ("Baidu", "baidu"),
+                                         ("Bing", "bing")],],
                                         label="Translation Method",
                                         value="google"),
                             gr.Dropdown([("animeace_i", "fonts/animeace_i.ttf"),

requirements

@@ -0,0 +1,8 @@
+deep-translator==1.11.4
+translators==5.9.1
+huggingface-hub==0.22.2
+manga-ocr==0.1.11
+numpy==1.24.2
+opencv-python==4.9.0.80
+pillow==10.3.0
+ultralytics==8.1.43

translator.py

@@ -1,11 +1,18 @@
-from deep_translator import GoogleTranslator
+from deep_translator import GoogleTranslator, LingueeTranslator
 from transformers import pipeline
+import translators as ts
 
 
 class MangaTranslator:
     def __init__(self):
         self.target = "en"
         self.source = "ja"
+        self.translators = {
+            "google": self._translate_with_google,
+            "hf": self._translate_with_hf,
+            "baidu": self._translate_with_baidu,
+            "bing": self._translate_with_bing
+        }
 
     def translate(self, text, method="google"):
         """
@@ -13,16 +20,18 @@ class MangaTranslator:
 
         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)
+            method (str):"google" for Google Translator, 
+                         "hf" for Helsinki-NLP's opus-mt-ja-en model (HF pipeline)
+                         "baidu" for Baidu Translate
+                         "bing" for Microsoft Bing Translator
 
         Returns:
             str: The translated text.
         """
-        if method == "hf":
-            return self._translate_with_hf(self._preprocess_text(text))
-        elif method == "google":
-            return self._translate_with_google(self._preprocess_text(text))
+        translator_func = self.translators.get(method)
+        
+        if translator_func:
+            return translator_func(self._preprocess_text(text))
         else:
             raise ValueError("Invalid translation method.")
             
@@ -36,6 +45,18 @@ class MangaTranslator:
         translated_text = pipe(text)[0]["translation_text"]
         return translated_text if translated_text != None else text
 
+    def _translate_with_baidu(self, text):
+        translated_text = ts.translate_text(text, translator="baidu",
+                                            from_language="jp", 
+                                            to_language=self.target)
+        return translated_text if translated_text != None else text
+
+    def _translate_with_bing(self, text):
+        translated_text = ts.translate_text(text, translator="bing",
+                                            from_language=self.source, 
+                                            to_language=self.target)
+        return translated_text if translated_text != None else text
+
     def _preprocess_text(self, text):
         preprocessed_text = text.replace("．", ".")
         return preprocessed_text