Update app.py

app.py

@@ -15,10 +15,11 @@ def get_current_time(city: str) -> str:
         city (str): Il nome della città per cui ottenere l'ora.
     
     Returns:
-        str: L'ora attuale nella città richiesta oppure un messaggio di errore.
+        str: Una stringa contenente l'ora attuale nella città richiesta oppure un messaggio di errore.
     """
     try:
-        timezone = pytz.timezone(pytz.country_timezones.get(city.lower(), 'Europe/Rome'))
+        # Mapping della città alla sua timezone
+        timezone = pytz.timezone(city)
         local_time = datetime.datetime.now(timezone).strftime("%Y-%m-%d %H:%M:%S")
         return f"L'ora attuale a {city.capitalize()} è: {local_time}"
     except Exception as e:
@@ -57,19 +58,21 @@ def interpret_and_respond(query: str) -> str:
     """Interpreta la richiesta dell'utente e restituisce l'ora per una città specificata.
     
     Args:
-        query (str): La domanda dell'utente.
+        query (str): La domanda dell'utente contenente la città richiesta.
     
     Returns:
-        str: L'ora attuale nella città richiesta oppure un messaggio di errore.
+        str: Una stringa contenente l'ora attuale nella città richiesta oppure un messaggio di errore.
     """
     words = query.lower().split()
     if "time" in words or "ora" in words:
         for word in words:
-            if word in pytz.all_timezones:
+            try:
                 return get_current_time(word)
-        return "Per favore, specifica una città."
+            except:
+                continue
+        return "Per favore, specifica una città valida."
     return "Mi dispiace, non ho capito la richiesta. Puoi riformularla?"
 
 agent.tools.append(interpret_and_respond)
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()