Update app.py

app.py

@@ -1,10 +1,17 @@
 import requests
+import geopy
+import joblib
 import gradio as gr
-import joblib as joblib
 
 # Load the trained model
 model = joblib.load('hackathonrf.joblib')
 
+# Function to get latitude and longitude from location name
+def get_coordinates(location):
+    geolocator = geopy.geocoders.Nominatim(user_agent="air_quality_app")
+    location = geolocator.geocode(location)
+    return location.latitude, location.longitude
+
 # Function to get AQI value from OpenWeatherMap API
 def get_aqi(latitude, longitude, api_key):
     url = f"http://api.openweathermap.org/data/2.5/air_pollution?lat={latitude}&lon={longitude}&appid={api_key}"
@@ -14,15 +21,16 @@ def get_aqi(latitude, longitude, api_key):
     return aqi_value
 
 # Function to make prediction
-def predict_air_quality(latitude, longitude, api_key):
+def predict_air_quality(location, api_key):
+    latitude, longitude = get_coordinates(location)
     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", "text", "text"], 
+                      inputs=["text", "text"], 
                       outputs="text", 
                       title="Air Quality Prediction",
-                      description="Enter latitude, longitude, and OpenWeatherMap API key:")
+                      description="Enter location and OpenWeatherMap API key:")
 iface.launch()