Update AlphaEdu.py

AlphaEdu.py

@@ -1,10 +1,11 @@
 import streamlit as st
 import pandas as pd
 from plms.language_model import TransformersQG
+import time
 import os
 import numpy as np
 
-st.set_page_config(page_icon='🧪', page_title='ViQAG for Vietnamese Education', layout='wide', initial_sidebar_state="collapsed")
+st.set_page_config(page_icon='🧪', page_title='ViQAG for Vietnamese Education', layout='centered', initial_sidebar_state="collapsed")
 
 with open(r"./static/styles.css") as f:
     st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
@@ -60,54 +61,44 @@ def generateQA(context, model_path = 'shnl/vit5-vinewsqa-qg-ae'):
     
 # =====================================================================================================
 
-col_1, col_2, col_3, col_4, col_5 = st.columns(spec=[1, 1, 3, 3, 1])
+col_1, col_2, col_3 = st.columns(spec=[2.5, 1.5, 6])
+subject = col_1.selectbox(label='Subject:', options=filenames, label_visibility='visible')
 
-col_1.markdown("<h8 style='text-align: left; font-weight: normal'>Select your subject:</h8>", unsafe_allow_html=True)
-subject = col_1.selectbox(label='Select your subject:', options=filenames, label_visibility='collapsed')
-
-col_2.markdown("<h8 style='text-align: left; font-weight: normal'>Select your grade:</h8>", unsafe_allow_html=True)
 list_grades, df = load_grades(file_name=subject)
-grade = col_2.selectbox(label='Select your grade:', options=list_grades, label_visibility='collapsed')
+grade = col_2.selectbox(label='Grade:', options=list_grades, label_visibility='visible')
 
-col_3.markdown("<h8 style='text-align: left; font-weight: normal'>Select your chapter:</h8>", unsafe_allow_html=True)
 list_chapters, df = load_chapters(df=df, grade_name=grade)
-chapter = col_3.selectbox(label='Select your chapter:', options=list_chapters, label_visibility='collapsed')
+chapter = col_3.selectbox(label='Chapter:', options=list_chapters, label_visibility='visible')
 
-col_4.markdown("<h8 style='text-align: left; font-weight: normal'>Select your lesson:</h8>", unsafe_allow_html=True)
+col_11, col_21 = st.columns(spec=[8, 2])
 lesson_names = load_lessons(df=df, grade_name=grade, chapter_name=chapter)
-lesson = col_4.selectbox(label='Select your lesson:', options=lesson_names, label_visibility='collapsed')
+lesson = col_11.selectbox(label='Lesson:', options=lesson_names, label_visibility='visible')
 
-col_5.markdown("<h8 style='text-align: left; font-weight: normal'>Paragraph:</h8>", unsafe_allow_html=True)
 total_paragraph, context_values = load_context(df=df, grade_name=grade, chapter_name=chapter, lesson_name=lesson)
+paragraph_idx = col_21.selectbox(label='Paragraph:', options=list(np.arange(1, total_paragraph + 1)), label_visibility='visible')
+st.text_area(label='Paragraph content', label_visibility='visible', height=200, value=context_values[paragraph_idx - 1])
 
-paragraph_idx = col_5.selectbox(label='Select your lesson:', options=list(np.arange(1, total_paragraph + 1)), label_visibility='collapsed')
-
-col_11, col_21 = st.columns(spec=[9, 1])
-col_11.markdown("<h8 style='text-align: left; font-weight: normal'>Paragraph related:</h8>", unsafe_allow_html=True)
-paragraph = col_11.text_area(label='Paragraph related', label_visibility='collapsed', height=300, value=context_values[paragraph_idx - 1])
-
-col_21.markdown("<h8 style='text-align: left; font-weight: normal'>QAG modes:</h8>", unsafe_allow_html=True)
-col_21.selectbox(label='QAG model:', options=['ViT5-ViNewsQA'], label_visibility='collapsed')
-
-col_21.markdown("<h8 style='text-align: left; font-weight: normal'>Options:</h8>", unsafe_allow_html=True)
-btn_show_answer = col_21.toggle(label='Show answers', disabled=False)
+col_13, col_23, col_33, col_43 = st.columns(spec=[0.8, 1.2, 1.2, 0.8])
+col_23.selectbox(label='QAG model:', options=['ViT5-ViNewsQA'], label_visibility='visible')
+btn_show_answer = col_33.toggle(label='Show answers', disabled=False)
 
-btn_generate = col_21.button(label='Generate questions', use_container_width=True)
+col_14, col_24, col_34, col_44, col_54 = st.columns(spec=[1, 1, 1, 1, 1])
+btn_generate = col_34.button(label='Generate', use_container_width=True)
 
 if btn_generate == True:
-    with st.spinner(text='Generating QA pairs from the selected context. Please wait ...'):
-        st.session_state.output = generateQA(context=paragraph)
+    with st.spinner(text='Generating QA pairs from the selected paragraph. Please wait ...'):
+        st.session_state.output = generateQA(context=context_values)
 
 if btn_show_answer:
     if st.session_state.output != '':
-        st.markdown("<h8 style='text-align: left; font-weight: normal'>Your questions and answers has been generated:</h8>", unsafe_allow_html=True)
-        st.code(body=st.session_state.output, language='wiki')
+        st.markdown("<h8 style='text-align: left; font-weight: normal'>Generated QA pairs:</h8>", unsafe_allow_html=True)
+        st.code(body=st.session_state.output, language='latex')
     else:
         pass
 else:
     if st.session_state.output != '':
-        st.markdown("<h8 style='text-align: left; font-weight: normal'>Your questions and answers has been generated:</h8>", unsafe_allow_html=True)
+        st.markdown("<h8 style='text-align: left; font-weight: normal'>Generated QA pairs:</h8>", unsafe_allow_html=True)
         output_no_answer = st.session_state.output.split(' [SEP] ')[0].split(', answer: ')[0].replace('question: ', '')
-        st.code(body=output_no_answer, language='wiki')
+        st.code(body=output_no_answer, language='latex')
     else:
         pass