complex roots

app.py

@@ -26,18 +26,16 @@ def quadratic_roots(a:int, b:int, c:int)-> str:
     # Calculate the discriminant
     discriminant = b**2 - 4*a*c
 
-    if discriminant > 0:
-        # Two distinct real roots
+    if discriminant >= 0:
+        # Real roots
         root1 = (-b + math.sqrt(discriminant)) / (2 * a)
         root2 = (-b - math.sqrt(discriminant)) / (2 * a)
-        return str(root1), str(root2)
-    elif discriminant == 0:
-        # One real root (double root)
-        root = -b / (2 * a)
-        return str(root), str(root)
     else:
-        # No real roots
-        return "No real roots exist."
+        # Complex roots
+        root1 = (-b + cmath.sqrt(discriminant)) / (2 * a)
+        root2 = (-b - cmath.sqrt(discriminant)) / (2 * a)
+
+    return str(root1), str(root2)
 
 
 @tool