Update app/webui/process.py

app/webui/process.py

@@ -7,11 +7,12 @@ from app.webui.patch import calculate_chunk_size, multichunk_initial_translation
 
 from llama_index.core.node_parser import SentenceSplitter
 
+from translatepy.translators import GoogleTranslate, DeeplTranslate
+from translatepy import Translator
 from translatepy.exceptions import UnknownLanguage
-from translatepy.translators.google import GoogleTranslate
 from translatepy import Language
 
-gtranslate = GoogleTranslate()
+translator = Translator(services_list=[DeeplTranslate, GoogleTranslate])
 
 progress=gr.Progress()
 
@@ -163,7 +164,7 @@ def translator_sec(
                 language = Language(target_lang)
             except Exception as e:
                 raise gr.Error(f"An unexpected error occurred: {e}")                
-            init_translation = gtranslate.translate(source_text, language)
+            init_translation = translator.translate(source_text, target_lang)
         else:
             init_translation = one_chunk_initial_translation(
                 source_lang, target_lang, source_text
@@ -209,7 +210,7 @@ def translator_sec(
                 language = Language(target_lang)
             except Exception as e:
                 raise gr.Error(f"An unexpected error occurred: {e}")
-            translation_1_chunks = gtranslate.translate(source_text_chunks, language)
+            translation_1_chunks = translator.translate(source_text_chunks, target_lang)
         else:
             translation_1_chunks = multichunk_initial_translation(
                 source_lang, target_lang, source_text_chunks