Update app.py

app.py

@@ -167,12 +167,19 @@ if st.button('Submit'):
     # doc1 = st.text_area('Document 1', value="1. What up bruh??")
     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]))
-    # write_to_file('ui_doc.json', ui_doc_list)
+    # write_to_file('ui_doc.json', ui_inp_keys_list.jsondoc_list)
+    # ref_doc_indices[i], section_names[i], document_name
+    write_to_file("inp_keys_list.json", inp_keys_list)
+    write_to_file("section_names.json", section_names)
+    write_to_file("document_name.pickle", document_name)
     
 
 if st.session_state.submit_clicked:
     if st.button('Retrieve'):
         # ui_doc_list=read_from_file('ui_doc.json')
+        inp_keys_list = read_from_file("inp_keys_list.json")
+        section_names = read_from_file("section_names.json")
+        document_name = read_from_file("document_name.pickle")        
         ui_retrieved_doc_list=[]
         if 'organize_clicked' not in st.session_state:
             st.session_state.organize_clicked = False