import math
number = int(input("Enter a number: "))
while number < 0:
number = int(input("Enter a number: "))
if number % 2 == 0:
if number == 2:
print("Your number is a prime.")
exit()
print("Your number is not a prime.")
exit()
elif number % 2 == 1:
i = 2
while i <= math.sqrt(number):
if number % i == 0:
print("Your number is not a prime ")
exit()
i += 1
print("Your number is a prime.")
I hope this helps!