Update app.py

app.py

@@ -7,18 +7,26 @@ from langdetect import detect
 from geopy.geocoders import Nominatim
 from timezonefinder import TimezoneFinder
 import gradio as gr
+import re
 
 load_dotenv()
 
-def geocode_city(city: str):
+def geocode_city(city: str, lang: str = "en"):
+    """
+    Geocode city with fallback to simpler name and language.
+    """
     geolocator = Nominatim(user_agent="weather_assistant")
-    location = geolocator.geocode(city, language="en")
+    location = geolocator.geocode(city, language=lang)
     if not location:
+        # fallback: first word
         simple = city.split()[0]
-        location = geolocator.geocode(simple, language="en")
+        location = geolocator.geocode(simple, language=lang)
     return location
 
 def get_timezone_by_city(city: str) -> str:
+    """
+    Get timezone for a city.
+    """
     location = geocode_city(city)
     if not location:
         return f"❌ Не удалось найти населенный пункт '{city}'."
@@ -28,6 +36,9 @@ def get_timezone_by_city(city: str) -> str:
     return tz
 
 def get_current_time_in_timezone(timezone: str) -> str:
+    """
+    Get local current time in a timezone.
+    """
     try:
         now = datetime.datetime.now(pytz.timezone(timezone))
         return now.strftime("%Y-%m-%d %H:%M:%S")
@@ -35,6 +46,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"❌ Ошибка получения времени: {e}"
 
 def get_air_quality(city: str, lang: str = "en") -> str:
+    """
+    Get translated air quality info.
+    """
     location = geocode_city(city)
     if not location:
         return f"❌ Не удалось найти населенный пункт '{city}'."
@@ -73,6 +87,34 @@ def get_air_quality(city: str, lang: str = "en") -> str:
             "pm10": "मोटे कण (PM10)",
             "co": "कार्बन मोनोऑक्साइड (CO)",
         },
+        "zh": {
+            "title": "😷 空气质量",
+            "aqi_desc": {1: "良好", 2: "一般", 3: "中等", 4: "较差", 5: "很差"},
+            "pm2_5": "细颗粒物 (PM2.5)",
+            "pm10": "可吸入颗粒物 (PM10)",
+            "co": "一氧化碳 (CO)",
+        },
+        "es": {
+            "title": "😷 Calidad del aire",
+            "aqi_desc": {1: "Buena", 2: "Aceptable", 3: "Moderada", 4: "Pobre", 5: "Muy mala"},
+            "pm2_5": "Partículas finas (PM2.5)",
+            "pm10": "Partículas gruesas (PM10)",
+            "co": "Monóxido de carbono (CO)",
+        },
+        "fr": {
+            "title": "😷 Qualité de l'air",
+            "aqi_desc": {1: "Bonne", 2: "Passable", 3: "Moyenne", 4: "Mauvaise", 5: "Très mauvaise"},
+            "pm2_5": "Particules fines (PM2.5)",
+            "pm10": "Particules grossières (PM10)",
+            "co": "Monoxyde de carbone (CO)",
+        },
+        "ko": {
+            "title": "😷 대기질",
+            "aqi_desc": {1: "좋음", 2: "보통", 3: "보통 이상", 4: "나쁨", 5: "매우 나쁨"},
+            "pm2_5": "미세먼지 (PM2.5)",
+            "pm10": "먼지 (PM10)",
+            "co": "일산화탄소 (CO)",
+        },
     }
     t = translations.get(lang, translations["en"])
     emoji = "🌱" if aqi <= 2 else "😷" if aqi <= 4 else "☣️"
@@ -84,6 +126,9 @@ def get_air_quality(city: str, lang: str = "en") -> str:
     )
 
 def get_weather(city: str, lang: str = "en") -> str:
+    """
+    Get translated weather info.
+    """
     location = geocode_city(city)
     if not location:
         return f"❌ Не удалось найти населенный пункт '{city}'."
@@ -131,6 +176,46 @@ def get_weather(city: str, lang: str = "en") -> str:
                 "Mist": "कोहरा 🌫", "Haze": "धुंध 🌫"
             }
         },
