Update app.py

joke generator tool

app.py

@@ -1,4 +1,6 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from datasets import load_dataset 
+import random
 import datetime
 import requests
 import pytz
@@ -18,6 +20,26 @@ def get_domain_name(email:str)-> str: #it's import to specify the return type
     domain = email.split('@')[1]
     return domain
 
+@tool
+def generate_jokes(joke_type:str)-> str: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """A tool that retrieves a joke of specified type. Two types are accepted 'dad jokes' and regular jokes
+    Args:
+        joke_type: a string that specifies the type of joke to retrieve. Accepted arguments 'dad' or 'regular'
+    """
+        if joke_type == 'dad':
+            path = 'gnumanth/dad-jokes'
+            col = 'joke'
+        else:
+            path = 'Amirkid/jokes'
+            col = 'text'
+        dataset = load_dataset(path, split = "train")[col]
+        # shuffle to make sure examples will be random
+        dataset = dataset.shuffle()
+        random_example = dataset[[col]][0]
+        
+    return random_example
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.