Update app.py

app.py

@@ -169,72 +169,72 @@ 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.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)
-            write_to_file("retrieved_docs_train.json", retrieved_out_train)
+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.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)
+        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'])
         
-            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
+        # Step 2: Lowercase the documents
+        
+        st.header('Retrieved Documents')
+        
+        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]
             
-            st.header('Retrieved Documents')
+            organize_list = []
+            ui_organize_list = []
             
-            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")
+            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")
                 topics_list = [slide_title]
-                
-                organize_list = []
-                ui_organize_list = []
-                
+                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 = 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
+                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