Update app.py

app.py

@@ -73,20 +73,21 @@ def conv_txt(extype):
 
 #===Read data===
 uploaded_file = st.file_uploader("Choose a file", type=['csv'])
-
+st.sidebar.header("Download Results")
+st.sidebar.text("Download the tagged results as a CSV file.")
 if uploaded_file is not None:
     df = pd.read_csv(uploaded_file, encoding='latin-1')
     st.dataframe(df)
+    
     # Tag the "Abstract" column with the corresponding categories
-    df['category'] = df['Abstract'].apply(predict_category)
+    df['category'] = df['Abstract'].apply(lambda x: predict_category(x)[0])
     st.dataframe(df)
-
-st.sidebar.header("Download Results")
-st.sidebar.text("Download the tagged results as a CSV file.")
-
-# Create a download button
-if st.sidebar.button("Download"):
-    csv = df.to_csv(index=False)
-    b64 = base64.b64encode(csv.encode()).decode()
-    href = f'<a href="data:file/csv;base64,{b64}" download="results.csv">Download csv file</a>'
-    st.markdown(href, unsafe_allow_html=True)
+    
+    csv = df.to_csv(index=False, encoding='utf-8-sig')  # Using utf-8-sig to handle possible BOM issues
+    b64 = base64.b64encode(csv.encode('utf-8-sig'))
+    st.sidebar.download_button(
+        label="Download", 
+        data=b64,
+        file_name="results.csv",
+        mime="text/csv"
+    )