Update app.py

app.py

@@ -198,12 +198,12 @@ 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')
+filtered_data['area_feature'] = np.where(filtered_data['Apriv'] != 0, filtered_data['Apriv'], filtered_data['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', area_feature])  # 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