Update app.py

app.py

@@ -206,57 +206,60 @@ def clean_text(text: str) -> str:
     
     return text
 
-def format_math_expression(text: str) -> str:
-    """수학 표현식을 LaTeX 형식으로 변환"""
+def clean_text(text: str) -> str:
+    """텍스트 전처리: 불필요한 문자 제거 및 LaTeX 형식으로 변환"""
     import re
     
-    # 텍스트 전처리
-    text = clean_text(text)
-    
-    # 기존 LaTeX 표현식은 보존
-    if text.startswith('$$') and text.endswith('$$'):
-        return text
-    
-    # 기본 수학 기호 변환
-    replacements = {
-        '÷': '\\div',
-        '×': '\\times',
-        '*': '\\cdot',
-        '≠': '\\neq',
-        '≤': '\\leq',
-        '≥': '\\geq',
-        '→': '\\rightarrow',
-        '←': '\\leftarrow',
-        '±': '\\pm',
-        '∞': '\\infty',
-        '√': '\\sqrt',
+    # LaTeX 명령어를 위한 이스케이프 처리
+    latex_commands = {
+        r'\div': r'\\div',
+        r'\ldots': r'\\ldots',
+        r'\dots': r'\\ldots',
+        r'\cdot': r'\\cdot',
+        r'\times': r'\\times'
     }
     
-    # 분수 패턴 찾기 및 변환
-    text = re.sub(r'(\d+)/(\d+)', r'\\frac{\1}{\2}', text)
-    
-    # 수학 기호 변환
-    for old, new in replacements.items():
+    for old, new in latex_commands.items():
         text = text.replace(old, new)
     
-    # 지수 표현 처리 (예: x^2)
-    text = re.sub(r'(\d+)\^(\d+)', r'$$\1^{\2}$$', text)
+    # 괄호 안의 수식을 LaTeX로 변환
+    def convert_to_latex(match):
+        content = match.group(1)
+        # 이미 $로 둘러싸여 있는지 확인
+        if not content.strip().startswith('$') and not content.strip().endswith('$'):
+            return f'$({content})$'
+        return f'({content})'
+    
+    # 괄호 안의 내용을 LaTeX로 변환
+    text = re.sub(r'\((.*?)\)', convert_to_latex, text)
     
     return text
 
-def display_math_question(question: str):
-    """수학 문제를 LaTeX 형식으로 표시"""
-    formatted_question = format_math_expression(question)
-    st.markdown(formatted_question)
+def format_math_expression(text: str) -> str:
+    """수학 표현식을 LaTeX 형식으로 변환"""
+    # 기본 클리닝
+    text = clean_text(text)
+    
+    # LaTeX 수식 기호 추가
+    if not text.startswith('$') and not text.endswith('$'):
+        text = f'${text}$'
+    
+    return text
 
-def format_math_choices(choices: dict) -> dict:
-    """선택지의 수학 표현식을 LaTeX 형식으로 변환"""
-    return {k: format_math_expression(v) for k, v in choices.items()}
+def display_math_content(content: str, is_question: bool = True):
+    """수학 내용(문제 또는 답안) 표시"""
+    formatted_content = format_math_expression(content)
+    
+    if is_question:
+        st.markdown(formatted_content)
+    else:
+        # 선택지의 경우 추가 포맷팅 필요할 수 있음
+        return formatted_content
 
 def display_math_question(question: str):
     """수학 문제를 LaTeX 형식으로 표시"""
     formatted_question = format_math_expression(question)
-    st.markdown(formatted_question, unsafe_allow_html=True)
+    st.markdown(formatted_question)
 
 def format_math_choices(choices: dict) -> dict:
     """선택지의 수학 표현식을 LaTeX 형식으로 변환"""