Update app.py

app.py

@@ -6,16 +6,17 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Tool personalizzato per ottenere l'ora attuale a Milano
+# Tool personalizzato per ottenere l'ora attuale in qualsiasi città
 @tool
-def get_current_time_milan() -> str:
-    """Restituisce l'ora attuale a Milano."""
+def get_current_time(city: str) -> str:
+    """Restituisce l'ora attuale per una città specificata dall'utente."""
     try:
-        tz = pytz.timezone('Europe/Rome')  # Milano è nella timezone di Roma
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"L'ora attuale a Milano è: {local_time}"
+        # Ricerca della timezone basata sulla città
+        timezone = pytz.timezone(pytz.country_timezones.get(city.lower(), 'UTC'))
+        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:
-        return f"Errore nel recupero dell'ora per Milano: {str(e)}"
+        return f"Errore nel recupero dell'ora per {city}: {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -34,7 +35,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_milan],  # Aggiunto il nuovo tool
+    tools=[final_answer, get_current_time],  # Aggiunto il nuovo tool per qualsiasi città
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -44,4 +45,18 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+# Funzione per interpretare la richiesta e restituire l'ora della città specificata
+@tool
+def interpret_and_respond(query: str) -> str:
+    """Interpreta la richiesta dell'utente e restituisce l'ora per una città specificata."""
+    words = query.lower().split()
+    if "time" in words or "ora" in words:
+        for word in words:
+            if word in pytz.all_timezones:
+                return get_current_time(word)
+        return "Per favore, specifica una città."
+    return "Mi dispiace, non ho capito la richiesta. Puoi riformularla?"
+
+agent.tools.append(interpret_and_respond)
+
 GradioUI(agent).launch()