关于#python#的问题，如何解决？

python函数：输入两个如果他们都是素数就求和，只要有其中一个不是就重新输入，判断素数要用自定义函数。急！

def isprime(n):  # 判断素数函数
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    else:
        return True
 
 
def f(n):  # 找小于n的素数并求和
    ls = []
    count = 0
    sum = 0
    for i in range(n, 1, -1):
        if isprime(i):
            count += 1
            if count <= 10:
                ls.append(i)
                sum += i
            else:
                break
    return ls, sum
 
 
p = int(input())
result = f(p)
result[0].sort()
for item in result[0]:
    if result[0].index(item) != len(result[0]) - 1:
        print(item, end='+')
    else:
        print(item, end='=')
print(result[1])