+        "zh": {
+            "title": "🌡️ 天气",
+            "temp": "温度",
+            "humidity": "湿度",
+            "conditions": {
+                "Clear": "晴 ☀️", "Clouds": "多云 ☁️", "Rain": "雨 🌧",
+                "Snow": "雪 ❄️", "Thunderstorm": "雷暴 ⛈", "Drizzle": "毛毛雨 🌦",
+                "Mist": "雾 🌫", "Haze": "霾 🌫"
+            }
+        },
+        "es": {
+            "title": "🌡️ Clima",
+            "temp": "Temperatura",
+            "humidity": "Humedad",
+            "conditions": {
+                "Clear": "Despejado ☀️", "Clouds": "Nublado ☁️", "Rain": "Lluvia 🌧",
+                "Snow": "Nieve ❄️", "Thunderstorm": "Tormenta ⛈", "Drizzle": "Llovizna 🌦",
+                "Mist": "Niebla 🌫", "Haze": "Calina 🌫"
+            }
+        },
+        "fr": {
+            "title": "🌡️ Météo",
+            "temp": "Température",
+            "humidity": "Humidité",
+            "conditions": {
+                "Clear": "Dégagé ☀️", "Clouds": "Nuageux ☁️", "Rain": "Pluie 🌧",
+                "Snow": "Neige ❄️", "Thunderstorm": "Orage ⛈", "Drizzle": "Bruine 🌦",
+                "Mist": "Brume 🌫", "Haze": "Brume légère 🌫"
+            }
+        },
+        "ko": {
+            "title": "🌡️ 날씨",
+            "temp": "온도",
+            "humidity": "습도",
+            "conditions": {
+                "Clear": "맑음 ☀️", "Clouds": "흐림 ☁️", "Rain": "비 🌧",
+                "Snow": "눈 ❄️", "Thunderstorm": "뇌우 ⛈", "Drizzle": "이슬비 🌦",
+                "Mist": "안개 🌫", "Haze": "실안개 🌫"
+            }
+        },
     }
     t = translations.get(lang, translations["en"])
     desc = t["conditions"].get(cond, cond)
@@ -141,18 +226,59 @@ def get_weather(city: str, lang: str = "en") -> str:
     )
 
 def process_input(user_input: str) -> str:
+
     lang = detect(user_input)
-    lang = lang if lang in ["en", "ru", "hi"] else "en"
+    if lang not in ["en","ru","hi","zh","es","fr","ko"]:
+        if re.search(r'[\u0400-\u04FF]', user_input):
+            lang = "ru"
+        else:
+            lang = "en"
     city = user_input.strip()
+
+    loc_disp = geocode_city(city, lang)
+    if not loc_disp:
+        return f"❌ Не удалось найти населенный пункт '{city}'."
+    parts = loc_disp.raw.get("display_name", "").split(", ")
+    city_disp = parts[0]
+    region_disp = parts[1] if len(parts) > 2 else ""
+    country_disp = parts[-1] if parts else ""
+
+    location_labels = {
+        "en": "📍 Location",
+        "ru": "📍 Местоположение",
+        "hi": "📍 स्थान",
+        "zh": "📍 位置",
+        "es": "📍 Ubicación",
+        "fr": "📍 Emplacement",
+        "ko": "📍 위치",
+    }
+    time_labels = {
+        "en": "🕒 Time",
+        "ru": "🕒 Время",
+        "hi": "🕒 समय",
+        "zh": "🕒 时间",
+        "es": "🕒 Hora",
+        "fr": "🕒 Heure",
+        "ko": "🕒 시간",
+    }
+
+    location_line = f"{location_labels.get(lang,'📍 Location')}: {city_disp}"
+    if region_disp:
+        location_line += f", {region_disp}"
+    if country_disp:
+        location_line += f", {country_disp}"
+
     tz = get_timezone_by_city(city)
     if tz.startswith("❌"):
-        return tz
-    ct = get_current_time_in_timezone(tz)
-    if ct.startswith("❌"):
-        return ct
+        time_line = tz
+    else:
+        ct = get_current_time_in_timezone(tz)
+        time_line = f"{time_labels.get(lang,'🕒 Time')}: {ct}"
+
     aq = get_air_quality(city, lang)
     w = get_weather(city, lang)
-    return f"🕒 {'Time' if lang=='en' else 'Время'}: {ct}\n\n{aq}\n\n{w}"
+
+    return f"{location_line}\n\n{time_line}\n\n{aq}\n\n{w}"
 
 if __name__ == "__main__":
     gr.Interface(