Update app.py

app.py

@@ -35,11 +35,16 @@ def save_list_as_excel(text):
     st.success(f"Character list saved as {file_name}")
     return file_name
 
-def get_download_link(file_path):
+def get_download_link(file_path, file_type):
     with open(file_path, 'rb') as f:
         data = f.read()
         b64 = base64.b64encode(data).decode()
-        href = f'<a href="data:application/octet-stream;base64,{b64}" download="{file_path}">Download {file_path}</a>'
+        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
 
 def perform_nlp(text):
@@ -101,9 +106,9 @@ def main():
                     df = pd.read_excel(file_name)
                     st.subheader("Saved Data")
                     st.dataframe(df)
-                    st.markdown(get_download_link(file_name), unsafe_allow_html=True)
-                    st.markdown(get_download_link(file_name.replace(".xlsx", ".csv")), unsafe_allow_html=True)
-                    st.markdown(get_download_link(file_name.replace(".xlsx", ".md")), unsafe_allow_html=True)
+                    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)
                 except:
                     pass