Update app.py

app.py

@@ -163,71 +163,74 @@ 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]))
-    
-    if st.button('Retrieve'):
-        if 'organize_clicked' not in st.session_state:
-            st.session_state.organize_clicked = False        
-        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)
-        write_to_file("retrieved_docs_train.json", retrieved_out_train)
-    
-        for ret_item in retrieved_out:
-            for key in ret_item['ref_abstract']:
-                # inp_doc_list.append(item['ref_abstract'][key])    
-                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')
+
+    st.session_state.submit_clicked = True
+
+    if st.session_state.submit_clicked:
+        if st.button('Retrieve'):
+            if 'organize_clicked' not in st.session_state:
+                st.session_state.organize_clicked = False        
+            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)
+            write_to_file("retrieved_docs_train.json", retrieved_out_train)
         
-        for i in range(len(section_names)):
-            ui_retrieved_doc_list.append(st.text_area(retrieve_prompt_template.format(inp_keys_list[i], ref_doc_indices[i], section_names[i], document_name), value=retrieved_doc_list[i]))
-    if st.session_state.retrieve_clicked:
-        if st.button('Organize'):
-            if 'summarize_clicked' not in st.session_state:
-                st.session_state.summarize_clicked = False                    
-            st.session_state.organize_clicked = True
-            st.header("Organization of the documents in the narrative")
-            topics_list = [slide_title]
+            for ret_item in retrieved_out:
+                for key in ret_item['ref_abstract']:
+                    # inp_doc_list.append(item['ref_abstract'][key])    
+                    retrieved_doc_list.append(ret_item['ref_abstract'][key]['abstract'])
             
-            organize_list = []
-            ui_organize_list = []
+            # Step 2: Lowercase the documents
+            st.session_state.retrieve_clicked = True
+            st.header('Retrieved Documents')
             
-            test_list = read_from_file("retrieved_docs.json")
-            train_list = read_from_file("retrieved_docs_train.json")
-            organize_out = get_retrieved_results("gpt4o", 1, "fixed", train_list, test_list, True)
-            for i in range(len(organize_out)):
-                organize_list.append(organize_out[i])
-                ui_organize_list.append(st.text_area("Section: " + topics_list[i], value=organize_out[i]))
-            write_to_file("organized_docs.json", organize_out)
-
-        if st.session_state.organize_clicked:
-            if st.button("Summarize"):
-                # if 'narrative_clicked' not in st.session_state:
-                #     st.session_state.narrative_clicked = False                                    
-                st.session_state.summarize_clicked = True
-                st.header("Intent-based multi-document summary")
+            for i in range(len(section_names)):
+                ui_retrieved_doc_list.append(st.text_area(retrieve_prompt_template.format(inp_keys_list[i], ref_doc_indices[i], section_names[i], document_name), value=retrieved_doc_list[i]))
+        if st.session_state.retrieve_clicked:
+            if st.button('Organize'):
+                if 'summarize_clicked' not in st.session_state:
+                    st.session_state.summarize_clicked = False                    
+                st.session_state.organize_clicked = True
+                st.header("Organization of the documents in the narrative")
                 topics_list = [slide_title]
-                generate_list = []
-                ui_generate_list = []
-                slides_list = []
+                
+                organize_list = []
+                ui_organize_list = []
+                
                 test_list = read_from_file("retrieved_docs.json")
-                train_list = read_from_file("retrieved_docs_train.json")                
-                organize_out = read_from_file("organized_docs.json")
-                gen_summary_dict = get_retrieved_results("gpt4o", 1, "fixed", train_list, test_list, False, organize_out)
-                for i in range(len(gen_summary_dict)):
-                    # highlighted_summary = highlight_sentences(gen_summary_dict[i], test_list[i]['abstract'])
-                    slides_list.append(get_slide(topics_list[i], gen_summary_dict[i]))
-                    # generate_list.append(.format(topics_list[i], gen_summary_dict[i]))
-                    st.markdown(f"## {topics_list[i]}")
-                    st.markdown(f"{gen_summary_dict[i]}")
-                    # st.markdown(highlighted_summary, unsafe_allow_html=True)
-                st.header("Generated Narrative")
-                for i in range(len(slides_list)):
-                    st.markdown("---")
-                    st.markdown(slides_list[i])
-                    st.markdown("---")
-            # if st.session_state.summarize_clicked:
-            #     if st.button("Narrative"):                
-            #         st.session_state.narrative_clicked = True
+                train_list = read_from_file("retrieved_docs_train.json")
+                organize_out = get_retrieved_results("gpt4o", 1, "fixed", train_list, test_list, True)
+                for i in range(len(organize_out)):
+                    organize_list.append(organize_out[i])
+                    ui_organize_list.append(st.text_area("Section: " + topics_list[i], value=organize_out[i]))
+                write_to_file("organized_docs.json", organize_out)
+    
+            if st.session_state.organize_clicked:
+                if st.button("Summarize"):
+                    # if 'narrative_clicked' not in st.session_state:
+                    #     st.session_state.narrative_clicked = False                                    
+                    st.session_state.summarize_clicked = True
+                    st.header("Intent-based multi-document summary")
+                    topics_list = [slide_title]
+                    generate_list = []
+                    ui_generate_list = []
+                    slides_list = []
+                    test_list = read_from_file("retrieved_docs.json")
+                    train_list = read_from_file("retrieved_docs_train.json")                
+                    organize_out = read_from_file("organized_docs.json")
+                    gen_summary_dict = get_retrieved_results("gpt4o", 1, "fixed", train_list, test_list, False, organize_out)
+                    for i in range(len(gen_summary_dict)):
+                        # highlighted_summary = highlight_sentences(gen_summary_dict[i], test_list[i]['abstract'])
+                        slides_list.append(get_slide(topics_list[i], gen_summary_dict[i]))
+                        # generate_list.append(.format(topics_list[i], gen_summary_dict[i]))
+                        st.markdown(f"## {topics_list[i]}")
+                        st.markdown(f"{gen_summary_dict[i]}")
+                        # st.markdown(highlighted_summary, unsafe_allow_html=True)
+                    st.header("Generated Narrative")
+                    for i in range(len(slides_list)):
+                        st.markdown("---")
+                        st.markdown(slides_list[i])
+                        st.markdown("---")
+                # if st.session_state.summarize_clicked:
+                #     if st.button("Narrative"):                
+                #         st.session_state.narrative_clicked = True