Update app.py

app.py

@@ -210,10 +210,14 @@ st.markdown(f"""<style>
 # Determine which area feature to use for prediction
 filtered_data['area_feature'] = np.where(filtered_data['Apriv'] != 0, filtered_data['Apriv'], filtered_data['Atotal'])
 
-# 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
-# Add predicted V_oferta values to filtered_data
-filtered_data['Predicted_V_oferta'] = predicted_V_oferta
+# Define the target column based on conditions
+filtered_data['target_column'] = np.where(filtered_data['Vunit_priv'] != 0, filtered_data['Vunit_priv'], filtered_data['Vunit_total'])
+
+# Apply KNN and get predicted target values
+predicted_target = knn_predict(filtered_data, 'target_column', ['latitude', 'longitude', 'area_feature'])  # Update with your features
+
+# Add predicted target values to filtered_data
+filtered_data['Predicted_target'] = predicted_target
 
 # Display the map and filtered_data
 with st.container():