Вычислить: высоту пирамиды; объем пирамиды; площадь ее полной поверхности

from math import sqrt, asin, tan, sin, cos, pi
 
n = 5
a = float(input("a = "))
b = float(input("b = "))
 
# высота пирамиды
R= a/(2*sin(pi/n))
h = round(sqrt(b**2-R**2), 1)
print(f"высота пирамиды {h}")
 
# объем пирамиды
So = n*a**2/(4*tan(pi/n))
V = round(So*h/3, 1)
print(f"объем пирамиды {V}")
 
# площадь полной поверхности
r = a/(2*tan(pi/n))
l = sqrt(h**2+r**2)
Sp = round(So+n*a*l/2, 1)
print(f"площадь полной поверхности {Sp}")
 
# радиус вписанной сферы
rs = round(h*sin(r/l)/(1+sin(r/l)), 1)
print(f"радиус вписанной сферы {rs}")
 
# радиус описанной сферы
Rs = round(b/sqrt(2-2*cos(pi-2*asin(h/b))), 1)
print(f"радиус описанной сферы {Rs}")

print("Введите номер задания, которое необходимо решить (1 или 2)")
task_num = input() # Осуществляется выбор задания, которое будем решать
while task_num != "1" and task_num != "2": # Если пользователь вводит неправильное число, просим ввести правильное
    print("Введите либо 1, либо 2!")
    task_num = input()
if task_num == "1":  # Если выбрано число 1, то решается задача 1
    try: # Код ниже может быть подвергнут попытке сломать программу, поэтому он включён в блок try
        A = int(input('Введите число A:'))
        A2 = A * A #формула для расчета числа A в степени 2
        A4 = A2 * A2 #формула для расчета числа A в степени 4
        A8 = A4 * A4 #формула для расчета числа A в степени 8
        A28 = A8 * A4 * A4 * A4 * A4 * A2 *A2 #формула для расчета числа A в степени 28
        print('Задано', A )
        print(A, ' в степени 28 =', A28)
    except ValueError: # Если пользователь ввёл неверные данные, пишем ошибку и аварийно завершаем программу
        print("Были введены неверные значения! Вам требовалось написать число! Программа аварийно завершена.")
if task_num == "2": # Если выбрано число 2, то решается задача 2
    try: # Код ниже может быть подвергнут попытке сломать программу, поэтому он включён в блок try
        S = int(input('Введите площадь основания пирамиды S:'))
        h = int(input('Введите высоту пирамиды h:'))
        V = S * h / 3 #формула для расчета объема пирамиды
        print(V, "объем пирамиды {V}")
    except ValueError: # Если пользователь ввёл неверные данные, пишем ошибку и аварийно завершаем программу
        print("Были введены неверные значения! Вам требовалось написать число! Программа аварийно завершена.")