Update app.py

app.py

@@ -20,7 +20,7 @@ fpretrain_lst = ["None", "-scsmall", "-scmedium", "-sclarge"]
 finetune_lst = ["-squad", "-scqa1", "-scqa2"]
 
 # 为每个选项创建下拉菜单
-st.markdown(f'bert', unsafe_allow_html=True)
+st.markdown(f'###### bert', unsafe_allow_html=True)
 size = st.selectbox("Choose a model size:", size_lst)
 cased = st.selectbox("Whether distinguish upper and lowercase letters:", cased_lst)
 fpretrain = st.selectbox("Further pretrained on a solar cell corpus:", fpretrain_lst)
@@ -119,23 +119,16 @@ if st.button('Extract the property'):
 st.markdown('## 🖥️ QA Dataset Auto Generation', unsafe_allow_html=True)
 st.markdown('### Algorithm Input', unsafe_allow_html=True)
 
-cde_lst = ["./CDE_properties.jsonl"]
-paper_lst = ["./reference_paper.json"]
-
-st.session_state['cde'] = "./CDE_properties.jsonl"
-st.session_state['cde'] = st.selectbox("ChemDataExtractor generated database path:", cde_lst)
-st.write("Example of the data record in the ChemDataExtractor generated database: ")
-with open(st.session_state['cde'], 'r') as file:
+st.markdown(f###### Example of the data record in the ChemDataExtractor generated database: ', unsafe_allow_html=True)
+with open("./CDE_properties.jsonl", 'r') as file:
     for line in file:
         json_data = json.loads(line.strip())
         json_string = json.dumps(json_data, indent=4)
         st.text_area("", value=json_string, height=200)
         break
 
-st.session_state['paper'] = "./reference_paper.json"
-st.session_state['paper'] = st.selectbox("Paper collection path:", paper_lst)
-st.write("Example of the paper collection: ")
-with open(st.session_state['paper'],'r+', encoding = "utf-8") as f:
+st.markdown('###### Example of the paper collection: ', unsafe_allow_html=True)
+with open("./reference_paper.json",'r+', encoding = "utf-8") as f:
     string = f.read()
     json_data = json.loads(string)
     json_string = json.dumps(json_data, indent=4)
@@ -143,6 +136,26 @@ with open(st.session_state['paper'],'r+', encoding = "utf-8") as f:
 
 st.markdown('### Algorithm Output', unsafe_allow_html=True)
 
+st.markdown('###### Extracted first-turn QA pairs: ', unsafe_allow_html=True)
+file_path = "./output_qa.jsonl"
+json_list = []
+with open(file_path, 'r', encoding="utf-8") as file:
+    for line in file:
+        json_dict = json.loads(line.strip())
+        json_list.append(json_dict)
+json_string = json.dumps(json_list, indent=4)
+st.text_area("", value=json_string, height=200)
+
+st.markdown('###### Extracted second-turn QA pairs: ', unsafe_allow_html=True)
+file_path = "output_secondqa.jsonl"
+json_list = []
+with open(file_path, 'r', encoding="utf-8") as file:
+    for line in file:
+        json_dict = json.loads(line.strip())
+        json_list.append(json_dict)
+json_string = json.dumps(json_list, indent=4)
+st.text_area("", value=json_string, height=200)
+
 st.write("To be updated")