Update app.py

app.py

@@ -297,15 +297,17 @@ if 'Predicted_target' in filtered_data.columns and not np.all(predicted_target =
 else:
     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)
+# Display the map for features
+with st.container():
+    folium_layermap = 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 heatmap layers for 'Valor_Urb', 'Valor_Eqp', and 'RENDA'
+    add_heatmap_layer(folium_layermap, filtered_data, 'Valor_Urb', 'RdBu_r')
+    add_heatmap_layer(folium_layermap, filtered_data, 'Valor_Eqp', 'RdBu_r')
+    add_heatmap_layer(folium_layermap, filtered_data, 'RENDA', 'RdBu_r')
 
-# Add layer control
-folium.LayerControl().add_to(folium_map)
+    # Add layer control
+    folium.LayerControl().add_to(folium_map)
 
-# Display the map using st_folium
-st_folium(folium_map, width="100%", height="100%")
+    # Display the map using st_folium
+    st_folium(folium_layermap, width="100%", height="100%")