File size: 3,061 Bytes
aeae383 fd5d5c2 aeae383 f52298a aeae383 f52298a aeae383 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
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
# generate qa button
st_generate_button = st.button('Generate', on_click=generate_qa)
# Display generated MCQs in Streamlit
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] # Assuming you want 3 distractors
options = correct_answer + distractors_subset
# Shuffle options
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('-------------------')
|