Update app.py

app.py

@@ -3,6 +3,9 @@ import pandas as pd
 import numpy as np
 from sklearn.neighbors import KNeighborsRegressor
 from geopy.distance import geodesic
+import googlemaps
+from geopy.exc import GeocoderTimedOut
+
 
 # Function to calculate distance in meters between two coordinates
 def calculate_distance(lat1, lon1, lat2, lon2):
@@ -72,21 +75,30 @@ zoom_level = round(15 - np.log10(max_distance))
 
 # Create a sidebar for controls
 with st.sidebar:
-    # Display a title
     st.title('avalia.se')
 
-    # Dropdown to select specific coordinates
     selected_coords = st.selectbox('Selecione Coordenadas', ['Random', 'Custom'])
+
     if selected_coords == 'Custom':
-        custom_lat = st.number_input('Enter Latitude', value=filtered_data['latitude'].mean())
-        custom_lon = st.number_input('Enter Longitude', value=filtered_data['longitude'].mean())
+        custom_address = st.text_input('Enter Address', 'Your Address Here')
         radius_visible = True  # Show radius slider for custom coordinates
         # No need to initialize max_distance_all here
     else:
-        custom_lat, custom_lon = data['latitude'].mean(), data['longitude'].mean()
+        custom_address = "Porto Alegre, Brazil"  # Default address
         radius_visible = False  # Hide radius slider for random coordinates
         max_distance_all = 0  # Initialize max_distance_all here
 
+        max_distance_all = 0  # Initialize max_distance_all here
+
+    # Geocode the custom address using the Google Maps API
+    gmaps = googlemaps.Client(key='AIzaSyDoJ6C7NE2CHqFcaHTnhreOfgJeTk4uSH0')  # Replace with your API key
+
+    try:
+        location = gmaps.geocode(custom_address)[0]['geometry']['location']
+        custom_lat, custom_lon = location['lat'], location['lng']
+    except (IndexError, GeocoderTimedOut):
+        st.error("Error: Unable to geocode the provided address. Please check and try again.")
+
     # Slider for setting the zoom level
     if selected_coords == 'Custom':
         zoom_level = st.slider('Nível de zoom', min_value=1, max_value=15, value=zoom_level)