Add nutrient_calculator tool

app.py

@@ -9,14 +9,39 @@ 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 nutrient_calculator(weight: float, activity_level: str) -> str:
+    """Calculates basic daily nutrient needs based on weight and activity level.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        weight: Weight in kilograms
+        activity_level: low/medium/high
     """
-    return "What magic will you build ?"
+    activity_multipliers = {
+        'low': 1.2,
+        'medium': 1.5,
+        'high': 1.8
+    }
+    
+    if activity_level.lower() not in activity_multipliers:
+        return "Invalid activity level. Use 'low', 'medium', or 'high'."
+    
+    if weight <= 0 or weight > 300:
+        return "Please enter a valid weight between 0 and 300 kg."
+    
+    multiplier = activity_multipliers[activity_level.lower()]
+    
+    # Basic calculations
+    calories = round(weight * 24 * multiplier)
+    protein = round(weight * 1.8)  # 1.8g per kg
+    carbs = round(calories * 0.45 / 4)  # 45% of calories from carbs
+    fats = round(calories * 0.25 / 9)  # 25% of calories from fats
+    water = round(weight * 0.033, 1)  # 33ml per kg
+    
+    return f"""Daily nutrient needs:
+Calories: {calories} kcal
+Protein: {protein}g
+Carbohydrates: {carbs}g
+Fats: {fats}g
+Water: {water} liters"""
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +80,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, nutrient_calculator], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,