Update app.py

app.py

@@ -6,6 +6,8 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+import random
+
 
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -32,7 +34,6 @@ def generate_image_with_prompt(prompt: str) -> str:
 @tool
 def random_fact() -> str:
     """Fetches a random fun fact."""
-    import random
     facts = [
         "Honey never spoils.",
         "Octopuses have three hearts.",
@@ -42,16 +43,17 @@ def random_fact() -> str:
 
 
 @tool
-def tell_joke()-> str:
+def tell_joke() -> str:
     """Returns a random joke."""
-    import random
-    jokes = [
-        "Why don't skeletons fight each other? They don't have the guts.",
-        "I told my wife she was drawing her eyebrows too high. She looked surprised.",
-        "Why don’t scientists trust atoms? Because they make up everything!"
-    ]
-    return random.choice(jokes)
-
+    try:
+        jokes = [
+            "Why don't skeletons fight each other? They don't have the guts.",
+            "I told my wife she was drawing her eyebrows too high. She looked surprised.",
+            "Why don’t scientists trust atoms? Because they make up everything!"
+        ]
+        return random.choice(jokes)
+    except Exception as e:
+        return f"Error fetching joke: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: