Update AlphaEdu.py

AlphaEdu.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import pandas as pd
-from plms.language_model import TransformersQG
+# from plms.language_model import TransformersQG
 import time
 import os
 import numpy as np
@@ -76,22 +76,22 @@ lesson = col_11.selectbox(label='Lesson:', options=lesson_names, label_visibilit
 
 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 = st.text_area(label='Paragraph content', label_visibility='visible', height=200, value=context_values[paragraph_idx - 1])
 
-col_13, col_23, col_33, col_43 = st.columns(spec=[0.8, 1.2, 1.2, 0.8])
+col_13, col_23, col_33 = st.columns(spec=[3.6, 2.4, 3.6])
 col_23.selectbox(label='QAG model:', options=['ViT5-ViNewsQA'], label_visibility='visible')
-btn_show_answer = col_33.toggle(label='Show answers', disabled=False)
+btn_show_answer = col_23.toggle(label='Show answers', disabled=False)
 
 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 paragraph. Please wait ...'):
-        st.session_state.output = generateQA(context=context_values)
+        st.session_state.output = generateQA(context=paragraph)
 
 if btn_show_answer:
     if st.session_state.output != '':
-        st.markdown("<h8 style='text-align: left; font-weight: normal'>Generated QA pairs:</h8>", unsafe_allow_html=True)
+        # 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