Решение уравнения

import math
 
print("A*x**2 + B*x + C = 0")
A = float(input("Введите число А: "))
B = float(input("Введите число B: "))
C = float(input("Введите число C: "))
D = B ** 2 - 4 * A * C
if D > 0:
    x1 = (-B + math.sqrt(D)) / (2 * A)
    x2 = (-B - math.sqrt(D)) / (2 * A)
    print(f'x1= {x1}')
    print(f'x2= {x2}')
elif D == 0:
    x = -B / (2 * A)
    print(f'x= {x}')
else:
    print("Корней нет")

def lin_equ(a,b):
    if a==0 and b==0:
        return (-1,0)
    elif a==0:
        return (0,0)
    else:
        return (1,-b/a)
        
def sqr_equ(a,b,c):
    if a==0 and b==0 and c==0:
        return (-1,0)
    elif a==0:
        return lin_equ(b,c)
    else:
        d=b*b-4*a*c
        if d>0:
            return (2,((-b+d**0.5)/(2*a),(-b-d**0.5)/(2*a)))
        elif d<0:
            return (0,0)
        else:
            return (1,-b/(2*a))
            
def task():
    while True:
        u=int(input("Что решаем? 1-линейное, 2-квадратное, 0-выход: "))
        if u==0:
            return
        if u==1:
            a=float(input("a="))
            b=float(input("b="))
            (n,r)=lin_equ(a,b)
            if (n==-1):
                print("Любое вещественное число")
            elif (n==0):
                print("Нет решений")
            else:
                print("Единственный корень=",r)
        elif u==2:
            a=float(input("a="))
            b=float(input("b="))
            c=float(input("c="))
            (n,r)=sqr_equ(a,b,c)
            if (n==-1):
                print("Любое вещественное число")
            elif (n==0):
                print("Нет вещественных решений")
            elif (n==1):
                print("Единственный (кратный) корень=",r)
            else:
                print("Два вещественных корня: x1=",r[0],"x2=",r[1])
                
task()

from sympy import  nroots, I
from sympy.abc import x
 
a = list(map(float, input('Введите через пробел коэффициенты квадратного уравнения :').split()))
y = f'{a[0]}*x*x + {a[1]}*x* + {a[2]}'
 
print('Корни уравнения:')
for i in nroots(y):
    print(i)