simple_calculator.py

# Simple Calculator (Improved Version)

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:
        return None
    return a / b


def get_numbers():
    while True:
        try:
            a = float(input("Enter Your First Number : "))
            b = float(input("Enter Your Second Number : "))
            return a, b
        except ValueError:
            print("❌ Invalid input! Please enter valid numbers.\n")


print("Choose an operation :")
print("1. Add")
print("2. Subtract")
print("3. Multiply")
print("4. Divide")
print("5. Exit")
print("-" * 40)

while True:
    try:
        choice = int(input("Enter Choice : "))

        if choice == 1:
            a, b = get_numbers()
            print(f"Sum of {a} and {b} = {add(a, b)}\n")

        elif choice == 2:
            a, b = get_numbers()
            print(f"Subtraction of {a} and {b} = {subtract(a, b)}\n")

        elif choice == 3:
            a, b = get_numbers()
            print(f"Multiplication of {a} and {b} = {multiply(a, b)}\n")

        elif choice == 4:
            a, b = get_numbers()
            result = divide(a, b)
            if result is None:
                print("❌ Error: Cannot divide by zero.\n")
            else:
                print(f"Division of {a} and {b} = {result}\n")

        elif choice == 5:
            print("Calculator exited. Goodbye 👋")
            break

        else:
            print("❌ Invalid choice. Please select between 1 and 5.\n")

        print("-" * 40)

    except ValueError:
        print("❌ Invalid choice! Please enter a number.\n")