Update app.py

app.py

@@ -11,8 +11,10 @@ from langdetect import detect
 import requests
 from smolagents import tool
 
-DEEPL_API_KEY = "" 
-DEEPL_FREE_ENDPOINT = "https://api-free.deepl.com/v2/translate"
+model_name = "facebook/m2m100_418M"
+tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+translation_model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+
 
 @tool
 def agent_translate(text: str) -> str:
@@ -43,14 +45,19 @@ def agent_translate(text: str) -> str:
     }
 
     try:
-        #response = requests.post(DEEPL_FREE_ENDPOINT, data=params)
-        #response.raise_for_status()
-        #result = response.json()
-        #translated_text = result["translations"][0]["text"]
-        #return translated_text
-        return ", ".join([detected_lang, target_lang, original])
+        inputs = tokenizer(original, return_tensors="pt")
+
+        forced_bos_token_id = tokenizer.get_lang_id(target_lang)
+
+        generated_tokens = translation_model.generate(
+            **inputs,
+            forced_bos_token_id=forced_bos_token_id
+        )
+
+        translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+        return translated_text
     except Exception as e:
-        return f"Error during translating using DeepL: {str(e)}"
+        return f"Error during translating: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: