Update app/webui/app.py

app/webui/app.py

@@ -1,5 +1,6 @@
 import sys
 import os
+from translatepy.translators.google import GoogleTranslate
 
 # Add the project root to the Python path
 project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
@@ -11,6 +12,8 @@ from glob import glob
 from app.webui.process import model_load, diff_texts, translator, translator_sec
 from llama_index.core import SimpleDirectoryReader
 
+gtranslate = GoogleTranslate()
+
 def huanik(
     endpoint: str,
     model: str,
@@ -27,6 +30,7 @@ def huanik(
     context_window: int,
     num_output: int,
     rpm: int,
+    gt: bool,
 ):
 
     if not source_text or source_lang == target_lang:
@@ -39,7 +43,7 @@ def huanik(
 
     source_text = re.sub(r'(?m)^\s*$\n?', '', source_text)
 
-    if choice:
+    if choice or gt:
         init_translation, reflect_translation, final_translation = translator_sec(
             endpoint2=endpoint2,
             model2=model2,
@@ -51,6 +55,7 @@ def huanik(
             source_text=source_text,
             country=country,
             max_tokens=max_tokens,
+            gt = gt,
         )
 
     else:
@@ -251,6 +256,10 @@ with gr.Blocks(theme="soft", css=CSS, fill_height=True) as demo:
                     value=60,
                     step=1,
                     )
+                gt = gr.Checkbox(
+                    label="Google Translate",
+                    info="Use Google Translate as Init Translation"
+                )
         with gr.Column(scale=4):
             source_text = gr.Textbox(
                 label="Source Text",
@@ -282,7 +291,7 @@ with gr.Blocks(theme="soft", css=CSS, fill_height=True) as demo:
     choice.select(fn=enable_sec, inputs=[choice], outputs=[AddEndpoint])
     endpoint2.change(fn=update_model, inputs=[endpoint2], outputs=[model2])
    
-    start_ta = submit.click(fn=huanik, inputs=[endpoint, model, api_key, choice, endpoint2, model2, api_key2, source_lang, target_lang, source_text, country, max_tokens, context_window, num_output, rpm], outputs=[output_init, output_reflect, output_final, output_diff])
+    start_ta = submit.click(fn=huanik, inputs=[endpoint, model, api_key, choice, endpoint2, model2, api_key2, source_lang, target_lang, source_text, country, max_tokens, context_window, num_output, rpm, gt], outputs=[output_init, output_reflect, output_final, output_diff])
     upload.upload(fn=read_doc, inputs = upload, outputs = source_text)
     output_final.change(fn=export_txt, inputs=output_final, outputs=[export])