Update of code

app.py

@@ -25,27 +25,31 @@ def get_current_weather(arg1:str)-> str: #it's important to specify the return t
 
 
 @tool
-def get_weather_prediction(arg1:str, arg2:list) -> str:
+def get_weather_prediction(arg1:str, arg2:str) -> str:
     """A tool that will give you weather prediction
     Args:
         arg1: The location
-        arg2: weather prediction it must be part of this list ['clear','sunny','partly cloudy','cloudy','overcast','mist','patchy rain possible','light rain','moderate rain','heavy rain','thunderstorm','snow showers','blizzard','fog'] and comma separated.
+        arg2: weather prediction(s), it must be a comma separated list where all members must be part of this list ['clear','sunny','partly cloudy','cloudy','overcast','mist','patchy rain possible','light rain','moderate rain','heavy rain','thunderstorm','snow showers','blizzard','fog'].
     """
+
+    if isinstance(arg2, str):
+        arg2 = arg2.split(',')
+
     url = f"https://wttr.in/{arg1}?format=j1"
     response = requests.get(url)
     expected_day = {}
     if response.status_code == 200:
         data = response.json()
-        for day in data["weather"]:
-            for hourly in day["hourly"]:
-                for expected_weather in arg2.split(','):
-                    if expected_weather.lower() in hourly["weatherDesc"][0]["value"].lower():
-                        expected_day[expected_weather] = day
+        for expected_weather in arg2:
+            for day in data["weather"]:
+                for hourly in day["hourly"]:
+                    if expected_weather.lower() in hourly["weatherDesc"][0]["value"].lower().strip(' ') and expected_weather.lower() not in expected_day:
+                        expected_day[expected_weather] = {'day' : day['date'], 'hour' : hourly['time']}
                     
     if len(expected_day):
         return str(expected_day)
     else:
-        return "Could not retrieve weather data"
+        return "It seems that the weather you are looking for is not in the forecast."
     
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -57,8 +61,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
         # Create timezone object
         tz = pytz.timezone(timezone)
         # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
+        local_day = datetime.datetime.now(tz).strftime("%Y-%m-%d")
+        local_time = datetime.datetime.now(tz).strftime("%H:%M:%S")
+        return f"The current local time in {timezone} is: day:{local_day} and time:{local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
@@ -96,4 +101,4 @@ agent = CodeAgent(
 )
 
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()