Update app.py

app.py

@@ -144,7 +144,8 @@ retrieved_doc_list = []
 
 if st.button('Submit'):
     if 'retrieve_clicked' not in st.session_state:
-        st.session_state.retrieve_clicked = False    
+        st.session_state.retrieve_clicked = False  
+    st.session_state.submit_clicked = True
     # for item, ret_item in zip(list_1, retrieved_out):
     for item in list_1:
         for key in item['ref_abstract']:
@@ -152,8 +153,8 @@ if st.button('Submit'):
             inp_keys_list.append(key)
             # retrieved_doc_list.append(ret_item['ref_abstract'][key]['abstract'])    
     # Initialize session state
-    if 'retrieve_clicked' not in st.session_state:
-        st.session_state.retrieve_clicked = False
+    # if 'retrieve_clicked' not in st.session_state:
+    #     st.session_state.retrieve_clicked = False
     
     retrieve_prompt_template = "{} : Document {} for the '{}' Section of the Article titled '{}'"
     
@@ -166,7 +167,7 @@ if st.button('Submit'):
     for i in range(len(section_names)):
         ui_doc_list.append(st.text_area(retrieve_prompt_template.format(inp_keys_list[i], ref_doc_indices[i], section_names[i], document_name), value=inp_doc_list[i]))
 
-    st.session_state.submit_clicked = True
+    
 
     if st.session_state.submit_clicked:
         if st.button('Retrieve'):