Составить программу вычисления функции в Python

def p(t, k):
    if t < k:
        return t / (t - k)
    elif t > k:
        return t / (t + k)
    else:
        return t
 
 
a, b = map(int, input().split())
s = p(a, b) + p(a - b, b) + p(a + b, b) + p(a, b - a)
print(s)

def p(t, k): #  вычисления p(t,k) использовать функцию пользователя, где p вычисляется по формуле...
    if t < k: #  если t<k,
        return t / (t - k) # то t/(t-k).
    elif t > k: # Если t>k,
        return t / (t + k) # t/(t+k).
    else: # иначе...(остается один вариант t=k,)
        return t # то t/(t+k)
 
 
a, b = map(int, input().split())    # Получаем строку на вход, делим ее по символу {пробел}, получаем список из 2 элементов,
                                    # к каждому элементу списка применяем функцию int(),
                                    # чтобы преобразовать к целочисленному типу. Распаковываем список из 2 элементов
                                    # и присваиваем значения переменным {a} и {b}.
s = p(a, b) + p(a - b, b) + p(a + b, b) + p(a, b - a) # считаем по твоей формуле значение, используя нашу функцию выше
print(s) # Выводим результат вычисления