Update app.py

app.py

@@ -102,11 +102,15 @@ if 'submit' not in st.session_state:
     st.session_state.submit = False
 if 'error' not in st.session_state:
     st.session_state.error = ""
+if 'success' not in st.session_state:
+    st.session_state.success = None
 
 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)
 
 uploaded_files = st.file_uploader("Upload PDFs Here", type="pdf", accept_multiple_files=True)
 st.success(f"{len(uploaded_files)} document(s) loaded...")
@@ -137,6 +141,7 @@ if st.session_state.submit:
             )
         except:
             st.session_state.error = "OpenAI API key is invalid"
+            st.session_state.success = None
             st.session_state.submit = False
             st.rerun()
         
@@ -156,16 +161,19 @@ if st.session_state.submit:
         df = df.drop('context', axis=1)
 
         csv = df.to_csv(index=False).encode('utf-8')
-        st.download_button(
+        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.submit = False
         st.rerun()
     else:
         st.session_state.error = "Please upload at least 1 PDF"
+        st.session_state.success = None
         st.session_state.submit = False
         st.rerun()