Change download button

app.py

@@ -1,5 +1,4 @@
 import io
-import os
 import pandas as pd
 import streamlit as st
 
@@ -137,9 +136,12 @@ if uploaded_files and submit:
                     dataframe.to_excel(writer, sheet_name='Original')
                     writer.close()
 
-                st.download_button(
-                    label="Save Result",
-                    data=buffer,
-                    file_name=f"{uploaded_file.name.replace('.pdf', '')}_{chunk_option}_{model.split('-')[0]}_{model_val.split('-')[0]}.xlsx",
-                    mime='application/vnd.ms-excel'
+                st.markdown(
+                    create_download_link(
+                        "application/vnd.ms-excel", 
+                        buffer.getvalue(), 
+                        f"{uploaded_file.name.replace('.pdf', '')}_{chunk_option}_{model.split('-')[0]}_{model_val.split('-')[0]}.xlsx", 
+                        "Save Result"
+                    ),
+                    unsafe_allow_html=True
                 )

utils.py

@@ -1,3 +1,4 @@
+import base64
 import os
 import pandas as pd
 import re
@@ -120,3 +121,7 @@ def integrate(df):
     df.reset_index(drop=True, inplace=True)
 
     return df
+
+def create_download_link(type, val, filename, message):
+    b64 = base64.b64encode(val)
+    return f'<a href="data:{type};base64,{b64.decode()}" download="{filename}">{message}</a>'