Update app.py

app.py

@@ -9,14 +9,41 @@ 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 create_fantasy_story(heros_name: str, number_of_chapter: int) -> str:
+    """A tool that generates a fantasy story based on a hero's name and a specified number of chapters.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        arg1: The name of the hero or heroine.
+        arg2: The number of chapters in the story.
     """
-    return "What magic will you build ?"
+    # Story templates for chapters
+    settings = [
+        "in the deep, enchanted forests",
+        "on the rugged, misty mountains",
+        "in the mysterious ruins of an ancient civilization",
+        "in the bustling city of magic"
+    ]
+    conflicts = [
+        "facing a dragon that terrorizes the land",
+        "battling a dark sorceress who unleashes sinister magic",
+        "confronting an ancient curse that turns villagers to stone",
+        "challenging an army of shadow warriors threatening the kingdom"
+    ]
+    resolutions = [
+        "bravely saving the day",
+        "finding a magical ally who helped defeat the threat",
+        "discovering the secret of an ancient spell",
+        "engaging in an epic battle that became legendary"
+    ]
+    
+    story = f"The Story of {arg1}:\n\n"
+    for i in range(1, arg2 + 1):
+        setting = random.choice(settings)
+        conflict = random.choice(conflicts)
+        resolution = random.choice(resolutions)
+        chapter = (f"Chapter {i}: {arg1} ventures {setting} to confront "
+                   f"{conflict} and ultimately {resolution}.\n")
+        story += chapter
+    return story
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +82,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=[create_fantasy_story,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,