Untitled

# ...

choice = input("Enter choice (1-4): ")

num1 = float(input("Enter first number: "))
num2 = float(input("Enter second number: "))

if choice == '1':
    print(num1, "+", num2, "=", add(num1, num2))
elif choice == '2':
    print(num1, "-", num2, "=", subtract(num1, num2))
elif choice == '3':
    print(num1, "*", num2, "=", multiply(num1, num2))
elif choice == '4':
    print(num1, "/", num2, "=", divide(num1, num2))
else:
    print("Invalid input.")

# ...

# Additional code for handling user's desire to continue calculating

while True:
    cont = input("Do you want to perform another calculation? (yes/no): ")
    if cont.lower() == "no":
        print("Calculator program ended.")
        break
    elif cont.lower() == "yes":
        choice = input("Enter choice (1-4): ")

        num1 = float(input("Enter first number: "))
        num2 = float(input("Enter second number: "))

        if choice == '1':
            print(num1, "+", num2, "=", add(num1, num2))
        elif choice == '2':
            print(num1, "-", num2, "=", subtract(num1, num2))
        elif choice == '3':
            print(num1, "*", num2, "=", multiply(num1, num2))
        elif choice == '4':
            print(num1, "/", num2, "=", divide(num1, num2))
        else:
            print("Invalid input.")
    else:
        print("Invalid input. Please enter 'yes' or 'no'.")