Переделать пирамиду в параллелепипед

from tkinter import Tk, Canvas
import numpy as np
 
 
# Колір заливки грані
def trColor(cn, step):
    ci = int(step) % 255
    if ci < 16:
        if cn == 0:
            return '#0' + hex(ci)[2:4] + '0000'
        elif cn == 1:
            return '#000' + hex(ci)[2:4] + '00'
        elif cn == 2:
            return '#00000' + hex(ci)[2:4]
        else:
            return '#000' + hex(ci)[2:4] + '0' + hex(ci)[2:4]
    else:
        if cn == 0:
            return '#' + hex(ci)[2:4] + '0000'
        elif cn == 1:
            return '#00' + hex(ci)[2:4] + '00'
        elif cn == 2:
            return '#0000' + hex(ci)[2:4]
        else:
            return '#00' + hex(ci)[2:4] + hex(ci)[2:4]
 
 
# Матриця переміщення на площині
def Move(p, dx, dy):
    return np.matmul([p[0], p[1], 1], np.array([[1, 0, 1],
                                                [0, 1, 0],
                                                [dx, dy, 1]]))
 
 
# Матриця повороту навколо осі Z
def Tz(psi):
    return np.array([[np.cos(psi), np.sin(psi), 0, 0],
                     [-np.sin(psi), np.cos(psi), 0, 0],
                     [0, 0, 1, 0],
                     [0, 0, 0, 1]])
 
 
# Загальна аксонометрична проекція
def AxonProjection(p, u, v):
    return np.matmul(p, np.matmul(np.array([[-np.cos(u), 0, np.sin(u), 0],
                                            [0, 1, 0, 0],
                                            [-np.sin(u), 0, -np.cos(u), 0],
                                            [0, 0, 0, 0]]),
                                  np.array([[1, 0, 0, 0],
                                            [0, -np.cos(v), -np.sin(v), 0],
                                            [0, np.sin(v), -np.cos(v), 0],
                                            [0, 0, 0, 0]])))
 
 
root = Tk()
 
h = 200  # Висота піраміди
a = 200  # Довжина сторони основи піраміди
r = np.sqrt(a ** 2 / 2)  # Радіус описаного кола навколо основи піраміди
points = [[0, 0, h, 0]]  # Вершина піраміди
for i in range(0, 4):
    points.append([r * np.cos(i * np.pi / 2), r * np.sin(i * np.pi / 2), 0, 1])
canvas = Canvas(root, width=700, height=700, bg='black')
 
c = 255  # Початкове значення інтенсивності кольору
dc = 0.1  # Приріст інтенсивності кольору
phi = np.pi / 6  # 30 градусів
theta = np.pi / 4  # 45 градусів
while True:
    c = c + dc
    if c < 0 or c > 255:
        dc = -dc
    canvas.delete("all")
    tr = []  # грані піраміди
    for i in range(0, 4):
        tr.append([AxonProjection(points[0], phi, theta),
                   AxonProjection(points[i + 1], phi, theta),
                   AxonProjection(points[(i + 1) % 4 + 1], phi, theta)])
    for j in range(0, 4):
        plainPoints = []  # Вершини полігона грані
        for i in range(0, 3):
            plainPoints.append(Move(tr[j][i], 350, 350)[0])
            plainPoints.append(Move(tr[j][i], 350, 350)[1])
        canvas.create_polygon(plainPoints, fill=trColor(j, c))
    points = list(map(lambda x: np.matmul(x, Tz(0.001)), points))  # Поворот навколо висоти піраміди
    canvas.pack()
    root.update()
 
root.mainloop()