Update solver.py

solver.py

@@ -1,10 +1,10 @@
 import sympy as sp
 from sympy.parsing.sympy_parser import parse_expr, standard_transformations, implicit_multiplication_application
 from sympy.solvers import solve
-from sympy import integrate, diff, simplify, expand, log, exp, sin, cos, tan, asin, acos, atan, Symbol, factorial, laplace_transform
+from sympy import integrate, diff, latex,simplify, expand, log, exp, sin, cos, tan, asin, acos, atan, Symbol, factorial, laplace_transform
 import re
 def format_expression(expr):
-    str_expr = str(expr)
+    latex_expr = latex(expr)
     replacements = {
         '**': '^',  # Power notation
         '*x': 'x',  # Remove unnecessary multiplication signs
@@ -41,12 +41,9 @@ def format_expression(expr):
         'Im': 'ℑ'  # Imaginary part
     }
     for old, new in replacements.items():
-        str_expr = str_expr.replace(old, new)
-    return str_expr
+        latex_expr = latex_expr.replace(old, new)
 
-    for old, new in replacements.items():
-        str_expr = str_expr.replace(old, new)
-    return str_expr
+    return f"$$ {latex_expr} $$"
 
 def preprocess_equation(equation_str):
     """Convert user-friendly equation format to SymPy format."""