Update app.py

app.py

@@ -63,7 +63,7 @@ data = pd.read_excel('ven_fim_PEDÓ_nov_23.xlsx')
 # Initialize variables to avoid NameError
 selected_coords = 'Direcionada'
 radius_visible = True
-custom_address_initial = 'Av. Senador Alberto Pasqualini, 177 - Centro, Lajeado - RS, 95900-034'  # Initial custom address
+custom_address_initial = 'Centro, Lajeado - RS, Brazil'  # Initial custom address
 custom_lat = data['latitude'].median()
 custom_lon = data['longitude'].median()
 radius_in_meters = 1500
@@ -197,10 +197,13 @@ st.markdown(f"""<style>
 }}
 </style>""", unsafe_allow_html=True)
 
+# Determine which area feature to use for prediction
+area_feature = np.where(filtered_data['Apriv'] != 0, 'Apriv', 'Atotal')
+
 # Check if KNN should be applied
 if selected_coords == 'Direcionada' and radius_visible:
     # Apply KNN and get predicted V_oferta values
-    predicted_V_oferta = knn_predict(filtered_data, 'V_oferta', ['latitude', 'longitude'])  # Update with your features
+    predicted_V_oferta = knn_predict(filtered_data, 'V_oferta', ['latitude', 'longitude', area_feature])  # Update with your features
     # Add predicted V_oferta values to filtered_data
     filtered_data['Predicted_V_oferta'] = predicted_V_oferta