Update app.py

app.py

@@ -9,20 +9,29 @@ client = InferenceClient()
 
 
 def preprocess_latex(content):
-    """Preprocess the AI response to format LaTeX properly."""
+    """
+    Preprocess the AI response using sympy to properly format mathematical expressions.
+    """
     lines = content.split("\n")  # Split response into lines
     processed_lines = []
-    
+
     for line in lines:
-        if "[" in line and "]" in line:  # Detect math expressions in brackets
+        if "[" in line and "]" in line:  # Detect mathematical expressions in brackets
             while "[" in line and "]" in line:
                 start = line.index("[")
                 end = line.index("]") + 1
-                math_expr = line[start:end].strip("[]").strip()
-                line = line[:start] + f"$$ {math_expr} $$" + line[end:]
+                math_expr = line[start + 1 : end - 1].strip()
+                try:
+                    # Parse the math expression and convert to LaTeX
+                    sympy_expr = sympify(math_expr)
+                    latex_expr = latex(sympy_expr)
+                    line = line[:start] + f"$$ {latex_expr} $$" + line[end:]
+                except Exception:
+                    # Fallback to raw math if parsing fails
+                    line = line[:start] + f"$$ {math_expr} $$" + line[end:]
         processed_lines.append(line)
 
-    return "\n".join(processed_lines)  # Join lines back into a single string
+    return "\n".join(processed_lines)