Найти площадь прямоугольника, треугольника или круга. Python

В зависимости от того, что выберет пользователь, вычислить площадь либо прямоугольника, либо треугольника, либо круга. Если выбраны прямоугольник или треугольник, то надо запросить длины сторон, если круг, то его радиус.

Площадь треугольника вычисляется по формуле Герона:

где p — это полупериметр, a, b, c — длины сторон. Полупериметр равен половине периметра, то есть половине суммы сторон.

Площадь прямоугольника равна произведению его сторон. Площадь круга вычисляется по формуле S = πr².

print("1-прямоугольник, 2-треугольник, 3-круг")
figure = input("Выберите фигуру: ")
 
if figure == '1':
    print("Длины сторон прямоугольника:")
    a = float(input("a = "))
    b = float(input("b = "))
    print("Площадь: %.2f" % (a * b))
elif figure == '2':
    print("Длины сторон треугольника:")
    a = float(input("a = "))
    b = float(input("b = "))
    c = float(input("c = "))
    p = (a + b + c) / 2
    from math import sqrt
    s = sqrt(p * (p - a) * (p - b) * (p - c))
    print("Площадь: %.2f" % s)
elif figure == '3':
    r = float(input("Радиус круга R = "))
    from math import pi
    print("Площадь: %.2f" % (pi * r ** 2))
else:
    print("Ошибка ввода")

Пример выполнение программы:

1-прямоугольник, 2-треугольник, 3-круг
Выберите фигуру: 2
Длины сторон треугольника:
a = 4
b = 5
c = 6
Площадь: 9.92