from testpack import generate_questionnaire
import streamlit as st
from read_files import read_docx, read_pdf
sts = st.session_state
          
def callback():
    with st.spinner("문제를 생성 중입니다."):
        sts.questionnaire = generate_questionnaire(sts.model, sts.input_text, sts.language, sts.num_questions)
    sts.current_step = "B"
    
def callback2():
    sts.current_step = "C"

if "current_step" not in sts:
    sts.current_step = "A"
if "input_text" not in sts:
    sts.input_text = ""

st.title("모의고사 자동제작 서비스")

if sts.current_step == "A":
    format = st.radio("파일 유형 선택", ["직접 입력", ".pdf", ".docx"], index=0, key="format")
    if format == "직접 입력":
        sts.input_text = st.text_area("내용 입력", sts.input_text)
    elif format == ".pdf":
        pdf_loc = st.file_uploader("최대 200MB까지 업로드할 수 있습니다")
        if pdf_loc:
            sts.input_text = read_pdf(pdf_loc)
    elif format == ".docx":
        docx_loc = st.file_uploader("최대 200MB까지 업로드할 수 있습니다")
        if docx_loc:
            sts.input_text = read_docx(docx_loc)
                
    with st.form("maker"):
        model = st.radio("모델 선택 (GPT-4를 권장합니다.)", ["GPT-4", "GPT-3.5"], index=0, key="model")
        language = st.radio("언어 선택 (영어가 더 정확합니다.)", ["한국어", "English"], index=0, key="language")
        num_questions = st.slider("생성할 문제 수 설정 (인공지능의 한계로 동일한 문제가 여러 개 생성될 수 있음)", min_value=1, max_value=20, value=5, key="num_questions")
        if sts.input_text:
            st.form_submit_button("모의고사 생성하기", on_click=callback)
        else:
            st.form_submit_button(":red[내용을 입력하거나 파일을 업로드해 주세요.]", disabled=True)

if sts.current_step == "B":
    with st.form("main_quiz"):            
        for q_no, question in enumerate(sts.questionnaire):
            labels = question["selection"]
            st.radio(question["question"], range(len(labels)), index=0, format_func=labels.__getitem__, key=f"chosen_{q_no}")
        st.form_submit_button("제출하기", on_click=callback2)
            
if sts.current_step == "C":
    score_list = [0] * len(sts.questionnaire)
    for q_no, question in enumerate(sts.questionnaire):
        selection_list = []
        for s_no, sel in enumerate(question["selection"]):                    
            if s_no == sts[f"chosen_{q_no}"]:
                if s_no == question["correct"]:
                    score_list[q_no] = 1
                    selection_list.append(f":green[{sel}]")
                else:
                    selection_list.append(f":red[{sel}]")
            else:
                if s_no == question["correct"]:
                    selection_list.append(f":green[{sel}]")
                else:
                    selection_list.append(sel)
        st.radio(question["question"], selection_list, index=sts[f"chosen_{q_no}"], disabled=True)
        result = "✅" if score_list[q_no] == 1 else "❌"
        st.write(f"{result} {question['explanation']}") 
    st.write(f"{len(sts.questionnaire)}개 중 {sum(score_list)}개 맞추셨습니다. 다시 이용하고 싶으시면 새로고침해 주세요.")