Незнаю как в паскале, но в Python ты можешь сделать это при и переменной с инпутом.
Объяснение:
Создаёшь условную переменную A объявляешь её инпутом, далее сравниваешь введённый в переменную A результат (к примеру 1), и рисуешь соответствующую фигуру.
