Update app.py

app.py

@@ -107,16 +107,17 @@ def generate_similar_question(wrong_q, misconception_id, generator):
         }
         
         logger.info(f"Prepared input data: {input_data}")
-        
-        # 유사 문제 생성 호출
-        generated_q, _ = generator.generate_similar_question_with_text(
-            construct_name=input_data['construct_name'],
-            subject_name=input_data['subject_name'],
-            question_text=input_data['question_text'],
-            correct_answer_text=input_data['correct_answer_text'],
-            wrong_answer_text=input_data['wrong_answer_text'],
-            misconception_id=input_data['misconception_id']
-        )
+
+        with st.spinner("📝 유사 문제를 생성하고 있습니다..."):
+            # 유사 문제 생성 호출
+            generated_q, _ = generator.generate_similar_question_with_text(
+                construct_name=input_data['construct_name'],
+                subject_name=input_data['subject_name'],
+                question_text=input_data['question_text'],
+                correct_answer_text=input_data['correct_answer_text'],
+                wrong_answer_text=input_data['wrong_answer_text'],
+                misconception_id=input_data['misconception_id']
+            )
         # Before module 3 integration code
         # if generated_q:
         #     return {
@@ -129,7 +130,8 @@ def generate_similar_question(wrong_q, misconception_id, generator):
         # integrating module3 
         if generated_q:
             verifier = load_answer_verifier()
-            with st.spinner("답안을 검증하고 있습니다..."):
+            with st.status("🤔 AI가 문제를 검토하고 있습니다..."):
+                st.write("답안의 정확성을 검증하고 있습니다...")
                 # 답안 검증 시도
                 verified_answer = verifier.verify_answer(
                     question=generated_q.question,
@@ -138,22 +140,30 @@ def generate_similar_question(wrong_q, misconception_id, generator):
                 
                 if verified_answer:
                     logger.info(f"Answer verified: {verified_answer}")
-                    # 검증된 답안으로 업데이트
-                    return {
+                    st.write("✅ 검증 완료!")
+                    # 세션 스테이트에 정답 저장
+                    result = {
                         'question': generated_q.question,
                         'choices': generated_q.choices,
                         '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")
-                    return {
+                    st.write("⚠️ 검증에 실패했습니다. 원본 답안을 사용합니다.")
+                    result = {
                         'question': generated_q.question,
                         'choices': generated_q.choices,
                         'correct': generated_q.correct_answer,
                         'explanation': generated_q.explanation
-                    }        
+                    }
+                    # 정답을 세션 스테이트에 저장
+                    st.session_state['current_similar_question_answer'] = generated_q.correct_answer
+                    return result
     except Exception as e:
         logger.error(f"Error in generate_similar_question: {str(e)}")
         st.error(f"문제 생성 중 오류가 발생했습니다: {str(e)}")