Update app.py

app.py

@@ -188,45 +188,79 @@ 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}}$$'
+    
+    # 기존 LaTeX 표현 보존
+    text = re.sub(r'\\begin{tabular}(.*?)\\end{tabular}', format_table, text, flags=re.DOTALL)
+    
+    # 기본 수학 기호 변환
     replacements = {
-        r'\÷': r'\div',
-        r'\×': r'\times',
-        r'\-': r'-',
-        r'\+': r'+',
-        r'=': r'=',
+        '÷': '\\div',
+        '×': '\\times',
+        '*': '\\cdot',  # 곱셈 점
+        '≠': '\\neq',
+        '≤': '\\leq',
+        '≥': '\\geq',
+        '→': '\\rightarrow',
+        '←': '\\leftarrow',
+        '±': '\\pm',
+        '∞': '\\infty',
+        '√': '\\sqrt',
     }
     
-    # 분수 패턴 찾기 (예: "1/2" → "\frac{1}{2}")
+    # 분수 패턴 찾기 및 변환
     text = re.sub(r'(\d+)/(\d+)', r'\\frac{\1}{\2}', text)
     
-    # 괄호 처리
-    text = re.sub(r'\(([^)]+)\)', r'\(\1\)', 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로 감싸기
+    # 괄호 안의 수식을 LaTeX로 처리
     def wrap_math(match):
         expr = match.group(1)
-        return f'${expr}$'
+        # 이미 LaTeX 달러 기호가 있는지 확인
+        if not expr.startswith('$') and not expr.endswith('$'):
+            return f'$${expr}$$'
+        return expr
+    
+    # 괄호로 둘러싸인 수식 찾아서 LaTeX로 변환
+    text = re.sub(r'\((.*?)\)', wrap_math, text)
     
-    # 수식 패턴 찾아서 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
     
     return text
 
 def display_math_question(question: str):
     """수학 문제를 LaTeX 형식으로 표시"""
     formatted_question = format_math_expression(question)
-    st.markdown(formatted_question)
+    st.markdown(formatted_question, unsafe_allow_html=True)
+
+def format_math_choices(choices: dict) -> dict:
+    """선택지의 수학 표현식을 LaTeX 형식으로 변환"""
+    return {k: format_math_expression(v) for k, v in choices.items()}
 
 def display_math_answer(answer: str):
     """수학 답안을 LaTeX 형식으로 표시"""
     formatted_answer = format_math_expression(answer)
     # 괄호 안의 수식만 LaTeX로 처리
     st.markdown(formatted_answer)
+
+    
 def main():
     """메인 애플리케이션 로직"""
     st.title("MisconcepTutor")