Update solver.py

solver.py

@@ -5,7 +5,15 @@ from sympy import integrate, diff, latex,simplify, expand, log, exp, sin, cos, t
 import re
 def format_expression(expr):
     latex_expr = latex(expr)
-    replacements = {
+    try:
+        # Convert string to float and handle sqrt
+        if isinstance(expr, sp.Number):
+            float_val = float(expr)
+            if float_val.is_integer():
+                return str(int(float_val))
+            return str(float_val)
+        # For other expressions
+        replacements = {
         '**': '^',  # Power notation
         '*x': 'x',  # Remove unnecessary multiplication signs
         '*(': '(',  # Remove multiplication before parentheses
@@ -40,10 +48,12 @@ def format_expression(expr):
         'Re': 'ℜ',  # Real part
         'Im': 'ℑ'  # Imaginary part
     }
-    for old, new in replacements.items():
-        latex_expr = latex_expr.replace(old, new)
-
-    return f"$$ {latex_expr} $$"
+        for old, new in replacements.items():
+            latex_expr = latex_expr.replace(old, new)
+        return latex_expr
+    except:
+        return latex_expr
+   
 
 def preprocess_equation(equation_str):
     """Convert user-friendly equation format to SymPy format."""