Armstrong数字两次显示错误输出，并在末尾更正(Python)

def is_armstrong_number(num):
    n = len(str(num))
    sum = 0
    temp = num
    while temp > 0:
        digit = temp % 10
        sum += digit ** n
        temp //= 10
    return num == sum

# 示例用法
num = 153
if is_armstrong_number(num):
    print(num, "是一个Armstrong数字")
else:
    print(num, "不是一个Armstrong数字")

153 是一个Armstrong数字

def fix_armstrong_number(num):
    if not is_armstrong_number(num):
        return num
    
    n = len(str(num))
    temp = num
    while temp > 0:
        digit = temp % 10
        temp //= 10
    new_num = (digit + 1) % 10**(n-1)
    fixed_num = num // 10 * 10 + new_num
    return fixed_num

# 示例用法
num = 153
fixed_num = fix_armstrong_number(num)
print("错误的Armstrong数字:", num)
print("修正后的Armstrong数字:", fixed_num)

错误的Armstrong数字: 153
修正后的Armstrong数字: 152