Update app.py

app.py

@@ -90,12 +90,22 @@ def generate_similar_question(wrong_q, misconception_id, generator):
         st.error("유사 문제 생성에 필요한 데이터 형식이 잘못되었습니다.")
         return None
         
-    # misconception_id가 유효한지 확인
-    if pd.isna(misconception_id):
-        logger.warning("misconception_id is NaN")
-        return None
-        
     try:
+        # misconception_id가 없거나 NaN인 경우 다른 misconception 사용
+        if pd.isna(misconception_id):
+            logger.info("Original misconception_id is NaN, trying to find alternative")
+            # 현재까지 나온 misconception들 중에서 선택
+            available_misconceptions = [m for m in st.session_state.misconceptions if not pd.isna(m)]
+            
+            if available_misconceptions:
+                # 가장 최근에 나온 misconception 선택
+                misconception_id = available_misconceptions[-1]
+                logger.info(f"Using alternative misconception_id: {misconception_id}")
+            else:
+                # 기본 misconception ID 사용 (예: 가장 기본적인 misconception)
+                misconception_id = 2001  # 적절한 기본값으로 수정 필요
+                logger.info(f"Using default misconception_id: {misconception_id}")
+
         # 데이터 준비 (튜플 변환 방지)
         input_data = {
             'construct_name': str(wrong_q.get('ConstructName', '')),
@@ -119,12 +129,10 @@ def generate_similar_question(wrong_q, misconception_id, generator):
                 misconception_id=input_data['misconception_id']
             )
 
-        # integrating module3 
         if generated_q:
             verifier = load_answer_verifier()
             with st.status("🤔 AI가 문제를 검토하고 있습니다..."):
                 st.write("답안의 정확성을 검증하고 있습니다...")
-                # 답안 검증 시도
                 verified_answer = verifier.verify_answer(
                     question=generated_q.question,
                     choices=generated_q.choices
@@ -133,18 +141,15 @@ def generate_similar_question(wrong_q, misconception_id, generator):
                 if verified_answer:
                     logger.info(f"Answer verified: {verified_answer}")
                     st.write("✅ 검증 완료!")
-                    # 세션 스테이트에 정답 저장
                     result = {
                         'question': generated_q.question,
                         'choices': generated_q.choices,
-                        'correct': verified_answer,  # 검증된 답안 사용
+                        'correct': verified_answer,
                         'explanation': generated_q.explanation
                     }
-                    # 정답을 세션 스테이트에 저장
                     st.session_state['current_similar_question_answer'] = verified_answer
                     return result
                 else:
-                    # 검증 실패 시 원본 답안 사용
                     logger.warning("Answer verification failed, using original answer")
                     st.write("⚠️ 검증에 실패했습니다. 원본 답안을 사용합니다.")
                     result = {
@@ -153,7 +158,6 @@ def generate_similar_question(wrong_q, misconception_id, generator):
                         'correct': generated_q.correct_answer,
                         'explanation': generated_q.explanation
                     }
-                    # 정답을 세션 스테이트에 저장
                     st.session_state['current_similar_question_answer'] = generated_q.correct_answer
                     return result