Create utils/translator.py

utils/translator.py

@@ -0,0 +1,34 @@
+# utils/translator.py
+
+import os
+from transformers import MarianMTModel, MarianTokenizer
+
+DEEPL_API_KEY = os.getenv("DEEPL_API_KEY")
+
+# Fallback model (HF)
+model_name = "Helsinki-NLP/opus-mt-en-pt"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
+
+def translate_text(text):
+    if not text.strip():
+        return "No input to translate."
+
+    try:
+        import requests
+
+        response = requests.post(
+            "https://api.deepl.com/v2/translate",
+            data={
+                "auth_key": DEEPL_API_KEY,
+                "text": text,
+                "target_lang": "PT"
+            },
+        )
+        return response.json()["translations"][0]["text"]
+
+    except Exception:
+        # Use Hugging Face fallback model
+        inputs = tokenizer.encode(text, return_tensors="pt", truncation=True)
+        outputs = model.generate(inputs, max_length=512, num_beams=4)
+        return tokenizer.decode(outputs[0], skip_special_tokens=True)