giúp mik vs mik cần gấp (LÀM CODE PYTHON GIÚP MIK NHA MN)

\begin{array}{c} \color{#db1616}{\texttt{#Khoadang09}} \end{array}

def 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
n = int(input())
print("YES" if prime(n) else "NO")

from math import sqrt
def isPrime(n):
    if n <= 1:
        return False
    else:
        for x in range(2, int(sqrt(n))):
            if n % x == 0:
                return False
        return True
n = int(input())
if isPrime(n):
    print("YES")
else:
    print("NO")