Spaces:
Sleeping
Sleeping
| import requests | |
| import streamlit as st | |
| import json | |
| from dotenv import load_dotenv | |
| import os | |
| from groq import Groq | |
| load_dotenv() | |
| # Get GROQ API key from environment variable | |
| client = Groq( | |
| api_key=os.getenv("GROQ_API_KEY"), | |
| ) | |
| # Get OpenWeatherMap API key from environment variable | |
| weather_api_key = os.getenv("OPENWEATHERMAP_API_KEY") | |
| # OpenWeatherMap API endpoint | |
| API_URL = "http://api.openweathermap.org/data/2.5/weather" | |
| def get_current_weather(location): | |
| """Get the current weather in a given location""" | |
| params = { | |
| "q": location, | |
| "appid": weather_api_key, | |
| "units": "metric" | |
| } | |
| response_data = requests.get(API_URL, params=params) | |
| if response_data.status_code == 200: | |
| data = response_data.json() | |
| print("data-", data) | |
| location_resp = data['name'] | |
| weather = data['weather'][0]['main'] | |
| temperature = round(data['main']['temp'], 2) | |
| description = data['weather'][0]['description'] | |
| humidity = data['main']['humidity'] | |
| pressure = data['main']['pressure'] | |
| wind = data['wind']['speed'] | |
| report = (f"We have an update on the weather for {location_resp} location. Currently, it is {weather} " | |
| f"({description}) with a temperature of {temperature} °C. There's {humidity}% humidity, " | |
| f"atmospheric pressure is {pressure} hPa, and winds are blowing at {wind} m/s.") | |
| return report | |
| else: | |
| return f"Uh oh! We're unable to retrieve weather data for {location}. Please try again later." | |
| def define_tools(): | |
| """Define tools for the OpenAI API call""" | |
| tools = [ | |
| { | |
| "type": "function", | |
| "function": { | |
| "name": "get_current_weather", | |
| "description": "Get the current weather in a given location", | |
| "parameters": { | |
| "type": "object", | |
| "properties": { | |
| "location": { | |
| "type": "string", | |
| "description": "The city and state, e.g. San Francisco, CA", | |
| }, | |
| }, | |
| "required": ["location"], | |
| }, | |
| }, | |
| } | |
| ] | |
| return tools | |
| st.title("Weather App") | |
| location = st.text_input("Enter a location (city and state):") | |
| if st.button("Get Weather"): | |
| tools = define_tools() | |
| response = client.chat.completions.create( | |
| model="mixtral-8x7b-32768", | |
| messages=[ | |
| { | |
| "role": "user", | |
| "content": f"What is the weather like in {location}?", | |
| } | |
| ], | |
| temperature=0, | |
| max_tokens=300, | |
| tools=tools, | |
| tool_choice="auto" | |
| ) | |
| # Handle potential NoneType errors | |
| if response and response.choices and response.choices[0].message: | |
| groq_response = response.choices[0].message | |
| if groq_response and groq_response.tool_calls and groq_response.tool_calls[0].function.arguments: | |
| args = json.loads(groq_response.tool_calls[0].function.arguments) | |
| weather_report = get_current_weather(**args) | |
| st.write("**Weather Report:**") | |
| st.write(weather_report) | |
| else: | |
| st.write("Oops! There seems to be an issue retrieving the weather data. Please check the location provided " | |
| "correctly") | |
| else: | |
| st.write("Oops! There seems to be an issue retrieving the weather data. Please try again later.") | |