Update app.py

app.py

@@ -140,7 +140,7 @@ if st.session_state.success != None:
         st.rerun()
 
 if selected_app == "1) Create CSVs":  
-    if st.session_state.success == None:
+    if not st.session_state.success:
         uploaded_files = st.file_uploader("Upload PDFs Here", type="pdf", accept_multiple_files=True)
         
         question_protocol = st.text_input("Provide instructions for how questions should be generated", "Write a question based on the text")
@@ -168,9 +168,8 @@ if selected_app == "1) Create CSVs":
                         ]
                     )
                 except:
+                    st.session_state.clear()
                     st.session_state.error = "OpenAI API key is invalid"
-                    st.session_state.success = None
-                    st.session_state.submit = False
                     st.rerun()
                     
                 with st.spinner("Loading, please be patient with us ... 🙏"):
@@ -190,19 +189,17 @@ if selected_app == "1) Create CSVs":
                     df = df.drop('context', axis=1)
         
                     csv = df.to_csv(index=False).encode('utf-8')
-                
-                    st.session_state.error = ""
+
+                    st.session_state.clear()
                     st.session_state.success = csv
-                    st.session_state.submit = False
                     st.rerun()
             else:
+                st.session_state.clear()
                 st.session_state.error = "Please upload at least 1 PDF"
-                st.session_state.success = None
-                st.session_state.submit = False
                 st.rerun()
 
 if selected_app == "2) Merge CSVs":
-    if st.session_state.success == None:
+    if not st.session_state.success:
         uploaded_files = st.file_uploader("Upload CSV files to merge", accept_multiple_files=True, type="csv")
     
         submit = st.button("Submit")
@@ -213,8 +210,6 @@ if selected_app == "2) Merge CSVs":
             if len(uploaded_files) > 1:
                 a = 1
             else:
+                st.session_state.clear()
                 st.session_state.error = "Please upload at least 2 CSVs to merge"
-                st.session_state.success = None
-                st.session_state.submit = False
-                st.rerun()
-    
+                st.rerun()