Update app.py

app.py

@@ -3,18 +3,18 @@ import requests
 import os
 from huggingface_hub import InferenceClient
 
-def preprocess_city_name(city_name):
+def preprocess_city_name_and_generate_insight(city_name):
     token = os.getenv("HF_TOKEN")
     if not token:
-        return "Hugging Face token is not set in the environment variables."
+        return city_name, "Hugging Face token is not set in the environment variables."
     client = InferenceClient(model="gpt2", token=token)
-    result = client(inputs=city_name)
-    # Assuming the result contains a more standardized city name
-    corrected_city_name = result['generated_text'].strip()
-    return corrected_city_name
+    prompt = f"Generate a short, interesting fact about {city_name} in less than 10 words:"
+    result = client(inputs=prompt)
+    insight = result['generated_text'].strip()
+    return city_name, insight
 
 def get_weather(city_name):
-    corrected_city_name = preprocess_city_name(city_name)
+    corrected_city_name, city_insight = preprocess_city_name_and_generate_insight(city_name)
     API_Key = os.getenv("OPENWEATHER_API_KEY")
     if not API_Key:
         return "API Key is not set. Please set the OPENWEATHER_API_KEY environment variable."
@@ -30,18 +30,19 @@ def get_weather(city_name):
         weather_emoji = "☀️" if "clear" in weather else "☁️" if "cloud" in weather else "🌧️" if "rain" in weather else "❄️" if "snow" in weather else "🌫️"
         return (f"In {corrected_city_name}, the weather is currently {weather} {weather_emoji}. "
                 f"The temperature is a comfy {temp}°C 🌡️. "
-                f"And the humidity levels are at {humidity}%, so keep hydrated! 💧")
+                f"And the humidity levels are at {humidity}% 💧. "
+                f"Did you know? {city_insight}")
     else:
         return "Failed to retrieve data. Please check the city name and try again."
 
-# Defining Gradio interface
+# Define Gradio interface
 iface = gr.Interface(
     fn=get_weather,
     inputs=gr.Textbox(label="Enter City Name", placeholder="Type here..."),
     outputs=gr.Textbox(label="Weather Update"),
-    title="Weather App with AI City Name Correction",
-    description="Enter a city name to get a lively description of the current weather, temperature, and humidity. The city name input is preprocessed by GPT-2 to correct any errors or standardize naming."
+    title="WeatherAssistantApp",
+    description="Enter a city name to get a lively description of the current weather, temperature, and humidity."
 )
 
-# Launching the interface
+# Launch the interface
 iface.launch()