Update app.py

app.py

@@ -44,18 +44,19 @@ def check_is_prime(number_to_check:int)-> str: #it's import to specify the retur
     # Negative numbers, 0 and 1 are not primes
     if number_to_check > 1:
   
-    # Iterate from 2 to n // 2
-    for i in range(2, (number_to_check//2)+1):
+        # Iterate from 2 to n // 2
+        for i in range(2, (number_to_check//2)+1):
       
-        # If num is divisible by any number between
-        # 2 and n / 2, it is not prime
-        if (number_to_check % i) == 0:
-            print(number_to_check, "is not a prime number")
-            break
+            # If num is divisible by any number between
+            # 2 and n / 2, it is not prime
+            if (number_to_check % i) == 0:
+                return_text = "is not a prime number")
+                break
+        else:
+            return_text = "is a prime number"
     else:
-        return_text = "is a prime number"
-else:
-    return_text = "is not a prime number"    
+        return_text = "is not a prime number"    
+
     return return_text
 
 @tool