Streamlit.py

import json
import traceback
import pandas as pd
from src.mcqgen.utils import read_file, get_table_data
from src.mcqgen.logger import logging
from src.mcqgen.MCQGen import generate_evaluate_chain
import streamlit as st


# load JSON
with open("./res.json", "r") as file:
    RESPONSE_JSON = json.load(file)

st.title("MCQ Generator")

with st.form("user_inputs"):
    file = st.file_uploader("Upload a PDF or TXT file")

    mcq_count = st.number_input("Number of MCQs", min_value=3, max_value=25)

    sub = st.text_input("Subject Name", max_chars=20)

    tone = st.text_input("Tone of Questions", max_chars=10, placeholder="Simple")

    button = st.form_submit_button("Generate MCQs")

    if button and file is not None and mcq_count and sub and tone:
        with st.spinner("Generating"):
            try:
                text = read_file(file)

                response = generate_evaluate_chain(
                    {
                        "text": text,
                        "number": mcq_count,
                        "subject": sub,
                        "tone": tone,
                        "response_json": json.dumps(RESPONSE_JSON),
                    }
                )

            except Exception as e:
                traceback.print_exception(type(e), e, e.__traceback__)
                st.error("Error")

            else:
                if isinstance(response, dict):
                    # Extract the quiz data from the response
                    quiz = response.get("quiz", None)
                    if quiz is not None:
                        table_data = get_table_data(quiz)
                        if table_data is not None:
                            df = pd.DataFrame(table_data)
                            df.index = df.index + 1
                            st.table(df)
                            # Display the review in atext box as well
                            st.text_area(label="Review", value=response["review"])
                        else:
                            st.error("Error in the table data")

                    else:
                        st.write("response")