Update translator.py

translator.py

@@ -1,30 +1,23 @@
-import logging
 from transformers import pipeline
 
-logger = logging.getLogger(__name__)
-
 class Translator:
     def __init__(self):
         self.model_name = "Helsinki-NLP/opus-mt-en-mul"
         self.translator = None
-        logger.info("[Translator] Initialized")
 
     def translate(self, text, target_lang):
-        if not text or not text.strip(): return ""
-        
-        lang_map = {
-            "Hindi": ">>hin<<", "Telugu": ">>tel<<", 
-            "Spanish": ">>spa<<", "French": ">>fra<<", "German": ">>deu<<"
-        }
+        if not text: return ""
         
-        prefix = lang_map.get(target_lang, ">>hin<<")
-        input_text = f"{prefix} {text}"
+        lang_codes = {"Hindi": ">>hin<<", "Telugu": ">>tel<<", "Spanish": ">>spa<<", 
+                      "French": ">>fra<<", "German": ">>deu<<", "Japanese": ">>jpn<<", "Arabic": ">>ara<<"}
         
+        prefix = lang_codes.get(target_lang, ">>hin<<")
         try:
             if self.translator is None:
-                self.translator = pipeline("translation", model=self.model_name)
+                # FIX: Using 'text2text-generation' as the task name
+                self.translator = pipeline("text2text-generation", model=self.model_name)
             
-            result = self.translator(input_text)
-            return result[0]['translation_text']
+            res = self.translator(f"{prefix} {text}")
+            return res[0]['generated_text']
         except Exception as e:
-            return f"[Translation Error: {e}]"
+            return f"[Translation error: {str(e)}]"