Update app.py

app.py

@@ -6,22 +6,17 @@ import gradio as gr
 from geopy.geocoders import Nominatim
 from timezonefinder import TimezoneFinder
 import requests
+from langdetect import detect
 
 @tool
 def get_timezone_by_city(city: str) -> str:
-    """Tool to fetch the timezone for a given city using geolocation.
-    Args:
-        city: A string representing the city name, optionally with region or country (e.g., 'Бураево', 'Осиновка, Бирский район', 'Идиофа, Африка').
-    Returns:
-        A string with the timezone (e.g., 'Europe/Samara') or an error message.
-    """
     if not isinstance(city, str):
         return f"Error: Expected a string for city, got: {type(city)}"
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
         location = geolocator.geocode(city, language="en")
         if not location:
-            city_name = city.split(",")[0].strip()
+            city_name = city.split(",")[0].strip().title()
             location = geolocator.geocode(city_name, language="en")
             if not location:
                 return f"Error: City '{city}' not found. Try a different spelling or provide more context."
@@ -35,34 +30,24 @@ def get_timezone_by_city(city: str) -> str:
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Tool to fetch the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'Europe/Moscow').
-    """
     if not isinstance(timezone, str):
         return f"Error: Expected a string for timezone, got: {type(timezone)}"
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"Current time in {timezone}: {local_time}"
+        return local_time
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def get_air_quality(city: str) -> str:
-    """Tool to fetch air quality data for a given city using OpenWeatherMap API.
-    Args:
-        city: A string representing the city name (e.g., 'Бирск', 'New York').
-    Returns:
-        A string with air quality index (AQI) and pollutant levels or an error message.
-    """
+def get_air_quality(city: str, lang: str = "en") -> str:
     if not isinstance(city, str):
         return f"Error: Expected a string for city, got: {type(city)}"
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
         location = geolocator.geocode(city, language="en")
         if not location:
-            city_name = city.split(",")[0].strip()
+            city_name = city.split(",")[0].strip().title()
             location = geolocator.geocode(city_name, language="en")
             if not location:
                 return f"Error: City '{city}' not found for air quality data."
@@ -76,22 +61,110 @@ def get_air_quality(city: str) -> str:
         data = response.json()
         aqi = data["list"][0]["main"]["aqi"]
         components = data["list"][0]["components"]
+
+        translations = {
+            "en": {
+                "aqi_desc": {1: "Good", 2: "Fair", 3: "Moderate", 4: "Poor", 5: "Very Poor"},
+                "air_quality": "Air quality",
+                "pm2_5": "Fine particles (PM2.5)",
+                "pm10": "Coarse particles (PM10)",
+                "co": "Carbon monoxide (CO)",
+                "no2": "Nitrogen dioxide (NO2)",
+                "so2": "Sulfur dioxide (SO2)",
+                "o3": "Ozone (O3)"
+            },
+            "ru": {
+                "aqi_desc": {1: "Хорошее", 2: "Удовлетворительное", 3: "Среднее", 4: "Плохое", 5: "Очень плохое"},
+                "air_quality": "Качество воздуха",
+                "pm2_5": "Мелкие частицы (PM2.5)",
+                "pm10": "Крупные частицы (PM10)",
+                "co": "Угарный газ (CO)",
+                "no2": "Диоксид азота (NO2)",
+                "so2": "Диоксид серы (SO2)",
+                "o3": "Озон (O3)"
+            }
+        }
+        lang = lang if lang in translations else "en"
+        t = translations[lang]
+        aqi_emoji = "🌱" if aqi <= 2 else "😷" if aqi <= 4 else "☣️"
         
-        aqi_description = {1: "Good", 2: "Fair", 3: "Moderate", 4: "Poor", 5: "Very Poor"}
         return (
-            f"Air quality in {city}: AQI {aqi} ({aqi_description.get(aqi, 'Unknown')})\n"
-            f"PM2.5: {components['pm2_5']} µg/m³, PM10: {components['pm10']} µg/m³\n"
-            f"CO: {components['co']} µg/m³, NO2: {components['no2']} µg/m³, SO2: {components['so2']} µg/m³, O3: {components['o3']} µg/m³"
+            f"😷 {t['air_quality']}: AQI {aqi} ({t['aqi_desc'].get(aqi, 'Unknown')}) {aqi_emoji}\n"
+            f"{t['pm2_5']}: {components['pm2_5']} µg/m³\n"
+            f"{t['pm10']}: {components['pm10']} µg/m³\n"
+            f"{t['co']}: {components['co']} µg/m³\n"
+            f"{t['no2']}: {components['no2']} µg/m³\n"
+            f"{t['so2']}: {components['so2']} µg/m³\n"
+            f"{t['o3']}: {components['o3']} µg/m³"
         )
     except Exception as e:
         return f"Error fetching air quality for '{city}': {str(e)}"
 
+@tool
+def get_weather(city: str, lang: str = "en") -> str:
+    if not isinstance(city, str):
+        return f"Error: Expected a string for city, got: {type(city)}"
+    try:
+        geolocator = Nominatim(user_agent="smolagents_bot")
+        location = geolocator.geocode(city, language="en")
+        if not location:
+            city_name = city.split(",")[0].strip().title()
+            location = geolocator.geocode(city_name, language="en")
+            if not location:
+                return f"Error: City '{city}' not found for weather data."
+        
+        api_key = os.getenv("OPENWEATHERMAP_API_KEY", "93f518fda8d24cf89aee7050c26b27e9")
+        url = f"http://api.openweathermap.org/data/2.5/weather?lat={location.latitude}&lon={location.longitude}&appid={api_key}&units=metric"
+        response = requests.get(url)
+        if response.status_code != 200:
+            return f"Error fetching weather for '{city}': API returned status {response.status_code}"
+        
+        data = response.json()
+        temp = data["main"]["temp"]
+        description = data["weather"][0]["description"]
+        weather_id = data["weather"][0]["id"]
+
+        translations = {
+            "en": {
+                "weather": "Weather in",
+                "clear": "Clear",
+                "clouds": "Cloudy",
+                "rain": "Rain",
+                "snow": "Snow",
+                "thunderstorm": "Thunderstorm",
+                "mist": "Mist"
+            },
+            "ru": {
+                "weather": "Погода в",
+                "clear": "Ясно",
+                "clouds": "Облачно",
+                "rain": "Дождь",
+                "snow": "Снег",
+                "thunderstorm": "Гроза",
+                "mist": "Туман"
+            }
+        }
+        lang = lang if lang in translations else "en"
+        t = translations[lang]
+        
+        weather_emoji = {
+            "2": "⛈️",  # Thunderstorm
+            "3": "🌧️",  # Drizzle
+            "5": "🌧️",  # Rain
+            "6": "❄️",  # Snow
+            "7": "🌫️",  # Mist
+            "800": "☀️", # Clear
+            "8": "☁️"   # Clouds
+        }
+        emoji = weather_emoji.get(str(weather_id)[0], "🌤️") if weather_id != 800 else weather_emoji.get("800")
+        
+        desc_trans = t.get(description, description.capitalize())
+        return f"🌤️ {t['weather']} {city}: {temp}°C, {desc_trans} {emoji}"
+    except Exception as e:
+        return f"Error fetching weather for '{city}': {str(e)}"
+
 @tool
 def final_answer(answer: str) -> str:
-    """Returns the final answer.
-    Args:
-        answer: The final answer as a string.
-    """
     return answer
 
 try:
@@ -106,115 +179,13 @@ except Exception as e:
 prompt_templates = {
     "system_prompt": (
         "You are a helpful assistant that responds in the same language as the user's query. "
-        "Detect the language and return only the result, without explanations, images, or repeating instructions. "
-        "If the query contains a city name (e.g., 'Бураево', 'Осиновка', 'Fosnavåg', 'न्यू यॉर्क') or phrases like 'сколько времени', 'what time', 'quelle heure', "
-        "or if a city is mentioned in any context (e.g., 'Что там в Бирске?'), treat it as a request for the current time and air quality in that city. "
-        "Use `get_timezone_by_city` to find the timezone, `get_current_time_in_timezone` for the time, and `get_air_quality` for air quality data. "
-        "Combine results in a single response. "
-        "Always return the response via `final_answer()` in a code block like this:\n"
+        "If the query contains a city name (e.g., 'Бураево', 'Sydney', 'न्यू यॉर्क') or phrases like 'сколько времени', 'what time', "
+        "or a city in any context (e.g., 'Что там в Бирске?'), return current time, air quality, and weather for that city. "
+        "Use `get_timezone_by_city`, `get_current_time_in_timezone`, `get_air_quality`, and `get_weather`. "
+        "Detect query language and pass it to `get_air_quality` and `get_weather` for translated output. "
+        "Return only the result via `final_answer()` in a code block:\n"
         "```py\n"
         "final_answer('YOUR ANSWER HERE')\n"
         "```\n"
-        "Execute tools in sequence: first call get_timezone_by_city(city), then get_current_time_in_timezone(timezone), then get_air_quality(city). "
-        "Do NOT include tool calls, formatting like 'Current time in X: ', or invalid dates (e.g., '2023-03-22') inside final_answer(). "
-        "If the city is not found or the query is unclear (e.g., no city mentioned), return an error message in the same language via final_answer(). "
-        "NEVER generate code without the ```py``` block or with invalid syntax. "
-        "Do NOT ask for clarifications or generate responses outside the ```py``` block."
-    ),
-    "default": "Response: {{question}}",
-    "planning": {
-        "initial_plan": (
-            "Analyze the query: {{question}}. If it contains a city name or phrases like 'сколько времени', "
-            "use `get_timezone_by_city`, `get_current_time_in_timezone`, and `get_air_quality`. "
-            "Write code in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR ANSWER HERE')\n"
-            "```"
-        ),
-        "update_plan_pre_messages": (
-            "Review the query: {{question}}. Adjust the plan if needed. "
-            "Write code in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR ANSWER HERE')\n"
-            "```"
-        ),
-        "update_plan_post_messages": (
-            "Review the query: {{question}} and results. Adjust the plan if needed. "
-            "Write code in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR ANSWER HERE')\n"
-            "```"
-        ),
-    },
-    "managed_agent": {
-        "execute": (
-            "Execute the task: {{question}}. If it contains a city name or phrases like 'сколько времени', use `get_timezone_by_city`, `get_current_time_in_timezone`, and `get_air_quality`. "
-            "Write code in a code block:\n"
-            "```py\n"
-            "query = '{{question}}'.strip()\n"
-            "city = query.split(' в ')[-1].strip() if ' в ' in query else query\n"
-            "timezone = get_timezone_by_city(city)\n"
-            "if timezone.startswith('Error'):\n"
-            "    final_answer(timezone)\n"
-            "else:\n"
-            "    time_result = get_current_time_in_timezone(timezone)\n"
-            "    air_result = get_air_quality(city)\n"
-            "    final_answer(time_result + '\\n' + air_result)\n"
-            "```"
-        ),
-        "report": (
-            "Summarize the execution of the task: {{question}}. "
-            "Return the result via final_answer() in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR ANSWER HERE')\n"
-            "```"
-        ),
-        "task": (
-            "Define the task based on the query: {{question}}. "
-            "Return the task definition via final_answer() in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR TASK DEFINITION HERE')\n"
-            "```"
-        ),
-    },
-    "final_answer": {
-        "pre_messages": (
-            "Prepare the final response for the query: {{question}}. "
-            "Return the response via final_answer() in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR ANSWER HERE')\n"
-            "```"
-        ),
-        "template": "Final response: {{answer}}",
-        "post_messages": (
-            "Review the formatted response for the query: {{question}}. "
-            "Return the final response via final_answer() in a code block:\n"
-            "```py\n"
-            "final_answer('YOUR ANSWER HERE')\n"
-            "```"
-        ),
-    },
-}
-
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_timezone_by_city, get_air_quality],
-    max_steps=3,
-    verbosity_level=2,
-    prompt_templates=prompt_templates,
-)
-
-def process_input(user_input):
-    try:
-        response = agent.run(user_input)
-        return response if response else "Error: No response received from the agent."
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-gr.Interface(
-    fn=process_input,
-    inputs="text",
-    outputs="text",
-    title="Helpful Assistant",
-    description="Hello! What city or town are you from? I can tell you the time, air quality, and more!"
-).launch()
+        "Do NOT include 'thoughts', 'code', explanations, or invalid dates. "
+        "If the city is not found or