Update app.py

app.py

@@ -184,25 +184,44 @@ def handle_answer(answer, current_q):
         st.session_state.current_step = 'review'
 
 # 수식 표현 
+def clean_text(text: str) -> str:
+    """텍스트 전처리: 불필요한 중괄호 제거 및 기본 클리닝"""
+    import re
+    
+    # LaTeX 수식 부분을 임시 저장
+    latex_parts = []
+    def save_latex(match):
+        latex_parts.append(match.group(0))
+        return f"LATEX_{len(latex_parts)-1}_PLACEHOLDER"
+    
+    # LaTeX 수식 부분을 임시로 치환
+    text = re.sub(r'\$\$.*?\$\$', save_latex, text)
+    
+    # 각 문자를 감싸고 있는 중괄호 제거
+    text = re.sub(r'\{(\w)\}', r'\1', text)
+    
+    # LaTeX 수식 부분 복원
+    for i, latex in enumerate(latex_parts):
+        text = text.replace(f"LATEX_{i}_PLACEHOLDER", latex)
+    
+    return text
+
 def format_math_expression(text: str) -> str:
     """수학 표현식을 LaTeX 형식으로 변환"""
     import re
     
-    # LaTeX 표 처리
-    def format_table(match):
-        table_content = match.group(1)
-        # 표 내용을 LaTeX 형식으로 정리
-        formatted_table = table_content.replace('\\\\', '\n')
-        return f'$$\\begin{{array}}{{|c|c|}} \\hline {formatted_table} \\hline \\end{{array}}$$'
+    # 텍스트 전처리
+    text = clean_text(text)
     
-    # 기존 LaTeX 표현 보존
-    text = re.sub(r'\\begin{tabular}(.*?)\\end{tabular}', format_table, text, flags=re.DOTALL)
+    # 기존 LaTeX 표현식은 보존
+    if text.startswith('$$') and text.endswith('$$'):
+        return text
     
     # 기본 수학 기호 변환
     replacements = {
         '÷': '\\div',
         '×': '\\times',
-        '*': '\\cdot',  # 곱셈 점
+        '*': '\\cdot',
         '≠': '\\neq',
         '≤': '\\leq',
         '≥': '\\geq',
@@ -216,35 +235,24 @@ def format_math_expression(text: str) -> str:
     # 분수 패턴 찾기 및 변환
     text = re.sub(r'(\d+)/(\d+)', r'\\frac{\1}{\2}', text)
     
-    # y=mx+b 형태의 방정식 처리
-    text = re.sub(r'y\s*=\s*([-\d.]+)x\s*([+-]\s*\d+)?', r'y = \1x\2', text)
-    
-    # 표현식 내의 변수 이탤릭체로 변환
-    text = re.sub(r'([a-zA-Z])(\d*)', r'{\1}\2', text)
-    
     # 수학 기호 변환
     for old, new in replacements.items():
         text = text.replace(old, new)
     
-    # 괄호 안의 수식을 LaTeX로 처리
-    def wrap_math(match):
-        expr = match.group(1)
-        # 이미 LaTeX 달러 기호가 있는지 확인
-        if not expr.startswith('$') and not expr.endswith('$'):
-            return f'$${expr}$$'
-        return expr
-    
-    # 괄호로 둘러싸인 수식 찾아서 LaTeX로 변환
-    text = re.sub(r'\((.*?)\)', wrap_math, text)
-    
-    # 이미 LaTeX로 처리된 부분은 건너뛰고, 나머지 수식을 LaTeX로 처리
-    def add_math_delimiters(text):
-        if '$$' not in text and '$' not in text:
-            return f'$${text}$$'
-        return text
+    # 지수 표현 처리 (예: x^2)
+    text = re.sub(r'(\d+)\^(\d+)', r'$$\1^{\2}$$', text)
     
     return text
 
+def display_math_question(question: str):
+    """수학 문제를 LaTeX 형식으로 표시"""
+    formatted_question = format_math_expression(question)
+    st.markdown(formatted_question)
+
+def format_math_choices(choices: dict) -> dict:
+    """선택지의 수학 표현식을 LaTeX 형식으로 변환"""
+    return {k: format_math_expression(v) for k, v in choices.items()}
+
 def display_math_question(question: str):
     """수학 문제를 LaTeX 형식으로 표시"""
     formatted_question = format_math_expression(question)