Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import requests
 import json
 import base64
+import re
 from PIL import Image
 from io import BytesIO
 
@@ -21,6 +22,7 @@ Rules you ALWAYS follow:
 6. For any calculation, write out every arithmetic operation explicitly before evaluating it. Never skip steps or compute mentally.
 7. For trigonometry, always state the formula, substitute values, then compute. Show sin/cos/tan values to 4 decimal places.
 8. Never use LaTeX notation. Use plain text only. Write fractions as a/b, exponents as x^2, and put each equation on its own line.
+
 Always format your response exactly like this:
 📘 **Problem:** [restate the problem clearly]
 ---
@@ -38,6 +40,25 @@ Always format your response exactly like this:
 💡 [One encouraging sentence]"""
 
 
+def strip_latex(text):
+    text = re.sub(r'\$\$(.+?)\$\$', r'\1', text, flags=re.DOTALL)
+    text = re.sub(r'\$(.+?)\$', r'\1', text)
+    text = re.sub(r'\\frac\{(.+?)\}\{(.+?)\}', r'\1/\2', text)
+    text = re.sub(r'\\text\{(.+?)\}', r'\1', text)
+    text = text.replace(r'\cos', 'cos')
+    text = text.replace(r'\sin', 'sin')
+    text = text.replace(r'\tan', 'tan')
+    text = text.replace(r'\times', '×')
+    text = text.replace(r'\div', '÷')
+    text = text.replace(r'\approx', '≈')
+    text = text.replace(r'\cdot', '·')
+    text = text.replace(r'\left', '').replace(r'\right', '')
+    text = text.replace(r'\sqrt', 'sqrt')
+    text = text.replace(r'\pi', 'π')
+    text = re.sub(r'\{|\}', '', text)
+    return text
+
+
 def image_to_base64(pil_image):
     buffered = BytesIO()
     pil_image.save(buffered, format="PNG")
@@ -70,7 +91,6 @@ def solve_math(image_input, text_input, grade_level, user_key):
 
     try:
         if has_image:
-            # Use vision model to read the image directly
             img_b64 = image_to_base64(image_input)
             text_note = f"\nStudent's additional note: {text_input.strip()}" if has_text else ""
             user_content = [
@@ -133,7 +153,7 @@ def solve_math(image_input, text_input, grade_level, user_key):
                 token = json.loads(line)["choices"][0]["delta"].get("content", "")
                 if token:
                     output += token
-                    yield output
+                    yield strip_latex(output)
             except Exception:
                 continue