Update solver.py

solver.py

@@ -4,12 +4,8 @@ from sympy.solvers import solve
 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):
-    try:
-        # Handle numeric evaluation first
-        if isinstance(expr, (sp.Integer, sp.Float)):
-            return str(float(expr))
-        latex_expr = latex(expr)
-        replacements = {
+    latex_expr = latex(expr)
+    replacements = {
         '**': '^',  # Power notation
         '*x': 'x',  # Remove unnecessary multiplication signs
         '*(': '(',  # Remove multiplication before parentheses
@@ -44,12 +40,10 @@ def format_expression(expr):
         'Re': 'ℜ',  # Real part
         'Im': 'ℑ'  # Imaginary part
     }
-        for old, new in replacements.items():
-            latex_expr = latex_expr.replace(old, new)
-        return latex_expr
-    except:
-        return latex_expr
-   
+    for old, new in replacements.items():
+        latex_expr = latex_expr.replace(old, new)
+
+    return f"$$ {latex_expr} $$"
 
 def preprocess_equation(equation_str):
     """Convert user-friendly equation format to SymPy format."""
@@ -118,6 +112,13 @@ def process_expression(expr_str):
             expr = parse_expr(expr_to_diff, transformations=(standard_transformations + (implicit_multiplication_application,)))
             result = diff(expr, x)
             return f"d/dx({format_expression(expr)}) = {format_expression(result)}"
+        
+        elif 'sqrt' in processed_expr:  # Handle square roots
+            expr = parse_expr(processed_expr, transformations=(standard_transformations + (implicit_multiplication_application,)))
+            result = sp.sqrt(expr)
+            positive_root = result
+            negative_root = -result
+            return f"√{format_expression(expr)} = ±{format_expression(positive_root)}"
 
         elif 'factorial' in processed_expr:  # Factorial case
             expr = parse_expr(processed_expr, transformations=(standard_transformations + (implicit_multiplication_application,)))