Update AlphaEdu.py

AlphaEdu.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 from plms.language_model import TransformersQG
 import os
+import numpy as np
 
 st.set_page_config(page_icon='🧪', page_title='ViQAG for Vietnamese Education', layout='wide', initial_sidebar_state="collapsed")
 
@@ -13,7 +14,7 @@ st.markdown(f"""
         <img src="./app/static/AlphaEdu_logo_trans.png"/>
     </div>
     """, unsafe_allow_html=True)
-st.markdown("<h1 style='text-align: center;'>ViQAG: An Automate Question Answer Generation System for Vietnamese Education</h1>", unsafe_allow_html=True)
+st.markdown("<h1 style='text-align: center;'>AlphaEdu</h1>", unsafe_allow_html=True)
 
 # =====================================================================================================
 
@@ -42,7 +43,7 @@ def load_lessons(df, grade_name, chapter_name):
 
 def load_context(df, grade_name, chapter_name, lesson_name):
     context = df[(df['grade'] == grade_name) & (df['chapter'] == chapter_name) & (df['lesson'] == lesson_name)]['context'].values
-    return '\n'.join([str(item) for item in context.tolist()])
+    return len(context), context
 
 def generateQA(context, model_path = 'shnl/vit5-vinewsqa-qg-ae'):
     unique_qa_pairs = set()
@@ -59,7 +60,7 @@ def generateQA(context, model_path = 'shnl/vit5-vinewsqa-qg-ae'):
     
 # =====================================================================================================
 
-col_1, col_2, col_3, col_4 = st.columns(spec=[1, 1, 3, 4])
+col_1, col_2, col_3, col_4, col_5 = st.columns(spec=[1, 1, 3, 3, 1])
 
 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')
@@ -76,25 +77,26 @@ col_4.markdown("<h8 style='text-align: left; font-weight: normal'>Select your le
 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')
 
-col_11, col_21 = st.columns(spec=[8, 2])
+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_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)
-context_values = load_context(df=df, grade_name=grade, chapter_name=chapter, lesson_name=lesson)
-col_11.text_area(label='Paragraph related', label_visibility='collapsed', height=300, value=context_values)
+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'>Choose question generation modes:</h8>", unsafe_allow_html=True)
-col_21.checkbox(label='Question Answer Generation', value=True)
-col_21.checkbox(label='Multiple Choice Question Generation (Coming soon)', disabled=True)
-col_21.checkbox(label='Fill-in-the-Blank Question Generation (Coming soon)', disabled=True)
+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 the answers', disabled=False)
+btn_show_answer = col_21.toggle(label='Show answers', disabled=False)
 
 btn_generate = col_21.button(label='Generate questions', use_container_width=True)
 
 if btn_generate == True:
-    with st.spinner(text='Generating 10 pairs questions-answers. Please wait ...'):
-        small_context = context_values.split('\n')[0]
-        st.session_state.output = generateQA(context=small_context)
+    with st.spinner(text='Generating QA pairs from the selected context. Please wait ...'):
+        st.session_state.output = generateQA(context=context_values)
 
 if btn_show_answer:
     if st.session_state.output != '':