Update app.py

app.py

@@ -5,6 +5,7 @@ from fuzzywuzzy import process
 import string
 from nltk.corpus import stopwords
 import nltk
+import io
 
 nltk.download('stopwords')
 
@@ -95,12 +96,16 @@ if st.button("Process"):
             st.write(f"{keyword}: {count}")
 
         # Save to Excel
-        save_results = st.checkbox("Save results to Excel")
-        if save_results:
-            df = pd.DataFrame(list(results.items()), columns=["Keyword", "Count"])
-            st.download_button(
-                label="Download Results as Excel",
-                data=df.to_excel(index=False, engine='openpyxl'),
-                file_name="results.xlsx",
-                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-            )
+        st.subheader("Download Results")
+        df = pd.DataFrame(list(results.items()), columns=["Keyword", "Count"])
+        output = io.BytesIO()
+        with pd.ExcelWriter(output, engine='openpyxl') as writer:
+            df.to_excel(writer, index=False, sheet_name="Results")
+        output.seek(0)
+
+        st.download_button(
+            label="Download Results as Excel",
+            data=output,
+            file_name="results.xlsx",
+            mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+        )