Update app.py

app.py

@@ -36,6 +36,36 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def recommend_diet_brunch(diet_type: str) -> str:
+    """A tool that recommends a brunch based on the type of diet.
+    
+    Args:
+        diet_type: A description of the type of diet (e.g., vegan, keto, etc.).
+        
+    Returns:
+        A string with a brunch recommendation.
+    """
+    # Normalize input to lowercase for matching.
+    diet_type_lower = diet_type.lower()
+    
+    # Map diet types to brunch suggestions.
+    diet_to_brunch = {
+        "vegan": "Tofu scramble with avocado toast, fruit salad, and a green smoothie.",
+        "keto": "Avocado and egg breakfast bowl with bacon and spinach.",
+        "vegetarian": "Vegetable frittata with a side of roasted potatoes and fresh fruit.",
+        "paleo": "Grilled chicken with roasted sweet potatoes, sautéed spinach, and avocado.",
+        "gluten-free": "Gluten-free pancakes with berries and almond butter.",
+        "low-carb": "Egg muffins with spinach, cheese, and bacon, served with a side of avocado.",
+        "intermittent fasting": "A light salad with lean protein like grilled chicken, nuts, and avocado.",
+        "mediterranean": "Greek yogurt with honey, walnuts, and a side of olives and cucumber."
+    }
+    
+    # Default to a vegetarian brunch if diet type is unknown.
+    recommended_brunch = diet_to_brunch.get(diet_type_lower, "Vegetable frittata with a side of roasted potatoes and fresh fruit.")
+    
+    return f"Based on your diet type ('{diet_type}'), I recommend this brunch: {recommended_brunch}"
+
 
 final_answer = FinalAnswerTool()
 generate_recipe = RecipeGeneratorTool()
@@ -60,7 +90,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_recipe], ## add your tools here (don't remove final answer)
+    tools=[final_answer, recommend_diet_brunch], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,