calculator.py

def main():

    OPERATIONS = {
    '+' : add,
    '-' : subtract,
    '*' : multiply,
    '/' : divide,
}
 
    while True:
        try:
            first_n = float(input("Enter your first number: "))
            second_n = float(input("Enter your second number: "))
            operator = get_operator()

            result = OPERATIONS[operator](first_n, second_n)
        
        except ValueError:
            print("Invalid input!")
            continue
            

        except ZeroDivisionError as e:
            print(e)
            continue
        

        print(f"{first_n} {operator} {second_n} = {result}")
        break



            
# getting operators
def get_operator():
    while True:
        operator = input("Choose an operator (+, -, *, /): ").strip()
        if operator in ['+', '-', '*', '/']:
            return operator
        else:
            print("Invalid operator!")
            continue
            
        

# creat functions for add, substruct, multiply, divide
def add(a, b):
    return  a+b
def subtract(a, b):
    return a-b
def multiply(a, b):
    return a*b
def divide(a, b):
    if b == 0:
        raise ZeroDivisionError("Cannot divide by zero")
    
    return a/b



if __name__ == "__main__":
    main()