Update app.py

Update CodeAgent

app.py

@@ -9,24 +9,21 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def restaurant_service_tool(query: str) -> str: #it's import to specify the return type
+def suggest_menu(query: 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
     """
-    This tool returns the highest-rated restaurant in Copenhagen.
-    
+    Suggests a menu based on the occasion.
     Args:
-        query: A search term for finding restaurant.
+        occasion: The type of occasion for the party.
     """
-    # Example list of catering services and their ratings
-    services = {
-        "Geranium": 4.7,
-        "Alchemist": 4.9,
-        "Maple Casual Dining": 4.8,
-    }
-    # Find the highest rated catering service (simulating search query filtering)
-    best_service = max(services, key=services.get)
-     
-    return best_service
+    if occasion == "casual":
+        return "Pizza, snacks, and drinks."
+    elif occasion == "formal":
+        return "3-course dinner with wine and dessert."
+    elif occasion == "superhero":
+        return "Buffet with high-energy and healthy food."
+    else:
+        return "Custom menu for the butler."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -66,7 +63,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[restaurant_service_tool, DuckDuckGoSearchTool(), final_answer], ## add your tools here (don't remove final answer)
+    tools=[DuckDuckGoSearchTool(), suggest_menu, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,