Update app.py

app.py

@@ -42,7 +42,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         timezone (str): Timezone name.
 
     Returns:
-        str: Current time.
+        str: Current time or error.
     """
     try:
         tz = pytz.timezone(timezone)
@@ -59,7 +59,7 @@ def get_air_quality(city: str, lang: str = "en") -> str:
         lang (str): Language code.
 
     Returns:
-        str: AQI string.
+        str: Translated air quality info.
     """
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
@@ -67,7 +67,10 @@ def get_air_quality(city: str, lang: str = "en") -> str:
         if not location:
             return f"❌ Город '{city}' не найден."
         api_key = os.environ["OPENWEATHERMAP_API_KEY"]
-        url = f"http://api.openweathermap.org/data/2.5/air_pollution?lat={location.latitude}&lon={location.longitude}&appid={api_key}"
+        url = (
+            f"http://api.openweathermap.org/data/2.5/air_pollution"
+            f"?lat={location.latitude}&lon={location.longitude}&appid={api_key}"
+        )
         data = requests.get(url).json()
         aqi = data["list"][0]["main"]["aqi"]
         c = data["list"][0]["components"]
@@ -84,7 +87,7 @@ def get_air_quality(city: str, lang: str = "en") -> str:
                 "title": "😷 Качество воздуха",
                 "aqi_desc": {1: "Хорошее", 2: "Удовлетворительное", 3: "Среднее", 4: "Плохое", 5: "Очень плохое"},
                 "pm2_5": "Мелкодисперсные частицы (PM2.5)",
