Update app.py

app.py

@@ -222,6 +222,23 @@ class HuggingFaceImageQATool(Tool):
         return self.processor.decode(out[0], skip_special_tokens=True)
 
 
+from transformers import pipeline
+
+class HuggingFaceTranslationTool(Tool):
+    name = "translate"
+    description = "Translate text from English to another language."
+    inputs = {
+        "text": {"type": "string", "description": "Text to translate"}
+    }
+    output_type = "string"
+
+    def __init__(self):
+        self.translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
+
+    def forward(self, text: str) -> str:
+        return self.translator(text)[0]["translation_text"]
+
+
 # ===== Code Execution =====
 class PythonCodeExecutionTool:
     def __init__(self):