Added new tools for health

app.py

@@ -4,19 +4,74 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from typing import Tuple
 
 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 calculate_bmi(weight: float, height: float) -> str:
+    """A tool that calculates the Body Mass Index (BMI).
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        weight: The weight of the person in kilograms.
+        height: The height of the person in meters.
+    
+    Returns:
+        A string displaying the BMI and its classification.
     """
-    return "What magic will you build ?"
+    try:
+        bmi = weight / (height ** 2)
+        classification = (
+            "Underweight" if bmi < 18.5 else
+            "Normal weight" if bmi < 24.9 else
+            "Overweight" if bmi < 29.9 else
+            "Obese"
+        )
+        return f"BMI: {bmi:.2f} ({classification})"
+    except Exception as e:
+        return f"Error calculating BMI: {str(e)}"
+
+@tool
+def calculate_bmr(weight: float, height: float, age: int, sex: str) -> str:
+    """A tool that calculates the Basal Metabolic Rate (BMR) using the Mifflin-St Jeor Equation.
+    
+    Args:
+        weight: The weight of the person in kilograms.
+        height: The height of the person in centimeters.
+        age: The age of the person in years.
+        sex: The biological sex of the person ('male' or 'female').
+    
+    Returns:
+        A string displaying the calculated BMR.
+    """
+    try:
+        if sex.lower() == "male":
+            bmr = 88.36 + (13.4 * weight) + (4.8 * height) - (5.7 * age)
+        elif sex.lower() == "female":
+            bmr = 447.6 + (9.2 * weight) + (3.1 * height) - (4.3 * age)
+        else:
+            return "Error: Invalid sex input. Use 'male' or 'female'."
+        
+        return f"BMR: {bmr:.2f} kcal/day"
+    except Exception as e:
+        return f"Error calculating BMR: {str(e)}"
+
+@tool
+def calculate_water_requirement(weight: float) -> str:
+    """A tool that calculates the daily water intake requirement.
+    
+    Args:
+        weight: The weight of the person in kilograms.
+    
+    Returns:
+        A string displaying the recommended daily water intake in liters.
+    """
+    try:
+        water_intake = 0.033 * weight
+        return f"Recommended daily water intake: {water_intake:.2f} liters"
+    except Exception as e:
+        return f"Error calculating water requirement: {str(e)}"
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +110,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, get_current_time_in_timezone, DuckDuckGoSearchTool(), calculate_water_requirement, calculate_bmi, calculate_bmr], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,