Запросить у пользователя 2 числа и выполнить действие по выбору

import math
 
 
def find_divisors(num):
    """
    Функция для нахождения всех простых делителей числа
    """
    divisors = []
    for i in range(2, math.isqrt(num) + 1):
        while num % i == 0:
            divisors.append(i)
            num //= i
    if num > 1:
        divisors.append(num)
    return divisors
 
 
num1 = int(input("Введите 1-е число: "))
num2 = int(input("Введите 2-е число: "))
 
while True:
    print("Выберите действие, которое хотите выполнить:")
    print("1 - Посчитать их сумму")
    print("2 - Найти максимальное")
    print("3 - Проверить делится ли каждое из чисел на 10")
    print("4 - Найти все простые делители 1-го числа")
    choice = int(input("Введите выбранное действие: "))
 
    if choice == 1:
        print("Сумма чисел равна:", num1 + num2)
    elif choice == 2:
        print("Максимальное число:", max(num1, num2))
    elif choice == 3:
        if num1 % 10 == 0 and num2 % 10 == 0:
            print("Оба числа делятся на 10")
        elif num1 % 10 == 0:
            print("Первое число делится на 10")
        elif num2 % 10 == 0:
            print("Второе число делится на 10")
        else:
            print("Ни одно число не делится на 10")
    elif choice == 4:
        divisors = find_divisors(num1)
        if divisors:
            print("Простые делители первого числа:", divisors)
        else:
            print("Первое число не имеет простых делителей")
    else:
        print("Вы ввели неправильный выбор, программа завершена")
        break