Python: Создать функцию, которая находит следующее простое число

Формулировка задачи:

Дано простое число.Создать функцию, которая находит следующее за ним простое число. на языке питон

Решение задачи: «Python: Создать функцию, которая находит следующее простое число»

textual
                            Листинг программы 
                        

                            var
  n: integer;
 
function IsPrime(x: integer): boolean;
var
  i: integer;
begin
  IsPrime := true;
  for i := 2 to Round(Sqrt(x)) do
    if x mod i = 0 then
      IsPrime := false;
end;
 
function NextPrime(x: integer): integer;
begin
  if not IsPrime(x) then
    exit
  else
  begin
    repeat
      Inc(x);
    until IsPrime(x);
    NextPrime := x;
  end
end;
 
begin
  repeat
    Write('Введите простое число: ');
    Readln(n);
  until IsPrime(n);
  Writeln('Следующее простое = ', NextPrime(n));
  Readln
end.
                        