Update app.py

app.py

@@ -81,13 +81,16 @@ if uploaded_file is not None:
     df['category'] = df['Abstract'].apply(predict_category)
     st.dataframe(df)
 
-st.sidebar.header("Download Results")
-st.sidebar.text("Download the tagged results as a CSV file.")
-
-# Create a download button
-# 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)
+# Create download tab
+with st.sidebar:
+    st.header("Download")
+    if st.button("Download CSV file"):
+        download_csv(df)
+
+def download_csv(df):
+    """Downloads a CSV file from the given DataFrame."""
+    b64_df = base64.b64encode(df.to_csv().encode()).decode()
+    headers = {"Content-Disposition": "attachment; filename=data.csv"}
+    st.download_file(b64_df, "data.csv", headers=headers)
+
+