Update app.py

app.py

@@ -18,6 +18,58 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def weather_forecast(location: str, date: str) -> str:
+    """A tool to get the weather forecast for a specific location and date.
+    Args:
+        location: The location for which to get the weather forecast.
+        date: The date for the weather forecast in YYYY-MM-DD format.
+    """
+    # Example implementation using a weather API
+    response = requests.get(f"https://api.weatherapi.com/v1/forecast.json?key=YOUR_API_KEY&q={location}&dt={date}")
+    data = response.json()
+    forecast = data['forecast']['forecastday'][0]['day']
+    return f"Weather forecast for {location} on {date}: {forecast['condition']['text']}, Temperature: {forecast['avgtemp_c']}°C"
+
+@tool
+def time_zone_converter(time: str, from_tz: str, to_tz: str) -> str:
+    """A tool to convert time from one time zone to another.
+    Args:
+        time: The time to convert in HH:MM format.
+        from_tz: The time zone to convert from.
+        to_tz: The time zone to convert to.
+    """
+    from_zone = pytz.timezone(from_tz)
+    to_zone = pytz.timezone(to_tz)
+    from_time = datetime.datetime.strptime(time, '%H:%M')
+    from_time = from_zone.localize(from_time)
+    to_time = from_time.astimezone(to_zone)
+    return to_time.strftime('%H:%M %Z')
+
+@tool
+def currency_converter(amount: float, from_currency: str, to_currency: str) -> str:
+    """A tool to convert currency from one type to another.
+    Args:
+        amount: The amount of money to convert.
+        from_currency: The currency code to convert from.
+        to_currency: The currency code to convert to.
+    """
+    response = requests.get(f"https://api.exchangerate-api.com/v4/latest/{from_currency}")
+    data = response.json()
+    rate = data['rates'][to_currency]
+    converted_amount = amount * rate
+    return f"{amount} {from_currency} is equal to {converted_amount:.2f} {to_currency}"
+
+@tool
+def task_reminder(task: str, due_date: str) -> str:
+    """A tool to set a reminder for a task.
+    Args:
+        task: The task to be reminded of.
+        due_date: The due date for the task in YYYY-MM-DD format.
+    """
+    return f"Reminder set for task: '{task}' due on {due_date}"
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -42,7 +94,7 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' #'Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
@@ -55,7 +107,13 @@ 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,
+          get_current_time_in_timezone,
+          task_reminder,
+          currency_converter,
+          time_zone_converter,
+          weather_forecast 
+          ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,