Update app.py

app.py

@@ -110,7 +110,14 @@ st.title("Chatbot Guide")
 if st.session_state.error != "":
     st.error(st.session_state.error)
 if st.session_state.success != None:
-    st.success(st.session_state.success)
+    st.success(
+        st.download_button(
+            label="Download Q/A pairs as CSV",
+            data=csv,
+            file_name='questions_answers.csv',
+            mime='text/csv',
+        )
+    )
 
 uploaded_files = st.file_uploader("Upload PDFs Here", type="pdf", accept_multiple_files=True)
 st.success(f"{len(uploaded_files)} document(s) loaded...")
@@ -161,15 +168,9 @@ if st.session_state.submit:
         df = df.drop('context', axis=1)
 
         csv = df.to_csv(index=False).encode('utf-8')
-        download = st.download_button(
-            label="Download Q/A pairs as CSV",
-            data=csv,
-            file_name='questions_answers.csv',
-            mime='text/csv',
-        )
         
         st.session_state.error = ""
-        st.session_state.success = download
+        st.session_state.success = csv
         st.session_state.submit = False
         st.rerun()
     else: