uses crt;
function ktnt(x: longint): boolean;
var i: longint;
begin
    if x <= 1 then exit(false);
    for i:=2 to trunc(sqrt(x)) do
        if x mod i = 0 then exit(false);
    exit(true);
end;

var n, t: longint;
begin
    clrscr;
    readln(n);
    while n <> 0 do 
        begin
            t:=t + sqr(n mod 10);
            n:=n div 10;
        end;
    if ktnt(t) then write('YES') else write('NO');
readln;
end.

Python:

n = int(input())

d = str(n)

c = list(d)

v = []

for i in c:
      z = int(i)
      v.append(z)

t = 0

for i in v:
      m = i * i
      t += m

def KTSNT(t):
      o = 0
      for i in range(1, t + 1):
                if t % i == 0:
                          o += 1
      if o == 2:
                   return 1
      if o != 2:
                 return 0

if KTSNT(t) == 1:
     print("Yes.")

if KTSNT(t) == 0:
     print("No.")