Write a function which returns if a number passed as an argument is a prime or composite number.

def prime_composite(n):
    if n <= 1:
        return "Neither Prime nor Composite"

    count = 0
    for i in range(1, n + 1):
        if n % i == 0:
            count = count + 1

    if count == 2:
        return "Prime Number"
    else:
        return "Composite Number"

num = int(input("Enter a number: "))
result = prime_composite(num)
print(result)

Enter a number: 7
Prime Number

Enter a number: 12
Composite Number