Update app.py

app.py

@@ -84,9 +84,8 @@ if uploaded_file is not None:
 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()
+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, "data.csv", headers=headers)
+    return st.download_file(b64_df, "data.csv", headers=headers)