Update app.py

app.py

@@ -10,6 +10,7 @@ import folium
 from branca.colormap import LinearColormap
 import base64
 
+
 # Function to add heatmap layer to folium map
 def add_heatmap_layer(map_obj, data, column_name, colormap_name, radius=15):
     heat_data = data[['latitude', 'longitude', column_name]].dropna()
@@ -258,13 +259,25 @@ with st.container():
     st.map(filtered_data, zoom=zoom_level, use_container_width=True)
     st.write("Dados:", filtered_data)  # Debug: Print filtered_data
 
-    if st.button("Baixar arquivo do Excel"):
-        with st.spinner('Downloading...'):
-            excel_file = filtered_data.to_excel("filtered_data.xlsx", index=False)
-                # Create a download link
-            b64 = base64.b64encode(excel_file.encode()).decode()
-            href = f'<a href="data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64}" download="sample_data.xlsx">Download Excel File</a>'
-            st.markdown(href, unsafe_allow_html=True)
+    if st.button('baixar EXCEL'):
+        st.write("Baixando...")
+        # Set up the file to be downloaded
+        output_df = filtered_data
+
+        # Create a BytesIO buffer to hold the Excel file
+        excel_buffer = BytesIO()
+
+        # Convert DataFrame to Excel and save to the buffer
+        with pd.ExcelWriter(excel_buffer, engine="xlsxwriter") as writer:
+            output_df.to_excel(writer, index=False, sheet_name="Sheet1")
+
+        # Reset the buffer position to the beginning
+        excel_buffer.seek(0)
+
+        # Create a download link
+        b64 = base64.b64encode(excel_buffer.read()).decode()
+        href = f'<a href="data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64}" download="sample_data.xlsx">Download Excel File</a>'
+        st.markdown(href, unsafe_allow_html=True)
     
     folium_layermap = folium.Map(location=[custom_lat, custom_lon], zoom_start=zoom_level, control_scale=True)