import gradio as gr
import os
import requests
from groq import Groq 

def preprocess_city_name_and_generate_insight(city_name):
    api_key = os.getenv("GROQ_API_KEY")
    if not api_key:
        return city_name, "GROQ API key is not set in the environment variables."

    try:
        client = Groq(api_key=api_key)

        prompt = f"Generate a short, interesting fact about {city_name} in less than 10 words."
        result = client.query(prompt=prompt, model_id="llama3-70b-8192")  # Adjust parameters as necessary

        insight = result.get('generated_text', "No insight generated.").strip()
    except Exception as e:
        print(f"Error generating insight: {str(e)}")
        insight = "Could not generate insight due to an error."

    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."

# Defining 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."
)

# Launching the interface
iface.launch()