Update utils/translator.py

utils/translator.py

@@ -70,6 +70,18 @@ def translate_to_english(text: str) -> str:
     chunks = chunk_text(clean_text(text))
     return translate_chunks(chunks, tokenizer_pt_en, model_pt_en)
 
+def translate_text(text: str, direction: str = "en-pt") -> str:
+    """
+    🌐 Wrapper para tradução direta via Gradio.
+    direction = "en-pt" ou "pt-en"
+    """
+    if direction == "en-pt":
+        return translate_to_portuguese(text)
+    elif direction == "pt-en":
+        return translate_to_english(text)
+    else:
+        return "Unsupported translation direction."
+
 # ========== Bilingual View ==========
 
 def bilingual_clauses(english_text: str) -> str: