Update app.py

app.py

@@ -7,16 +7,69 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+from typing import Union
+
 @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 unit_converter_tool(value:int, from_unit:str, to_unit:str)-> Union[int, float, str]: 
+    """Converts a value from one unit to another.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        value (float or int): The numerical value to convert.
+        from_unit (str): The unit of the input value (e.g., "cm", "inch", "kg", "celsius").
+        to_unit (str): The unit to convert the value to (e.g., "meter", "feet", "lbs", "fahrenheit").
+
+    Returns:
+        float or int: The converted value.
+        str: "Conversion not supported." if the requested conversion is not available.
+        float or int: Returns the original value if from_unit and to_unit are the same.
+
+    Raises:
+        TypeError: If value is not a number (int or float).
+
+    Supported Conversions:
+        Length: cm, inch, meter, feet
+        Weight: kg, lbs, ton, quintal
+        Temperature: celsius, fahrenheit
+        
     """
-    return "What magic will you build ?"
+    conversions = {
+        "cm_to_inch": lambda x: x * 0.393701,
+        "inch_to_cm": lambda x: x / 0.393701,
+        "cm_to_meter": lambda x: x / 100,
+        "meter_to_cm": lambda x: x * 100,
+        "cm_to_feet": lambda x: x * 0.0328084,
+        "feet_to_cm": lambda x: x / 0.0328084,
+        "inch_to_meter": lambda x: x * 0.0254,
+        "meter_to_inch": lambda x: x / 0.0254,
+        "inch_to_feet": lambda x: x * 0.0833333,
+        "feet_to_inch": lambda x: x / 0.0833333,
+        "meter_to_feet": lambda x: x * 3.28084,
+        "feet_to_meter": lambda x: x / 3.28084,
+        "kg_to_lbs": lambda x: x * 2.20462,
+        "lbs_to_kg": lambda x: x / 2.20462,
+        "kg_to_ton": lambda x: x * 0.001,
+        "ton_to_kg": lambda x: x / 0.001,
+        "kg_to_quintal": lambda x: x * 0.01,
+        "quintal_to_kg": lambda x: x / 0.01,
+        "ton_to_quintal": lambda x: x * 10,
+        "quintal_to_ton": lambda x: x / 10,
+        "celsius_to_fahrenheit": lambda x: (x * 9/5) + 32,
+        "fahrenheit_to_celsius": lambda x: (x - 32) * 5/9,
+    }
+    
+    if not isinstance(value, (int, float)):
+        raise TypeError("Value must be a number (int or float).")
+    
+    conversion_name = f"{from_unit}_to_{to_unit}"
+
+    # Handle cases where units might be the same:
+    if from_unit == to_unit:
+        return value
+
+    if conversion_name in conversions:
+        return conversions[conversion_name](value)
+    else:
+        return "Conversion not supported."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +108,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, unit_converter_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,