Viết hàm giải phương trình bậc nhất và phương trình bậc hai

def level_1(a, b):

    if a == 0:

        if b == 0:

            return f"The equation({a}x + {b} = 0) has countless solutions"

        return f"The equation({a}x + {b} = 0) has no solution"

    return f"Solution of the equation({a}x + {b} = 0) is x = {-b/a}"

def level_2(a, b, c):

    if a == 0:

        return level_1(b, c)

    else:

        delta = b*b - 4*a*c

        if delta > 0:

            return f"Solution of the equation({a}x^2 + {b}x + {c} = 0) is:\nx1 = {(-b + delta**0.5)/(2*a)}\nx2 = {(-b - delta**0.5)/(2*a)}"

        elif delta == 0:

            return f"Solution of the equation({a}x^2 + {b}x + {c} = 0) is:\nx1 = x2 = {-b/(2*a)}"

        else:

            return f"The equation({a}x^2 + {b}x + {c} = 0) has no real solution (complex solution)"

try:

    function = int(input("Please choose an function:\nchoose 1: Solve the equation Ax + B = 0\nchoose 2: Solve the equation Ax^2 + Bx + C = 0\nYou choose: "))

    coefficient = input("Please input coefficient of the equation: ")

    print("-"*20)

    if function == 1:

        a, b = map(float, coefficient.split())

        print(level_1(a, b))

    elif function == 2:

        a, b, c = map(float, coefficient.split())

        print(level_2(a, b, c))

    else:

        print(f"Please choose an function:\nchoose 1: Solve the equation Ax + B = 0\nchoose 2: Solve the equation Ax^2 + B + C = 0")

except ValueError:

    print("input value error")

def solve(a,b,c=0):
    if c == 0:
        if a == 0:
            if b == 0:
                print("Phương trình có vô số nghiệm")
            else:
                print("Phương trình vô nghiệm")
        else:
            x1 = -b / a
            print(f"Phương trình có 1 nghiệm duy nhất:\nx = {x1}")
    else:
        if a == 0:
            if b == 0:
                print("Phương trình vô nghiệm")
            else:
                x1 = x2 = -c / b
                print(f"Phương trình có 1 nghiệm duy nhất:\nx = {x1}")
        else:
            delta = b**2 - 4*a*c

            if delta < 0:
                print("Phương trình vô nghiệm")    
            elif delta == 0:
                x1 = -b / (2 * a)
                print(f"Phương trình có nghiệm kép:\nx1 = x2 = {x1}")
            else:
                from math import sqrt
                x1 = (-b + sqrt(delta)) / (2 * a)
                x2 = (-b - sqrt(delta)) / (2 * a)
                print(f"Phương trình có hai nghiệm phân biệt:\nx1 = {x1}\nx2 = {x2}")

while True:
    try:
        option = int(input("Nhập option: "))

        if option == 1:
            print("Bạn đã chọn chức năng: Giải phương trình bậc 1")
            a,b = map(float,input("Nhập a,b: ").split())
            solve(a,b)
            break
        elif option == 2:
            print("Bạn đã chọn chức năng: Giải phương trình bậc 2")
            a,b,c = map(float,input("Nhập a,b,c: ").split())
            solve(a,b,c)
            break
        else:
            print("Vui lòng chọn một trong hai chức năng:\n1. Giải phương trình bậc nhất\n2. Giải phương trình bậc hai")
    except:
        print("Định dạng đầu vào không hợp lệ!")

import math

