Basic calculator

[vedanshagarwal4]

No description provided.

[ritesharrma]

The whole basic calculator code is fine. But I want to suggest some basic enhancement changes.
def operate(a, b, choice):
operations = {
1: a + b,
2: a - b,
3: a * b,
4: a / b if b != 0 else "Division by zero error",
5: a ** b,
6: a % b if b != 0 else "Modulo by zero error"
}

return operations.get(choice, "Invalid choice")

if name == 'main':
# Menu - Available choices
print("1. Addition\n2. Subtraction\n3. Multiplication\n4. Division\n5. Exponentiation\n6. Modulo\n")

try:
    first_num = float(input('Enter first number: '))
    second_num = float(input('Enter second number: '))
    choice = int(input('Enter choice from the above list: '))

    result = operate(first_num, second_num, choice)
    print(f"Result: {result}")
except ValueError:
    print("Invalid input. Please enter numerical values for numbers and an integer for the choice.")