Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.unit_conversion import UnitConversionTool
 
 from Gradio_UI import GradioUI
 
@@ -32,40 +33,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-@tool
-def unit_conversion(value: float, from_unit: str, to_unit: str) -> str:
-    """A tool that converts units like distance, weight, and temperature.
-    
-    Args:
-        value: The numerical value to convert.
-        from_unit: The unit to convert from (e.g., 'miles', 'celsius').
-        to_unit: The unit to convert to (e.g., 'kilometers', 'fahrenheit').
-    
-    Returns:
-        The converted value in the target unit.
-    """
-    
-    conversion_factors = {
-        ("miles", "kilometers"): 1.60934,
-        ("kilometers", "miles"): 0.621371,
-        ("pounds", "kilograms"): 0.453592,
-        ("kilograms", "pounds"): 2.20462,
-        ("celsius", "fahrenheit"): lambda c: (c * 9/5) + 32,
-        ("fahrenheit", "celsius"): lambda f: (f - 32) * 5/9
-    }
-
-    key = (from_unit.lower(), to_unit.lower())
-
-    if key in conversion_factors:
-        conversion = conversion_factors[key]
-        result = conversion(value) if callable(conversion) else value * conversion
-        return f"{value} {from_unit} is {result:.2f} {to_unit}."
-    else:
-        return "Sorry, conversion not supported."
-
-        
+   
 final_answer = FinalAnswerTool()
+unit_conversion = UnitConversionTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -86,7 +56,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, unit_conversion], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,