update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import io
 import time
 import json
 import glob
@@ -28,6 +29,7 @@ Entrez.email = "[email protected]"
 Entrez.api_key = "3d7a71231fe7f2d2bd7599e022535199a908"
 
 fold = -1
+buffer = io.BytesIO()
 # chunk_size = 8000
 
 st.cache_data()
@@ -219,19 +221,22 @@ if uploaded_file:
                                 })
                             except Exception as e:
                                 print(e)
-                # result = pd.DataFrame(L)
+                # # result = pd.DataFrame(L)
                 st.dataframe(pd.DataFrame(L))
-                csv = convert_df(pd.DataFrame(L))
+                # csv = convert_df(pd.DataFrame(L))       
+                with pd.ExcelWriter(buffer, engine='xlsxwriter') as writer:
+                    # Write each dataframe to a different worksheet
+                    pd.DataFrame(L).to_excel(writer, sheet_name='Result')
+                    writer.close()
 
                 st.download_button(
                     label="Save Result",
-                    data=csv,
-                    file_name=str(uploaded_file.name).replace('.pdf', ''),
-                    mime='text/csv'
+                    data=buffer,
+                    file_name=str(uploaded_file.name).replace('.pdf', '') + '.xlsx',
+                    mime='application/vnd.ms-excel'
                 )
 
-            except json.JSONDecodeError:
+            except json.JSONDecodeError as j:
                 st.write("Sorry, we are experiencing difficulties in extracting the information. Please try again with different context length.")
-                # st.write(e)
-            except:
+            except Exception as e:
                 st.write("Sorry, we are experiencing difficulties in extracting the information. Please ensure that you input an uncorrupted file.")