Update app.py

Added a few tools to the agent.

app.py

@@ -9,14 +9,42 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> 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 does nothing yet 
+def generate_funny_poem(theme: str, lines: int) -> str:
+    """A tool that generates a short, humorous poem based on a theme.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        theme: The theme of the poem (e.g., 'cats', 'space', 'love').
+        lines: The number of lines in the poem (between 2 and 6).
     """
-    return "What magic will you build ?"
+    funny_lines = {
+        "cats": [
+            "The cat just stole my dinner plate,",
+            "It looked at me, then licked its fate.",
+            "Nine lives, but none for rent,",
+            "Sleeps all day, pays no rent!"
+        ],
+        "space": [
+            "A rocket zoomed across the night,",
+            "Aliens waved, quite polite!",
+            "I asked them for a cup of tea,",
+            "They said, 'No thanks, we drink gravity!'"
+        ],
+        "love": [
+            "Roses are red, violets are blue,",
+            "I bought you coffee, now marry me too!",
+            "You stole my heart, just like my fries,",
+            "But I forgive you—those puppy eyes!"
+        ],
+        "default": [
+            "Life’s a joke, and so am I,",
+            "Still, I try and still I sigh.",
+            "If laughter’s gold, then I'm quite broke,",
+            "Guess I'll survive on just my jokes!"
+        ]
+    }
+
+    selected_lines = funny_lines.get(theme, funny_lines["default"])
+    poem = "\n".join(random.sample(selected_lines, min(lines, len(selected_lines))))
+    return f"Here's a fun little poem about {theme}:\n\n{poem}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +83,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_funny_poem, image_generation_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,