Write a Program to develop prime number sever (client will send any number to sever ,server will send the response the number is prime or not)

def prime(n):
    for i in range(2,n):
        if n % i == 0:
            return False
    return True

n = int(input("Enter any number:"))
if prime(n):
    print('Prime')
else:
    print('Not prime')