Update app.py

app.py

@@ -61,7 +61,7 @@ def get_air_quality(city: str, lang: str = "en") -> str:
     
     Args:
         city: A string representing the city name (e.g., 'Бирск', 'Sydney').
-        lang: A string representing the language code (e.g., 'en', 'ru').
+        lang: A string representing the language code (e.g., 'en', 'ru', 'hi').
     
     Returns:
         A string with air quality index (AQI) and pollutant levels or an error message.
@@ -101,6 +101,13 @@ def get_air_quality(city: str, lang: str = "en") -> str:
                 "pm2_5": "Мелкие частицы (PM2.5)",
                 "pm10": "Крупные частицы (PM10)",
                 "co": "Угарный газ (CO)"
+            },
+            "hi": {
+                "aqi_desc": {1: "अच्छा", 2: "संतोषजनक", 3: "मध्यम", 4: "खराब", 5: "बहुत खराब"},
+                "air_quality": "वायु गुणवत्ता",
+                "pm2_5": "बारीक कण (PM2.5)",
+                "pm10": "मोटे कण (PM10)",
+                "co": "कार्बन मोनोऑक्साइड (CO)"
             }
         }
         lang = lang if lang in translations else "en"
@@ -122,7 +129,7 @@ def get_weather(city: str, lang: str = "en") -> str:
     
     Args:
         city: A string representing the city name (e.g., 'Бирск', 'Sydney').
-        lang: A string representing the language code (e.g., 'en', 'ru').
+        lang: A string representing the language code (e.g., 'en', 'ru', 'hi').
     
     Returns:
         A string with weather conditions, temperature, and humidity or an error message.
@@ -161,7 +168,8 @@ def get_weather(city: str, lang: str = "en") -> str:
                     "Snow": "Snow ❄️",
                     "Thunderstorm": "Thunderstorm ⛈",
                     "Drizzle": "Drizzle 🌦",
-                    "Mist": "Mist 🌫"
+                    "Mist": "Mist 🌫",
+                    "Haze": "Haze 🌫"
                 }
             },
             "ru": {
@@ -175,7 +183,23 @@ def get_weather(city: str, lang: str = "en") -> str:
                     "Snow": "Снег ❄️",
                     "Thunderstorm": "Гроза ⛈",
                     "Drizzle": "Морось 🌦",
-                    "Mist": "Туман 🌫"
+                    "Mist": "Туман 🌫",
+                    "Haze": "Туман 🌫"
+                }
+            },
+            "hi": {
+                "weather": "मौसम",
+                "temp": "तापमान",
+                "humidity": "नमी",
+                "conditions": {
+                    "Clear": "साफ ☀️",
+                    "Clouds": "बादल ☁️",
+                    "Rain": "बारिश 🌧",
+                    "Snow": "बर्फ ❄️",
+                    "Thunderstorm": "तूफान ⛈",
+                    "Drizzle": "बूंदाबांदी 🌦",
+                    "Mist": "कोहरा 🌫",
+                    "Haze": "धुंध 🌫"
                 }
             }
         }
@@ -218,15 +242,20 @@ prompt_templates = {
         "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 ONLY the provided functions: `get_timezone_by_city`, `get_current_time_in_timezone`, `get_air_quality`, `get_weather`, `final_answer`. "
-        "Detect query language with `detect` and pass it to `get_air_quality` and `get_weather` for translated output. "
+        "Do NOT use `detect` or any other functions directly in the code. "
         "Return ONLY the result via `final_answer()` in a code block:\n"
         "```py\n"
         "final_answer('YOUR ANSWER HERE')\n"
         "```\n"
-        "Do NOT include 'thoughts', 'code', explanations, extra imports, or invalid dates. "
-        "Do NOT define new functions or modify existing ones. "
+        "Do NOT include 'thoughts', explanations, extra imports, new functions, or text outside the ```py``` block. "
         "If the city is not found or query is unclear, return an error via `final_answer()`. "
-        "NEVER generate code without ```py``` block or with invalid syntax."
+        "Ensure the output matches the format:\n"
+        "🕒 Время в [City]: [Time]\n"
+        "😷 [Air Quality Label]: AQI [Number] ([Description]) [Emoji]\n"
+        "[Pollutants]\n"
+        "🌡️ [Weather Label]: [Condition]\n"
+        "[Temperature Label]: [Temp]°C\n"
+        "[Humidity Label]: [Humidity]%"
     ),
     "default": "Response: {{question}}",
     "planning": {
@@ -256,17 +285,21 @@ prompt_templates = {
             "Return code:\n"
             "```py\n"
             "from langdetect import detect\n"
-            "query = '{{question}}'.strip()\n"
-            "city = query.split(' в ')[-1].strip().title() if ' в ' in query else query.title()\n"
-            "lang = detect(query) if query else 'en'\n"
-            "timezone = get_timezone_by_city(city)\n"
-            "if timezone.startswith('Error'):\n"
-            "    final_answer(timezone)\n"
+            "query = '{{question}}'.strip().lower()\n"
+            "city = query.split(' в ')[-1].strip().title() if ' в ' in query else query.strip().title()\n"
+            "if city in ['привет', 'что', 'там', 'сколько', 'времени', '']:\n"
+            "    final_answer('Error: Please specify a valid city name')\n"
             "else:\n"
-            "    time_result = get_current_time_in_timezone(timezone)\n"
-            "    air_result = get_air_quality(city, lang)\n"
-            "    weather_result = get_weather(city, lang)\n"
-            "    final_answer(f'🕒 Время в {city}: {time_result}\\n' + air_result + '\\n' + weather_result)\n"
+            "    lang = detect(query) if query else 'en'\n"
+            "    city_normalized = city.replace('Нью-Дели', 'New Delhi').replace('Нью-Йорк', 'New York')\n"
+            "    timezone = get_timezone_by_city(city_normalized)\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_normalized, lang)\n"
+            "        weather_result = get_weather(city_normalized, lang)\n"
+            "        final_answer(f'🕒 Время в {city_normalized}: {time_result}\\n' + air_result + '\\n' + weather_result)\n"
             "```"
         ),
         "report": (
@@ -302,7 +335,7 @@ prompt_templates = {
 agent = CodeAgent(
     model=model,
     tools=[final_answer, get_current_time_in_timezone, get_timezone_by_city, get_air_quality, get_weather],
-    max_steps=3,
+    max_steps=1,
     verbosity_level=2,
     prompt_templates=prompt_templates,
 )
@@ -314,10 +347,11 @@ def process_input(user_input):
     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, weather, and more!"
-).launch()
+if __name__ == "__main__":
+    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, weather, and more!"
+    ).launch()