Update app/webui/app.py

app/webui/app.py

@@ -27,6 +27,7 @@ def huanik(
     context_window: int,
     num_output: int,
     rpm: int,
+    gt: bool,
 ):
 
     if not source_text or source_lang == target_lang:
@@ -39,7 +40,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 +52,7 @@ def huanik(
             source_text=source_text,
             country=country,
             max_tokens=max_tokens,
+            gt = gt,
         )
 
     else:
@@ -120,6 +122,12 @@ def closeBtnHide(output_final):
     if output_final:
         return gr.update(visible=True), gr.update(visible=False)
 
+def hide_add(gt):
+    if gt:
+        return gr.update(visible=False), gr.update(visible=False)
+    else:
+        return gr.update(value=False, visible=True), gr.update(visible=False)
+
 TITLE = """
     <div style="display: inline-flex;">
         <div style="margin-left: 6px; font-size:32px; color: #6366f1"><b>Translation Agent</b> WebUI</div>