Update app.py

app.py

@@ -35,17 +35,26 @@ def save_list_as_excel(text):
     st.success(f"Character list saved as {file_name}")
     return file_name
 
-def get_download_link(file_path, file_type):
-    with open(file_path, 'rb') as f:
-        data = f.read()
-        b64 = base64.b64encode(data).decode()
-        if file_type == "xlsx":
-            href = f'<a href="data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64}" download="{file_path}">Download {file_path}</a>'
-        elif file_type == "csv":
-            href = f'<a href="data:text/csv;base64,{b64}" download="{file_path}">Download {file_path}</a>'
-        elif file_type == "md":
-            href = f'<a href="data:text/markdown;base64,{b64}" download="{file_path}">Download {file_path}</a>'
-        return href
+@st.cache_resource
+def get_table_download_link(file_path):
+    try:
+        with open(file_path, 'rb') as file:
+            data = file.read()
+            b64 = base64.b64encode(data).decode()
+            file_name = os.path.basename(file_path)
+            ext = os.path.splitext(file_name)[1]  # get the file extension
+            if ext == '.xlsx':
+                mime_type = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
+            elif ext == '.csv':
+                mime_type = 'text/csv'
+            elif ext == '.md':
+                mime_type = 'text/markdown'
+            else:
+                mime_type = 'application/octet-stream'  # general binary data type
+            href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
+            return href
+    except:
+        return ''
 
 def perform_nlp(text):
     sentences = sent_tokenize(text)
@@ -106,9 +115,9 @@ def main():
                     df = pd.read_excel(file_name)
                     st.subheader("Saved Data")
                     st.dataframe(df)
-                    st.markdown(get_download_link(file_name, "xlsx"), unsafe_allow_html=True)
-                    st.markdown(get_download_link(file_name.replace(".xlsx", ".csv"), "csv"), unsafe_allow_html=True)
-                    st.markdown(get_download_link(file_name.replace(".xlsx", ".md"), "md"), unsafe_allow_html=True)
+                    st.markdown(get_table_download_link(file_name), unsafe_allow_html=True)
+                    st.markdown(get_table_download_link(file_name.replace(".xlsx", ".csv")), unsafe_allow_html=True)
+                    st.markdown(get_table_download_link(file_name.replace(".xlsx", ".md")), unsafe_allow_html=True)
                 except:
                     pass