try:

    Chucnang = int(input('chon chuc nang:'))

    def phuongtrinhbac1(Func):

        if Func == 1:

            print('giai phuong trinh bac 1')

            a,b = map(float,input('nhap a va b lan luot la he so cua phuong trinh ax + b = 0 : ').split())

            if a != 0 :

                return ('phuong trinh co 1 nghiem duy nhat x = {}'.format(-b/a))

            elif a== 0 and b==0 :

                return('phuong trinh co vo so nghiem')

            elif a==0 and b != 0 :

                return('phuong trinh vo nghiem')

    opt1=phuongtrinhbac1(Chucnang)

    def phuongtrinhbac2(Func):

        if Func == 2:

            print('giai phuong trinh bac 2')

            a,b,c = map(float,input('nhap a ,b ,c lan luot la he so cua phuong trinh ax**2 + bx + c = 0 : ').split())

            delta = b**2 - (4*a*c)

            if a == 0 :

                return('phuong trinh co 1 nghiem duy nhat x = {}'.format(-c/b))

            elif delta < 0 :

                return ('phuong trinh vo nghiem')

            elif delta == 0 :

                return ('phuong trinh co nghiem kep x1=x2={}'.format(-b/(2*a)))

            elif delta > 0 :

                return ('phuong trinh co 2 nghiem phan biet x1 = {} , x2 = {}'.format((-b + math.sqrt(delta))/(2*a),(-b - math.sqrt(delta))/(2*a)))

    Opt2 = phuongtrinhbac2(Chucnang)

    def result(Func,opt1 ,opt2):

        if Func == 1:

            return  opt1

        elif Func == 2:

            return  opt2

        return '''Vui long chọn 1 trong 2 chức năng

    1.Giai Phuong Trinh Bac 1

    2.Giai Phuong Trinh Bac 2'''

    print(result(Chucnang,opt1,Opt2))

except:

    print('Dau vào không hợp lệ')

import math

# Hàm giải phương trình bậc hai
def phuongtrinhbachai(a,b,c):
    delta = b*b - 4*a*c
    if a == 0 and b == 0 and c == 0:
        txt = 'Phuong trinh co vo so nghiem'
    elif a == 0 and b == 0 and c != 0:
        txt = 'Phương trình vô nghiệm'
    elif a == 0 and b != 0:
        x = -c/b
        txt = 'Phương trình có một nghiệm duy nhất:\nx = {}'.format(x)
    elif a != 0:
        if delta < 0:
            txt = 'Phương trình vô nghiệm'
        elif delta == 0:
            x = -b/(2*a)
            txt = 'Phương trình có nghiệm kép:\nx1 = x2 = {}'.format(x)
        elif delta > 0:
            x1 = (-b + math.sqrt(delta))/a
            x2 = (-b - math.sqrt(delta))/a
            txt = 'Phương trình có hai nghiệm phân biệt:\nx1 = {}\nx2 = {}'.format(x1,x2)
    return txt

# Hàm giải phương trình bậc nhất
def phuongtrinhbacnhat(a,b):
    if a==0 and b==0:
        txt = 'Phuong trinh co vo so nghiem'
    elif a==0 and b!=0:
        txt = 'Phương trình vô nghiệm'
    elif a!=0 and b==0:
        txt = 'Phương trình có một nghiệm duy nhất:\nx = 0'
    elif a!=0 and b!=0:
        x = -b/a
        txt = 'Phương trình có một nghiệm duy nhất:\nx = {}'.format(x)
    return txt

try:
    loaiphuongtrinh = int(input())
    if loaiphuongtrinh == 1:
        a,b = map(float,input().split())
        print(phuongtrinhbacnhat(a,b))
    elif loaiphuongtrinh == 2:
        a,b,c = map(float,input().split())
        print(phuongtrinhbachai(a,b,c))
    else:
        print('Vui long chon mot trong hai chuc nang:\n1. Giai phuong trinh bac nhat\n2. Giai phuong trinh bac hai')
except:
    print('Dinh dang dau vao khong hop le!')

import math
def linear_equation(a, b):
    return "x = {}".format(-b/a)
def quadratic_equation(a, b, c):
    delta = b*b - 4*a*c
    if delta < 0:
        return "Phương trình vô nghiệm"
    elif delta == 0:
        return "x1 = x2 = {}".format(-b/(2*a))
    else:
        x1 = (-b - math.sqrt(delta))/(2*a)
        x2 = (-b + math.sqrt(delta))/(2*a)
        return "x1 = {}\nx2 = {}".format(x1, x2)
try:
    mode = input()
    if mode != "1" and mode != "2":
        print("Vui lòng chọn một trong hai chức năng: ")
        print("1. Giải phương trình bậc nhất", "\n2. Giải phương trình bậc hai")
    elif mode == 1:
        a, b = map(float, input().split())
        print(linear_equation(a, b))
    else:
        a, b, c = map(float, input().split())
        print(quadratic_equation(a, b, c))
except:
    print("Định dạng đầu vào không hợp lệ!")