שאלה דחופה בפייתון

צריך לכתוב פונקציה המבדקת את המספר
כדי להחזיר את הערכים מהפונקציה ל-def main(), ואז להשתמש ב-return ולשלוח את הערכים חזרה ל-main()

שואל השאלה:
אני יודע מה עושים אבל לא איך רושמים את זה ובגלל זה ביקשתי אם מישהו יכול לרשום לי בפייתון ,

'''
def is_prime(n):
if n < 2:
return false
for i in range(2, int(n ** 0.5) + 1):
if n % i == 0:
return false
return true

def main():
n = int(input("enter a number: "))
if is_prime(n):
print(n, "is a prime number")
else:
print(n, "is not a prime number")
''

השתמש בקוד שסופק כדי ליצור פונקציה שבודקת אם מספר הוא ראשוני ומחזירה את הערך ל-main().