Update app.py

app.py

@@ -4,21 +4,23 @@ import requests
 import pytz
 import yaml
 import json
+from googletrans import Translator
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 @tool
-def get_random_joke() -> str:
-    """A tool that fetches a random joke."""
-    url = "https://official-joke-api.appspot.com/random_joke"
-    
+def translate_to_norwegian(text: str) -> str:
+    """A tool that translates the input text to Norwegian using Google Translate.
+    Args:
+        text: The text to translate.
+    """
+    translator = Translator()
     try:
-        response = requests.get(url).json()
-        joke = f"{response['setup']} {response['punchline']}"
-        return joke
+        translation = translator.translate(text, dest="no")
+        return translation.text
     except Exception as e:
-        return f"Error fetching joke: {str(e)}"
+        return f"Error translating text: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -56,7 +58,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_random_joke, final_answer], ## add your tools here (don't remove final answer)
+    tools=[translate_to_norwegian, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,