Update app.py

app.py

@@ -1,86 +1,56 @@
 import os
-import json
-import gradio as gr
 import requests
-from groq import Groq  # Ensure this is the correct import
-
-# Groq client initialization with environment variables
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-
-def preprocess_city_name_and_generate_insight(city_name):
-    tools = [
-        {
-            "type": "function",
-            "function": {
-                "name": "generate_city_insight",
-                "description": "Generate a short, interesting fact about a specified city",
-                "parameters": {
-                    "type": "object",
-                    "properties": {
-                        "city_name": {
-                            "type": "string",
-                            "description": "The name of the city to generate insights for",
-                        },
-                    },
-                    "required": ["city_name"],
-                },
-            },   
+import gradio as gr
+from langchain_groq import ChatGroq
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+
+groq_api_key = os.getenv("groq")
+
+def weather(city):
+    url = 'http://api.openweathermap.org/data/2.5/weather?q=' + city + '&appid=' + os.getenv("openweather") + '&units=metric'
+    res = requests.get(url)
+    data = res.json()
+    if res.status_code == 200:
+        weather_details = {
+            "humidity": data.get('main', {}).get('humidity'),
+            "pressure": data.get('main', {}).get('pressure'),
+            "wind": data.get('wind', {}).get('speed'),
+            "description": data.get('weather', [{}])[0].get('description'),
+            "temp": data.get('main', {}).get('temp')
         }
-    ]
-
-    # Simulate a user message asking for insights about a city
-    response = client.chat.completions.create(
-        model="mixtral-8x7b-32768",
-        messages=[
-            {
-                "role": "user",
-                "content": f"What interesting fact can you tell me about {city_name}?",
-            }
-        ],
-        temperature=0,
-        max_tokens=300,
-        tools=tools,
-        tool_choice="auto"
-    )
-
-    # Extract and handle the response
-    groq_response = response.choices[0].message
-    args = json.loads(groq_response.tool_calls[0].function.arguments)
-
-    # Assuming generate_city_insight function is defined somewhere that returns an insight
-    insight = generate_city_insight(**args)
-    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}")
+        return weather_details
+    else:
+        return None
+
+def generate_weather_report(city):
+    weather_res = weather(city)
+    if weather_res:
+        humidity = weather_res['humidity']
+        pressure = weather_res['pressure']
+        wind = weather_res['wind']
+        description = weather_res['description']
+        temp = weather_res['temp']
+
+        # Adding weather emoji based on the description
+        weather_emoji = "☀️" if "clear" in description else "☁️" if "cloud" in description else "🌧️" if "rain" in description else "❄️" if "snow" in description else "🌫️"
+
+        system = f"You are a resourceful weather expert of {city}. Compile a short weather summary based on information like {temp}°C is the current temperature of the city, {humidity}% is the humidity percentage for the city, and the weather of city can be best described with an interesting insight about the city {description} {weather_emoji}."
+        human = "{text}"
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", system),
+            ("human", human)
+        ])
+        chat = ChatGroq(api_key=groq_api_key, model_name="llama3-70b-8192")
+        chain = prompt | chat | StrOutputParser()
+        output = chain.invoke({"text": city})
+        return output
     else:
-        return "Failed to retrieve data. Please check the city name and try again."
+        return "Weather data not available. Please check the city name or try again later."
 
-# Gradio interface definition
-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."
-)
+def main(city):
+    weather_report = generate_weather_report(city)
+    return weather_report
 
-# Launching the interface
+iface = gr.Interface(fn=main, inputs=[gr.Textbox(label="Enter City Name")], outputs=[gr.Textbox(label="Weather Report")])
 iface.launch()