Помогите пожалуйста с задачей в python

def add(x, y):
return x + y

def subtract(x, y):
return x - y

def multiply(x, y):
return x * y

def divide(x, y):
if y == 0:
return "Ошибка: Деление на ноль!"
return x / y

def calculator():
while True:
print("""
[1] Сложение
[2] Вычитание
[3] Умножение
[4] Деление
[0] Выход
""")

choice = input("Выберите операцию: ")

if choice == '0':
print("Выход из программы.")
break
elif choice in ('1', '2', '3', '4'):
try:
num1 = float(input("Введите первое число: "))
num2 = float(input("Введите второе число: "))

if choice == '1':
print("Результат:", add(num1, num2))
elif choice == '2':
print("Результат:", subtract(num1, num2))
elif choice == '3':
print("Результат:", multiply(num1, num2))
elif choice == '4':
print("Результат:", divide(num1, num2))
except ValueError:
print("Ошибка: Введите корректное число!")
else:
print("Ошибка: Неверный выбор операции!")

calculator()