update Armstrong_number.py

becuase it is not displaying any output and logic does not works well

Author: sundaram2021

Armstrong_number.py

@@ -1,15 +1,18 @@
 def is_armstrong_number(number):
-    numberstr = str(number)
-    length = len(numberstr)
-    num = number
-    rev = 0
-    temp = 0
-    while num != 0:
-        rem = num % 10
-        num //= 10
-        temp += rem**length
-    return temp == number
+    sum = 0
 
+    # find the sum of the cube of each digit
+    temp = number
+    while temp > 0:
+       digit = temp % 10
+       sum += digit ** 3
+       temp //= 10
+    
+    # display the result
+    if number == sum:
+       print(number,"is an Armstrong number")
+    else:
+       print(number,"is not an Armstrong number")
 
-number = int(input("Enter the number"))
+number = int(input("Enter the number : "))
 is_armstrong_number(number)