Update app.py

app.py

@@ -3,6 +3,7 @@ import geopy
 import joblib
 import gradio as gr
 
+
 # Load the trained model
 model = joblib.load('hackathonrf.joblib')
 
@@ -13,23 +14,21 @@ def get_coordinates(location):
     return location.latitude, location.longitude
 
 # Function to get AQI value from OpenWeatherMap API
-def get_aqi(latitude, longitude):
-    url = f"http://api.openweathermap.org/data/2.5/air_pollution/history?lat={latitude}&lon={longitude}&start=1614556800&end=1614643200&appid=78b94879cbb50e02397e93687aa24adc"  # Hardcoded API key
-    response = requests.get(url)
-    data = response.json()
-    return data['list']
+def get_aqi(latitude, longitude, api_key):
+    url = f"http://api.openweathermap.org/data/2.5/air_pollution/history?lat={latitude}&lon={longitude}&start=1614556800&end=1614643200&appid=78b94879cbb50e02397e93687aa24adc"
+    return aqi_value
 
-# Function to make prediction and fetch historical air pollution data
-def predict_air_quality(location):
+# Function to make prediction
+def predict_air_quality(location, api_key):
     latitude, longitude = get_coordinates(location)
-    historical_data = get_aqi(latitude, longitude)
-    prediction = model.predict([[historical_data[0]['main']['aqi'], historical_data[0]['main']['aqi'], historical_data[0]['main']['aqi'], historical_data[0]['main']['aqi']]])
-    return prediction[0], historical_data
+    aqi_value = get_aqi(latitude, longitude, api_key)
+    prediction = model.predict([[aqi_value, aqi_value, aqi_value, aqi_value]])
+    return prediction[0]
 
 # Create Gradio interface
 iface = gr.Interface(fn=predict_air_quality, 
-                      inputs=["text"], 
-                      outputs=["text", "text"], 
-                      title="Air Quality Prediction with Historical Data",
-                      description="Enter city name:")
-iface.launch()
+                      inputs=["text", "text"], 
+                      outputs="text", 
+                      title="Air Quality Prediction",
+                      description="Enter location:")
+iface.launch()