app.py

@st.cache_resource
def load_answer_verifier():
    """답안 검증 모델 로드"""
    from src.ThirdModule.module3 import AnswerVerifier
    return AnswerVerifier()

def generate_similar_question(wrong_q, misconception_id, generator):
    """유사 문제 생성 및 검증"""
    logger.info(f"Generating similar question for misconception_id: {misconception_id}")
    
    # 입력 데이터 유효성 검사
    if not isinstance(wrong_q, dict):
        logger.error(f"Invalid wrong_q type: {type(wrong_q)}")
        st.error("유사 문제 생성에 필요한 데이터 형식이 잘못되었습니다.")
        return None
    
    # misconception_id가 유효한지 확인
    if pd.isna(misconception_id):
        logger.warning("misconception_id is NaN")
        return None
    
    try:
        # 데이터 준비
        input_data = {
            'construct_name': str(wrong_q.get('ConstructName', '')),
            'subject_name': str(wrong_q.get('SubjectName', '')),
            'question_text': str(wrong_q.get('QuestionText', '')),
            'correct_answer_text': str(wrong_q.get(f'Answer{wrong_q["CorrectAnswer"]}Text', '')),
            'wrong_answer_text': str(wrong_q.get(f'Answer{st.session_state.selected_wrong_answer}Text', '')),
            'misconception_id': int(misconception_id)
        }
        
        # 유사 문제 생성
        generated_q, _ = generator.generate_similar_question_with_text(**input_data)
        
        if generated_q:
            verifier = load_answer_verifier()
            with st.spinner("답안을 검증하고 있습니다..."):
                # 답안 검증 시도
                verified_answer = verifier.verify_answer(
                    question=generated_q.question,
                    choices=generated_q.choices
                )
                
                if verified_answer:
                    logger.info(f"Answer verified: {verified_answer}")
                    # 검증된 답안으로 업데이트
                    return {
                        'question': generated_q.question,
                        'choices': generated_q.choices,
                        'correct': verified_answer,  # 검증된 답안 사용
                        'explanation': generated_q.explanation
                    }
                else:
                    # 검증 실패 시 원본 답안 사용
                    logger.warning("Answer verification failed, using original answer")
                    return {
                        'question': generated_q.question,
                        'choices': generated_q.choices,
                        'correct': generated_q.correct_answer,
                        'explanation': generated_q.explanation
                    }
    except Exception as e:
        logger.error(f"Error in generate_similar_question: {str(e)}")
        st.error(f"문제 생성 중 오류가 발생했습니다: {str(e)}")
        return None

    return None