Update app.py

app.py

@@ -68,12 +68,16 @@ with st.sidebar:
     if radius_visible:
         radius_in_meters = st.slider('Selecione raio (em metros)', min_value=100, max_value=5000, value=1000)
 
-# Filter data based on the radius
-if selected_coords == 'Custom':
+if selected_coords == 'Custom' and radius_visible:
+    # Filter data based on the radius and drop NaN values
     filtered_data = data[data.apply(lambda x: calculate_distance(x['latitude'], x['longitude'], custom_lat, custom_lon), axis=1) <= radius_in_meters]
-    filtered_data = filtered_data.dropna()  # Drop rows with NaN values
-else:
-    filtered_data = data
+    filtered_data = filtered_data.dropna()
+
+    # Apply KNN and get predicted Vunit values
+    predicted_vunit = knn_predict(filtered_data, 'Vunit', knn_features)
+
+    # Add predicted Vunit values to filtered_data
+    filtered_data['Predicted_Vunit'] = predicted_vunit
 
 
 # Add a custom CSS class to the map container