Update app.py

app.py

@@ -9,23 +9,27 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
+from smolagents import tool, DuckDuckGoSearchTool
+
 @tool
-def my_custom_tool(arg1: str) -> str:
-    """A tool that generates a funny coding joke.  
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """A tool that generates a funny coding joke.
 
-    Args:  
-        arg1 (str): Topic of the joke.  
+    Args:
+        arg1 (str): Level of funny (e.g., "low", "medium", "high").
+        arg2 (int): Topic of the joke (e.g., 1 for Python, 2 for JavaScript).
 
-    Returns:  
-        str: A generated joke.  
+    Returns:
+        str: A generated joke.
     """
-    search_query = f"funny programming joke about {arg1}"
+    search_query = f"funny programming joke about {arg2}"
     results = DuckDuckGoSearchTool(search_query)
 
     if results:
-        return f"The joke on {arg1} : {results[0]}"  
+        return f"The joke on {arg2} at level {arg1}: {results[0]}"
     else:
-        return f"Sorry, I couldn't find a joke about {arg1}."
+        return f"Sorry, I couldn't find a joke about {arg2}."
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: