Update app.py

app.py

@@ -4,8 +4,22 @@ import datetime
 import pytz
 import math
 import os
+from deep_translator import GoogleTranslator
 
 # ============ TOOLS ============
+@tool
+def translator_tool(text: str, target_language: str) -> str:
+    """Translates text into a specified language.
+    Args:
+        text: The text or phrase to translate.
+        target_language: The destination language (e.g., 'french', 'german', 'japanese', 'hindi').
+    """
+    try:
+        # GoogleTranslator handles full language names or ISO codes
+        translation = GoogleTranslator(source='auto', target=target_language).translate(text)
+        return f"Translated to {target_language.title()}: {translation}"
+    except Exception as e:
+        return f"Translation error: {str(e)}. Make sure the language name is correct."
 
 @tool
 def web_search_tool(query: str) -> str:
@@ -116,9 +130,9 @@ model = InferenceClientModel(
 
 agent = CodeAgent(
     model=model,
-    tools=[web_search_tool, time_tool, calculator_tool],
+    tools=[web_search_tool, time_tool, calculator_tool, translator_tool],
     max_steps=5,
-    additional_authorized_imports=['math', 'datetime', 'pytz']
+    additional_authorized_imports=['math', 'datetime', 'pytz', 'deep_translator']
 )
 
 # ============ GRADIO UI ============
@@ -141,6 +155,7 @@ demo = gr.ChatInterface(
         "What time is it in Tokyo?",
         "Convert 150 pounds to kg",
         "Calculate the square root of 144 plus 50",
+        "Translate 'Hello, how are you today?' into Japanese",
         "Search for the latest news on AI agents"
     ],
 )