Процедура prime_number определяет, является ли число простым или нет.

def prime_number(num):
is_prime = True
if num < 2:
is_prime = False
else:
for i in range(2, num):
if num % i == 0:
is_prime = False
break
return is_prime
n = int(input("Введите число: "))
result = prime_number(n)
if result:
print("{} - простое число".format(n))
else:
print("{} - не простое число".format(n))

Эта процедура принимает на вход одно число и проверяет, является ли оно простым. В случае если число является простым, процедура возвращает True, в противном случае - False.