Update app.py

app.py

@@ -117,30 +117,30 @@ if 'last_selected_app' in st.session_state:
 
 st.session_state.last_selected_app = selected_app
 
-if selected_app == "1) Create CSVs":
-    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
-    
-    if st.session_state.error != "":
-        st.error(st.session_state.error)
-        
-    if st.session_state.success != None:
-        st.success("Success! Download the Q/A pairs below / Click reset to upload more PDFs")
-        st.download_button(
-            label="Download CSV",
-            data=st.session_state.success,
-            file_name='questions_answers.csv',
-            mime='text/csv',
-        )
-        if st.button('Reset'):
-            st.session_state.clear()
-            st.rerun()
-            
-    else:
+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
+
+if st.session_state.error != "":
+    st.error(st.session_state.error)
+
+if st.session_state.success != None:
+    st.success("Success! Download the Q/A pairs below / Click reset to upload more PDFs")
+    st.download_button(
+        label="Download CSV",
+        data=st.session_state.success,
+        file_name='questions_answers.csv',
+        mime='text/csv',
+    )
+    if st.button('Reset'):
+        st.session_state.clear()
+        st.rerun()
+
+if selected_app == "1) Create CSVs":  
+    if st.session_state.success == None:
         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")
@@ -151,7 +151,6 @@ if selected_app == "1) Create CSVs":
         openai_api_key = st.text_input("Enter your OpenAI API key", type="password")
     
         submit = st.button("Submit")
-    
         if submit:
             st.session_state.submit = True
         
@@ -203,18 +202,19 @@ if selected_app == "1) Create CSVs":
                 st.rerun()
 
 if selected_app == "2) Merge CSVs":
-    uploaded_files = st.file_uploader("Upload CSV files to merge", accept_multiple_files=True, type="csv")
-
-    submit = st.button("Submit")
-    if submit:
-        st.session_state.submit = True
+    if st.session_state.success == None:
+        uploaded_files = st.file_uploader("Upload CSV files to merge", accept_multiple_files=True, type="csv")
     
-    if st.session_state.submit:
-        if len(uploaded_files) > 1:
-            a = 1
-        else:
-            st.session_state.error = "Please upload at least 2 CSVs to merge"
-            st.session_state.success = None
-            st.session_state.submit = False
-            st.rerun()
+        submit = st.button("Submit")
+        if submit:
+            st.session_state.submit = True
+        
+        if st.session_state.submit:
+            if len(uploaded_files) > 1:
+                a = 1
+            else:
+                st.session_state.error = "Please upload at least 2 CSVs to merge"
+                st.session_state.success = None
+                st.session_state.submit = False
+                st.rerun()