Update app.py

app.py

@@ -5,6 +5,8 @@ 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
+
 logging.basicConfig(level=logging.DEBUG)
 
 
@@ -183,90 +185,18 @@ def handle_answer(answer, current_q):
     if st.session_state.current_question_index >= 10:
         st.session_state.current_step = 'review'
 
-def format_question(question: str) -> str:
-    """문제 텍스트를 LaTeX 형식으로 변환"""
-    # 색상이 있는 텍스트 처리
-    colored_text_pattern = r'\\textcolor{([^}]+)}{([^}]+)}'
-    
-    def process_colored_text(match):
-        color = match.group(1)
-        text = match.group(2)
-        return f'\\textcolor{{{color}}}{{{format_latex_expression(text)}}}'
-    
-    question = re.sub(colored_text_pattern, process_colored_text, question)
-    return format_latex_expression(question)
-
-def format_answer_choice(choice: str) -> str:
-    """선택지 텍스트를 LaTeX 형식으로 변환"""
-    # 달러 기호가 포함된 선택지 특별 처리
-    if '$' in choice:
-        # 달러 기호를 LaTeX 명령어로 변환
-        choice = choice.replace('$', '\\$')
-    return format_latex_expression(choice)
+# 전역 LaTeX 포맷터 인스턴스 생성
+latex_formatter = LatexFormatter()
 
 def display_math_content(content: str):
     """수학 내용을 화면에 표시"""
-    formatted_content = format_latex_expression(content)
-    # LaTeX 렌더링을 위한 추가 설정
+    formatted_content = latex_formatter.format_expression(content)
     st.markdown(formatted_content, unsafe_allow_html=True)
 
-def format_latex_expression(text: str) -> str:
-    """복잡한 LaTeX 수식과 특수 기호를 처리하는 함수"""
-    import re
-    
-    # LaTeX 특수 명령어 매핑
-    latex_commands = {
-        r'\left': r'\\left',
-        r'\right': r'\\right',
-        r'\bigcirc': r'\\bigcirc',
-        r'\square': r'\\square',
-        r'\quad': r'\\quad'
-    }
-    
-    # 1. 이미 존재하는 LaTeX 수식 보존
-    latex_parts = []
-    def save_latex(match):
-        latex_parts.append(match.group(0))
-        return f"LATEX_{len(latex_parts)-1}_PLACEHOLDER"
-    
-    text = re.sub(r'\$\$.*?\$\$', save_latex, text)
-    
-    # 2. 특수 명령어 처리
-    for cmd, latex_cmd in latex_commands.items():
-        text = text.replace(cmd, latex_cmd)
-    
-    # 3. 붙어있는 단어 분리
-    text = re.sub(r'([a-z])([A-Z])', r'\1 \2', text)
-    text = re.sub(r'([A-Za-z])(\d)', r'\1 \2', text)
-    text = re.sub(r'(\d)([A-Za-z])', r'\1 \2', text)
-    
-    # 4. 문장 분석
-    sentences = text.split('$$')
-    formatted_sentences = []
-    
-    for i, sentence in enumerate(sentences):
-        if i % 2 == 0:  # 일반 텍스트
-            # 수식이 아닌 부분의 특수 문자 처리
-            for cmd, latex_cmd in latex_commands.items():
-                if cmd in sentence:
-                    sentence = f"$${sentence}$$"
-                    break
-            formatted_sentences.append(sentence)
-        else:  # 수식
-            formatted_sentences.append(f"$${sentence}$$")
-    
-    text = ''.join(formatted_sentences)
-    
-    # 5. LaTeX 수식 복원
-    for i, latex in enumerate(latex_parts):
-        text = text.replace(f"LATEX_{i}_PLACEHOLDER", latex)
-    
-    # 6. 마지막 정리
-    text = text.replace('\\\\', '\\')  # 중복된 백슬래시 제거
-    text = re.sub(r'\s+', ' ', text)  # 여러 개의 공백을 하나로
-    
-    return text
-    
+def format_answer_choice(choice: str) -> str:
+    """선택지 LaTeX 포맷팅"""
+    return latex_formatter.format_expression(choice)
+
 def main():
     """메인 애플리케이션 로직"""
     st.title("MisconcepTutor")