Update app.py

app.py

@@ -114,13 +114,17 @@ else:
     document_name = "Adobe Acrobat"
     section_names = ["Introduction"]*3+["History"]*3+["Document Cloud"]*2
     ref_doc_indices = np.arange(1,4).tolist() + np.arange(1,4).tolist() + np.arange(1,3).tolist()
+
+# Initialize session state
+if 'submit_clicked' not in st.session_state:
+    st.session_state.submit_clicked = False
     
 inp_doc_list = []
 inp_keys_list = []
 retrieved_doc_list = []
 
 
-if selection!='Select':
+if st.button('Submit'):
     # for item, ret_item in zip(list_1, retrieved_out):
     for item in list_1:
         for key in item['ref_abstract']: