Составить программу для нахождения корней квадратного уравнения ах2+вх+с

Предыдущий
вопрос Следующий
вопрос

import math
def quadratic_roots(a, b, c):
discriminant = b**2 - 4*a*c
if discriminant > 0:
root1 = (-b + math.sqrt(discriminant)) / (2*a)
root2 = (-b - math.sqrt(discriminant)) / (2*a)
return root1, root2
elif discriminant == 0:
root = -b / (2*a)
return root
else:
real_part = -b / (2*a)
imaginary_part = math.sqrt(abs(discriminant)) / (2*a)
root1 = complex(real_part, imaginary_part)
root2 = complex(real_part, -imaginary_part)
return root1, root2
a = float(input("Введите коэффициент a: "))
b = float(input("Введите коэффициент b: "))
c = float(input("Введите коэффициент c: "))
roots = quadratic_roots(a, b, c)
print("Корни уравнения: ", roots)

Программа принимает на вход значения коэффициентов квадратного уравнения, вычисляет дискриминант и в зависимости от его значения находит корни уравнения. Если дискриминант больше нуля, выводятся два действительных корня, если равен нулю - один действительный корень, если меньше нуля - два комплексных корня.