-                "pm10": "Крупные частицы (PM10)",
+                "pm10": "Крупнодисперсные частицы (PM10)",
                 "co": "Угарный газ (CO)",
             },
             "hi": {
@@ -116,7 +119,7 @@ def get_weather(city: str, lang: str = "en") -> str:
         lang (str): Language code.
 
     Returns:
-        str: Weather string.
+        str: Weather info string.
     """
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
@@ -124,7 +127,11 @@ def get_weather(city: str, lang: str = "en") -> str:
         if not location:
             return f"❌ Город '{city}' не найден."
         api_key = os.environ["OPENWEATHERMAP_API_KEY"]
-        url = f"http://api.openweathermap.org/data/2.5/weather?lat={location.latitude}&lon={location.longitude}&appid={api_key}&units=metric"
+        url = (
+            f"http://api.openweathermap.org/data/2.5/weather"
+            f"?lat={location.latitude}&lon={location.longitude}"
+            f"&appid={api_key}&units=metric"
+        )
         data = requests.get(url).json()
         w = data["weather"][0]["main"]
         temp = data["main"]["temp"]
@@ -137,7 +144,8 @@ def get_weather(city: str, lang: str = "en") -> str:
                 "humidity": "Humidity",
                 "conditions": {
                     "Clear": "Clear ☀️", "Clouds": "Cloudy ☁️", "Rain": "Rain 🌧",
-                    "Snow": "Snow ❄️", "Thunderstorm": "Storm ⛈", "Drizzle": "Drizzle 🌦", "Mist": "Mist 🌫", "Haze": "Haze 🌫"
+                    "Snow": "Snow ❄️", "Thunderstorm": "Storm ⛈", "Drizzle": "Drizzle 🌦",
+                    "Mist": "Mist 🌫", "Haze": "Haze 🌫"
                 }
             },
             "ru": {
@@ -146,7 +154,8 @@ def get_weather(city: str, lang: str = "en") -> str:
                 "humidity": "Влажность",
                 "conditions": {
                     "Clear": "Ясно ☀️", "Clouds": "Облачно ☁️", "Rain": "Дождь 🌧",
-                    "Snow": "Снег ❄️", "Thunderstorm": "Гроза ⛈", "Drizzle": "Морось 🌦", "Mist": "Туман 🌫", "Haze": "Дымка 🌫"
+                    "Snow": "Снег ❄️", "Thunderstorm": "Гроза ⛈", "Drizzle": "Морось 🌦",
+                    "Mist": "Туман 🌫", "Haze": "Дымка 🌫"
                 }
             },
             "hi": {
@@ -155,7 +164,8 @@ def get_weather(city: str, lang: str = "en") -> str:
                 "humidity": "नमी",
                 "conditions": {
                     "Clear": "साफ ☀️", "Clouds": "बादल ☁️", "Rain": "बारिश 🌧",
-                    "Snow": "बर्फ ❄️", "Thunderstorm": "तूफान ⛈", "Drizzle": "बूंदाबांदी 🌦", "Mist": "कोहरा 🌫", "Haze": "धुंध 🌫"
+                    "Snow": "बर्फ ❄️", "Thunderstorm": "तूфान ⛈", "Drizzle": "बूंदाबांदी 🌦",
+                    "Mist": "कोहरा 🌫", "Haze": "धुंध 🌫"
                 }
             }
         }
@@ -172,10 +182,18 @@ def get_weather(city: str, lang: str = "en") -> str:
 
 @tool
 def final_answer(answer: str) -> str:
-    """Final string output."""
+    """
+    Return the final answer string to the user.
+
+    Args:
+        answer (str): The final response to display.
+
+    Returns:
+        str: The same response passed in.
+    """
     return answer
 
-# 🤖 Модель
+# 🤖 Model setup
 model = HfApiModel(
     max_tokens=1000,
     temperature=0.5,
@@ -185,13 +203,14 @@ model = HfApiModel(
 # 📜 System Prompt
 system_prompt = (
     "Ты помощник. Используй только функции: "
-    "`get_timezone_by_city`, `get_current_time_in_timezone`, `get_air_quality`, `get_weather`, `final_answer`. "
-    "Передавай `lang = variables['lang']`, и `dark_mode = variables['dark_mode']`. "
-    "Ответ в виде:\n```py\nfinal_answer('...')\n```<end_code>\n"
+    "`get_timezone_by_city`, `get_current_time_in_timezone`, `get_air_quality`, "
+    "`get_weather`, `final_answer`. Передавай `lang = variables['lang']`, "
+    "и `dark_mode = variables['dark_mode']`. Ответ в формате:\n"
+    "```py\nfinal_answer('...')\n```<end_code>\n"
     "❌ Без пояснений, markdown, JSON."
 )
 
-# 🎯 Templates
+# 🎯 Prompt templates
 prompt_templates = {
     "system_prompt": system_prompt,
     "default": "Response: {{question}}",
@@ -220,7 +239,7 @@ agent = CodeAgent(
     verbosity_level=2
 )
 
-# 🧠 Тёмный юмор
+# 🦹‍♂️ Dark humor jokes
 def get_joke(lang):
     jokes = {
         "en": [
@@ -241,12 +260,12 @@ def get_joke(lang):
     }
     return random.choice(jokes.get(lang, jokes["en"]))
 
-# 💬 Обработка запроса
+# 💬 Request processing
 def process_input(user_input):
     try:
         lang = detect(user_input)
         lang = lang if lang in ["en", "ru", "hi"] else "en"
-        dark_mode = True  # ← можно выключить
+        dark_mode = True  # enable/disable humor
 
         agent.variables = {
             "lang": lang,
@@ -260,7 +279,7 @@ def process_input(user_input):
     except Exception as e:
         return f"❌ Ошибка: {str(e)}"
 
-# 🚀 Gradio UI
+# 🚀 Run
 if __name__ == "__main__":
     gr.Interface(
         fn=process_input,
@@ -268,4 +287,4 @@ if __name__ == "__main__":
         outputs="text",
         title="🦊 Умный ассистент",
         description="Напиши город — узнаешь время, погоду, воздух и получишь комментарий 😏"
-    ).launch() 
+    ).launch()