Update app.py

app.py

@@ -7,6 +7,21 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+# Translation tool
+@tool
+def translate_text(sentence: str, target_language: str) -> str:
+    """A tool that translates a given sentence into the specified target language.
+    Args:
+        sentence: The sentence to translate.
+        target_language: The language to translate the sentence into (e.g., 'fr' for French, 'es' for Spanish).
+    """
+    try:
+        translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-{target_language}")
+        translation = translator(sentence)[0]['translation_text']
+        return f"Translated text ({target_language}): {translation}"
+    except Exception as e:
+        return f"Error in translation: {str(e)}"
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -55,7 +70,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, translate_text], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,