Update app.py

Try to introduce fairytale tool.

app.py

@@ -9,14 +9,30 @@ 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
+def my_fairytale_tool(arg: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 
+    """A tool that returns a heads-up string of two fairytiles' titles
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        arg: random number
     """
-    return "What magic will you build ?"
+
+        fairytales = [
+        "Cinderella",
+        "Snow White",
+        "Little Red Riding Hood",
+        "The Ugly Duckling",
+        "The Frog Prince"
+    ]
+    
+    # Randomly select  fairytales from the list
+    selected_1 = fairytales[int(arg / 5)]
+    fairytales.pop(int(arg / 4))
+    selected_2 = fairytales[random.randint(0,4)]
+
+
+    
+    return f"What a crazy idea to see a story of {selected_1} vs {selected_2}"
+    #return "What magic will you build ?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +71,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, image_generation_tool, my_fairytale_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,