Next Prime Number-Python

#Q. Next Prime Number – Have the program find prime numbers until the user chooses to stop asking for the next one.
def is_prime(x):

if x == 2:
return True

if x % 2 == 0:
return False

for i in range(3, int(x**0.5)+1, 2):
if x % i == 0:
return False

return True
def generate_primes(current_prime):
“””
Returns the next prime number

“””

new_prime = current_prime + 1

while True:

if not is_prime(new_prime):
new_prime += 1
else:
break

return new_prime
def main(): # Wrapper function

current_prime = 2

while True:

ask = input(‘Would you like to know the next prime number? (Y/N): ‘)

if ask.lower().startswith(‘y’):
print(current_prime)
current_prime = generate_primes(current_prime)

else:
break

if __name__ == ‘__main__’:
main()