app.py

import gradio as gr
import requests
import os
from huggingface_hub import InferenceClient

def preprocess_city_name_and_generate_insight(city_name):
    token = os.getenv("HF_TOKEN")
    if not token:
        return city_name, "Hugging Face token is not set in the environment variables."
    client = InferenceClient(model="gpt2", token=token)
    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, 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."
        
    url = f'https://api.openweathermap.org/data/2.5/weather?q={corrected_city_name}&appid={API_Key}&units=metric'
    response = requests.get(url)
    
    if response.status_code == 200:
        data = response.json()
        weather = data['weather'][0]['description']
        temp = data['main']['temp']
        humidity = data['main']['humidity']
        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}% 💧. "
                f"Did you know? {city_insight}")
    else:
        return "Failed to retrieve data. Please check the city name and try again."

# 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="WeatherAssistantApp",
    description="Enter a city name to get a lively description of the current weather, temperature, and humidity."
)

# Launch the interface
iface.launch()