Update app.py

app.py

@@ -5,7 +5,6 @@ from src.SecondModule.module2 import SimilarQuestionGenerator
 from src.ThirdModule.module3 import AnswerVerifier
 import logging
 from typing import Optional, Tuple
-#from latex_formatter import LatexFormatter  # LaTeX 포맷터 import
 from pylatexenc.latex2text import LatexNodes2Text
 import re
 
@@ -339,15 +338,20 @@ def update_similar_question_display(new_question, i, answered=False):
     """Display similar question and its options"""
     display_math_content(new_question['question'])
     
-    col1, col2 = st.columns(2)
-    for j, (col, opt) in enumerate([(col1, 'A'), (col1, 'C'), (col2, 'B'), (col2, 'D')]):
-        with col:
-            st.markdown(f"### {opt})")
-            display_option_content(new_question['choices'][opt])
-            if not answered:
-                if st.button("선택", key=f"similar_{opt}_{i}"):
-                    handle_similar_answer(opt, i)
-                    st.rerun()
+    # Display options
+    for opt in ['A', 'B', 'C', 'D']:
+        with st.container():
+            col1, col2 = st.columns([1, 11])
+            with col1:
+                if st.button(f"{opt}.", key=f"sim_btn_{opt}_{i}", help="Click to select"):
+                    if not answered:
+                        st.session_state[f"similar_question_answered_{i}"] = True
+                        st.session_state[f"selected_answer_{i}"] = opt
+                        correct_answer = st.session_state.get('current_similar_question_answer')
+                        st.session_state[f"is_correct_{i}"] = (opt == correct_answer)
+                        st.rerun()
+            with col2:
+                display_option_content(new_question['choices'][opt])
 
 def main():
     """메인 애플리케이션 로직"""