Помогите решить задачу на Python

def is_lucky_number(num):
num_str = str(num)
if len(num_str) != 6:
return False

first_half = int(num_str[:3])
second_half = int(num_str[3:])

return sum(map(int, str(first_half))) == sum(map(int, str(second_half)))

def find_nearest_lucky_number(num):
while True:
num += 1
if is_lucky_number(num):
return num

# Вводим шестизначное натуральное число
try:
num = int(input("Введите шестизначное натуральное число: "))
except ValueError:
print("Ошибка! Введите корректное число.")
else:
if is_lucky_number(num):
print("Это счастливое число!")
else:
nearest_lucky_num = find_nearest_lucky_number(num)
print(f"Ближайшее счастливое число: {nearest_lucky_num}")