|
import streamlit as st |
|
import random |
|
from pipeline import Pipeline |
|
|
|
st.header("Automation of Question and Answer Generation") |
|
st.markdown( |
|
"I built this project based on this [paper](https://www.sciencedirect.com/science/article/pii/S0957417422014014#s0015), " |
|
"where they created End-to-End generation of Multiple-Choice questions using Text-to-Text transfer Transformer models (T5).\n" |
|
"This research focuses on using Transformer-based language models to automate the generation of multiple-choice questions (MCQs), " |
|
"with the aim of assisting or assisting educators in the process of creating reading comprehension (RC) assessments. " |
|
"This is relevant and timely as teachers can invest less time doing routine work and share more time with their students, " |
|
"thus building an engaging experience for face-to-face classroom interaction. " |
|
"This study addresses the issue of creating multiple-choice questionnaires from 3 viewpoints: QG, QA, and distractor generation (DG). " |
|
"An end-to-end pipeline for generating multiple-choice questions is proposed, based on a pre-trained T5 language model." |
|
) |
|
|
|
|
|
|
|
st.sidebar.info( |
|
"Note: The number of questions generated depends on the length of the context. " |
|
"You may find that the number of QA pairs does not match the number you want." |
|
) |
|
|
|
with st.sidebar: |
|
if "num_qa" not in st.session_state: |
|
st.session_state.num_qa = 5 |
|
|
|
def on_change(): |
|
st.session_state.num_qa = num_qa |
|
|
|
num_qa = st.slider("Select Number of QA questions", min_value=1, max_value=10, value=1, step=1, on_change=on_change) |
|
|
|
if 'context' not in st.session_state: |
|
st.session_state.context = "" |
|
st_text_area = st.text_area('Context to generate the QA', value=st.session_state.context, height=400) |
|
|
|
def generate_qa(): |
|
st.session_state.context = st_text_area |
|
mcq_generator = Pipeline() |
|
generator = mcq_generator.generate_mcqs(st_text_area, num_qa) |
|
st.session_state.generator = generator |
|
|
|
|
|
st_generate_button = st.button('Generate', on_click=generate_qa) |
|
|
|
|
|
if hasattr(st.session_state, 'generator') and len(st.session_state.generator) > 0: |
|
st.subheader("Generated MCQs") |
|
for i, question in enumerate(st.session_state.generator, start=1): |
|
correct_answer = [question.answerText] |
|
distractors_subset = question.distractors[:3] |
|
options = correct_answer + distractors_subset |
|
|
|
|
|
random.shuffle(options) |
|
|
|
options_with_labels = [{'label': chr(ord('A') + j), 'text': option} for j, option in enumerate(options)] |
|
|
|
st.write(f'Number {i}: {question.questionText}') |
|
for option in options_with_labels: |
|
if option["text"] == correct_answer[0]: |
|
st.write(f'<span style="color:green;">{option["label"]}. {option["text"]}</span>', unsafe_allow_html=True) |
|
else: |
|
st.write(f'{option["label"]}. {option["text"]}') |
|
st.write('-------------------') |
|
|
|
|
