Update AlphaEdu.py

AlphaEdu.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import pandas as pd
-import time
+from plms.language_model import TransformersQG
 import os
 
 st.set_page_config(page_icon='🧪', page_title='ViQAG for Vietnamese Education', layout='wide', initial_sidebar_state="collapsed")
@@ -17,6 +17,9 @@ st.markdown("<h1 style='text-align: center;'>ViQAG: An Automate Question Answer
 
 # =====================================================================================================
 
+if 'output' not in st.session_state:
+    st.session_state.output = ''
+
 def file_selector(folder_path=r'./Resources/'):
     filenames = os.listdir(folder_path)
     return filenames
@@ -39,8 +42,21 @@ 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(context.tolist())
-
+    return '\n'.join([str(item) for item in context.tolist()])
+
+def generateQA(context, model_path = 'shnl/vit5-vinewsqa-qg-ae'):
+    unique_qa_pairs = set()
+    model = TransformersQG(model=model_path, max_length=512)
+    output = model.generate_qa(context)
+    qa_pairs = ''
+    for item in output:
+        question, answer = item
+        if (question, answer) not in unique_qa_pairs:
+            qa_pairs += f'question: {question} \nanswer: {answer} [SEP] '
+            unique_qa_pairs.add((question, answer))
+    qa = '\n\n'.join(qa_pairs.split(' [SEP] '))
+    return qa
+    
 # =====================================================================================================
 
 col_1, col_2, col_3, col_4 = st.columns(spec=[1, 1, 3, 4])
@@ -71,16 +87,24 @@ col_21.checkbox(label='Multiple Choice Question Generation (Coming soon)', disab
 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'>Options:</h8>", unsafe_allow_html=True)
-btn_show_answer = col_21.toggle(label='Show the answers', disabled=True)
+btn_show_answer = col_21.toggle(label='Show the answers', disabled=False)
 
 btn_generate = col_21.button(label='Generate questions', use_container_width=True)
 
-if btn_generate:
-    with st.spinner(text='Generating...'):
-        time.sleep(5)
-    
-    st.markdown("<h8 style='text-align: left; font-weight: normal'>Your questions and answers has been generated:</h8>", unsafe_allow_html=True)
-    output = '''question: <question_1>, answer: <Có cái njt :v Chưa cập nhật xong :v> [SEP] \nquestion: <question_2>, answer: <answer_2> [SEP] \nquestion: <question_3>, answer: <answer_3> [SEP] \nquestion: <question_4>, answer: <answer_4> [SEP] \nquestion: <question_5>, answer: <answer_5> [SEP]\nquestion: <question_6>, answer: <answer_6> [SEP]\nquestion: <question_7>, answer: <answer_7> [SEP]\nquestion: <question_8>, answer: <answer_8> [SEP]\nquestion: <question_9>, answer: <answer_9> [SEP]\nquestion: <question_10>, answer: <answer_10> [SEP]'''
-    st.code(body=output, language='wiki')
-
-    
+if btn_generate == True:
+    with st.spinner(text='Generating 10 pairs questions-answers. 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')
+    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)
+        output_no_answer = st.session_state.output.split(' [SEP] ')[0].split(', answer: ')[0].replace('question: ', '')
+        st.code(body=output_no_answer, language='wiki')
+    else:
+        pass