Update app.py

app.py

@@ -10,18 +10,34 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def get_roots(a:int, b:int, c:int)-> (int, int): 
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that performs the quadratic formula to get the roots of a polynomial
+def get_roots(a: float, b: float, c: float) -> Tuple[float, float]:
+    """Compute the real roots of a quadratic equation ax^2 + bx + c = 0.
+
     Args:
-        a: the first argument
-        b: the second argument
-        c: the third argument
-    """   
-    first_root = ((-1 * b) + math.sqrt(math.pow(b, 2) - (4 * a * c))) / (2 * a)
-    second_root = ((-1 * b) - math.sqrt(math.pow(b, 2) - (4 * a * c))) / (2 * a)
-    
-    return (first_root, second_root)
+        a: Quadratic coefficient (must be non-zero).
+        b: Linear coefficient.
+        c: Constant term.
+
+    Returns:
+        A tuple (x1, x2) containing the two real roots.
+
+    Raises:
+        ValueError: If a == 0 or the discriminant is negative (no real roots).
+    """
+    if a == 0:
+        raise ValueError("Coefficient 'a' must be non-zero for a quadratic equation.")
+
+    discriminant = b * b - 4 * a * c
+    if discriminant < 0:
+        raise ValueError(
+            f"Discriminant is negative ({discriminant}); equation has no real roots."
+        )
+
+    sqrt_disc = math.sqrt(discriminant)
+    denom = 2 * a
+    x1 = (-b + sqrt_disc) / denom
+    x2 = (-b - sqrt_disc) / denom
+    return (x1, x2)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: