Update app.py

app.py

@@ -254,71 +254,144 @@ def main():
                     #             st.write("**📝 해설:**", new_question['explanation'])
                     #         else:
                     #             st.error("유사 문제를 생성할 수 없습니다.")
-                    # 유사 문제 생성 버튼
+                    
+                    # # 유사 문제 생성 버튼
+                    # 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'])
+                                
+                    #             # 정답을 세션 스테이트에 저장
+                    #             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 st.button(f"📚 유사 문제 풀기 #{i + 1}", key=f"retry_{i}"):
+                        # 유사 문제 생성 상태를 세션에 저장
+                        st.session_state[f"show_similar_question_{i}"] = True
+                        st.session_state[f"similar_question_answered_{i}"] = False
+                        # 기존 답변 기록 초기화
+                        st.rerun()
+                    
+                    # 유사 문제가 생성된 상태인 경우
+                    if st.session_state.get(f"show_similar_question_{i}", False):
                         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'])
                                 
-                                # 정답을 세션 스테이트에 저장
-                                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']
+                                # 답변 상태 확인
+                                answered = st.session_state.get(f"similar_question_answered_{i}", False)
+                                selected_answer = st.session_state.get(f"selected_answer_{i}", None)
+                                
+                                # 보기 표시
+                                st.write("**보기:**")
+                                col1, col2 = st.columns(2)
                                 
-                                # 아직 답변하지 않은 경우에만 보기 버튼 표시
-                                if not st.session_state[f"similar_question_answered_{i}"]:
-                                    st.write("**보기:**")
-                                    col1, col2 = st.columns(2)
+                                # 답변하지 않은 경우에만 버튼 활성화
+                                if not answered:
                                     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()
-                                            
+                                        for option in ['A', 'C']:
+                                            if st.button(
+                                                f"{option}) {new_question['choices'][option]}", 
+                                                key=f"similar_{option}_{i}"
+                                            ):
+                                                st.session_state[f"similar_question_answered_{i}"] = True
+                                                st.session_state[f"selected_answer_{i}"] = option
+                                                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()
+                                        for option in ['B', 'D']:
+                                            if st.button(
+                                                f"{option}) {new_question['choices'][option]}", 
+                                                key=f"similar_{option}_{i}"
+                                            ):
+                                                st.session_state[f"similar_question_answered_{i}"] = True
+                                                st.session_state[f"selected_answer_{i}"] = option
+                                                st.rerun()
                                 
-                                # 답변한 경우 결과 및 해설 표시
-                                if st.session_state[f"similar_question_answered_{i}"]:
+                                # 답변한 경우 결과 표시
+                                if answered:
+                                    selected = st.session_state[f"selected_answer_{i}"]
+                                    if selected == new_question['correct']:
+                                        st.success("✅ 정답입니다!")
+                                    else:
+                                        st.error(f"❌ 틀렸습니다. 정답은 {new_question['correct']}입니다.")
+                                    
+                                    # 해설 표시
                                     st.write("---")
                                     st.write("**📝 해설:**", new_question['explanation'])
+                                    
                                     # 다시 풀기 버튼
                                     if st.button("🔄 다시 풀기", key=f"reset_{i}"):
                                         st.session_state[f"similar_question_answered_{i}"] = False
+                                        st.session_state[f"selected_answer_{i}"] = None
                                         st.rerun()
+                                
+                                # 문제 닫기 버튼
+                                if st.button("❌ 문제 닫기", key=f"close_{i}"):
+                                    st.session_state[f"show_similar_question_{i}"] = False
+                                    st.rerun()
                             else:
                                 st.error("유사 문제를 생성할 수 없습니다.")
-
-
+                                if st.button("❌ 닫기", key=f"close_error_{i}"):
+                                    st.session_state[f"show_similar_question_{i}"] = False
+                                    st.rerun()
 if __name__ == "__main__":
     main()