Update app.py

app.py

@@ -81,16 +81,12 @@ if uploaded_file is not None:
     df['category'] = df['Abstract'].apply(predict_category)
     st.dataframe(df)
 
-# 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)
-
+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()
+    headers = {"Content-Disposition": "attachment; filename=data.csv"}
+    st.download_file(b64, "data.csv", headers=headers)