Update app.py

app.py

@@ -172,7 +172,8 @@ if st.button('Submit'):
     if st.session_state.submit_clicked:
         if st.button('Retrieve'):
             if 'organize_clicked' not in st.session_state:
-                st.session_state.organize_clicked = False        
+                st.session_state.organize_clicked = False
+            st.session_state.retrieve_clicked = True
             retrieved_out = get_retrieved_results("gpt4o", 0, "fixed", list_2, list_1)
             write_to_file("retrieved_docs.json", retrieved_out)
             retrieved_out_train = get_retrieved_results("gpt4o", 0, "fixed", list_1, list_2)
@@ -184,7 +185,7 @@ if st.button('Submit'):
                     retrieved_doc_list.append(ret_item['ref_abstract'][key]['abstract'])
             
             # Step 2: Lowercase the documents
-            st.session_state.retrieve_clicked = True
+            
             st.header('Retrieved Documents')
             
             for i in range(len(section_names)):