Update app.py

app.py

@@ -238,14 +238,22 @@ def main():
                         st.info(f"Misconception ID: {int(misconception_id)}\n\n{misconception_text}")
                     else:
                         st.info("Misconception 정보가 없습니다.")
-                    # try:
-                    #     misconception_text = generator.get_misconception_text(misconception_id)
-                    #     st.info(f"Misconception ID: {int(misconception_id)}\n\n{misconception_text}")
-                    # except Exception as e:
-                    #     logger.error(f"Error in get_misconception_text: {e}")
-                    #     st.info("Misconception 정보가 없습니다.")
-                    #     return None, None
-                    
+
+                    # # 유사 문제 생성 버튼
+                    # if st.button(f"📚 유사 문제 풀기 #{i + 1}", key=f"retry_{i}"):
+                    #     with st.spinner("유사 문제를 생성하고 있습니다..."):
+                    #         logger.info(f"Generating similar question for misconception_id: {misconception_id}")
+                    #         new_question = generate_similar_question(wrong_q, misconception_id, generator)
+                    #         if new_question:
+                    #             st.write("### 🎯 유사 문제")
+                    #             st.write(new_question['question'])
+                    #             st.write("**보기:**")
+                    #             for choice, text in new_question['choices'].items():
+                    #                 st.write(f"{choice}) {text}")
+                    #             st.write("**✅ 정답:**", new_question['correct'])
+                    #             st.write("**📝 해설:**", new_question['explanation'])
+                    #         else:
+                    #             st.error("유사 문제를 생성할 수 없습니다.")
                     # 유사 문제 생성 버튼
                     if st.button(f"📚 유사 문제 풀기 #{i + 1}", key=f"retry_{i}"):
                         with st.spinner("유사 문제를 생성하고 있습니다..."):
@@ -254,13 +262,63 @@ def main():
                             if new_question:
                                 st.write("### 🎯 유사 문제")
                                 st.write(new_question['question'])
-                                st.write("**보기:**")
-                                for choice, text in new_question['choices'].items():
-                                    st.write(f"{choice}) {text}")
-                                st.write("**✅ 정답:**", new_question['correct'])
-                                st.write("**📝 해설:**", new_question['explanation'])
+                                
+                                # 정답을 세션 스테이트에 저장
+                                if f"similar_question_answered_{i}" not in st.session_state:
+                                    st.session_state[f"similar_question_answered_{i}"] = False
+                                if f"similar_question_correct_{i}" not in st.session_state:
+                                    st.session_state[f"similar_question_correct_{i}"] = new_question['correct']
+                                
+                                # 아직 답변하지 않은 경우에만 보기 버튼 표시
+                                if not st.session_state[f"similar_question_answered_{i}"]:
+                                    st.write("**보기:**")
+                                    col1, col2 = st.columns(2)
+                                    with col1:
+                                        if st.button(f"A) {new_question['choices']['A']}", key=f"similar_A_{i}"):
+                                            st.session_state[f"similar_question_answered_{i}"] = True
+                                            if 'A' == new_question['correct']:
+                                                st.success("✅ 정답입니다!")
+                                            else:
+                                                st.error("❌ 틀렸습니다.")
+                                            st.rerun()
+                                            
+                                        if st.button(f"C) {new_question['choices']['C']}", key=f"similar_C_{i}"):
+                                            st.session_state[f"similar_question_answered_{i}"] = True
+                                            if 'C' == new_question['correct']:
+                                                st.success("✅ 정답입니다!")
+                                            else:
+                                                st.error("❌ 틀렸습니다.")
+                                            st.rerun()
+                                            
+                                    with col2:
+                                        if st.button(f"B) {new_question['choices']['B']}", key=f"similar_B_{i}"):
+                                            st.session_state[f"similar_question_answered_{i}"] = True
+                                            if 'B' == new_question['correct']:
+                                                st.success("✅ 정답입니다!")
+                                            else:
+                                                st.error("❌ 틀렸습니다.")
+                                            st.rerun()
+                                            
+                                        if st.button(f"D) {new_question['choices']['D']}", key=f"similar_D_{i}"):
+                                            st.session_state[f"similar_question_answered_{i}"] = True
+                                            if 'D' == new_question['correct']:
+                                                st.success("✅ 정답입니다!")
+                                            else:
+                                                st.error("❌ 틀렸습니다.")
+                                            st.rerun()
+                                
+                                # 답변한 경우 결과 및 해설 표시
+                                if st.session_state[f"similar_question_answered_{i}"]:
+                                    st.write("---")
+                                    st.write("**📝 해설:**", new_question['explanation'])
+                                    # 다시 풀기 버튼
+                                    if st.button("🔄 다시 풀기", key=f"reset_{i}"):
+                                        st.session_state[f"similar_question_answered_{i}"] = False
+                                        st.rerun()
                             else:
                                 st.error("유사 문제를 생성할 수 없습니다.")
+
+
 if __name__ == "__main__":
     main()