Update app.py

app.py

@@ -255,26 +255,9 @@ predicted_target = knn_predict(filtered_data, 'target_column', ['latitude', 'lon
 # Add predicted target values to filtered_data
 filtered_data['Predicted_target'] = predicted_target
 
-# Display the map and filtered_data
-#with st.container():
-    #st.map(filtered_data, zoom=zoom_level, use_container_width=True)
-    #st.write("Dados:", filtered_data)  # Debug: Print filtered_data
-
 # Display the map and filtered_data
 with st.container():
-    folium_map = folium.Map(location=[custom_lat, custom_lon], zoom_start=zoom_level, control_scale=True)
-
-    # Add heatmap layers for 'Valor_Urb', 'Valor_Eqp', and 'RENDA'
-    add_heatmap_layer(folium_map, filtered_data, 'Valor_Urb', 'RdBu_r')
-    add_heatmap_layer(folium_map, filtered_data, 'Valor_Eqp', 'RdBu_r')
-    add_heatmap_layer(folium_map, filtered_data, 'RENDA', 'RdBu_r')
-
-    # Add layer control
-    folium.LayerControl().add_to(folium_map)
-
-    # Display the map using st_folium
-    st_folium(folium_map, height=700, responsive=True)
-
+    st.map(filtered_data, zoom=zoom_level, use_container_width=True)
     st.write("Dados:", filtered_data)  # Debug: Print filtered_data
 
 k_threshold = 5
@@ -312,4 +295,17 @@ if 'Predicted_target' in filtered_data.columns and not np.all(predicted_target =
     st.write(f"Valor médio (Reais/m²) para as características selecionadas: ${mean_value:.2f}$ Reais")
     st.write(f"Os valores podem variar entre ${lower_bound:.2f}$ e ${higher_bound:.2f}$ Reais, dependendo das características dos imóveis.")
 else:
-    st.warning(f"**Dados insuficientes para inferência do valor. Mínimo necessário:** {k_threshold}")
+    st.warning(f"**Dados insuficientes para inferência do valor. Mínimo necessário:** {k_threshold}")
+
+folium_map = folium.Map(location=[custom_lat, custom_lon], zoom_start=zoom_level, control_scale=True)
+
+# Add heatmap layers for 'Valor_Urb', 'Valor_Eqp', and 'RENDA'
+add_heatmap_layer(folium_map, filtered_data, 'Valor_Urb', 'RdBu_r')
+add_heatmap_layer(folium_map, filtered_data, 'Valor_Eqp', 'RdBu_r')
+add_heatmap_layer(folium_map, filtered_data, 'RENDA', 'RdBu_r')
+
+# Add layer control
+folium.LayerControl().add_to(folium_map)
+
+# Display the map using st_folium
+st_folium(folium_map, width="100%", height="100%")