Нужна помощь с визуализацией доски Гальтона

import pygame as pg
import pymunk.pygame_util
import pymunk
import numpy as np
from random import randrange
 
# Установка параметров pygame и pymunk
pg.init()
RES = WIDTH, HEIGHT = 1440, 900
FPS = 100
surface = pg.display.set_mode(RES)
clock = pg.time.Clock()
draw_options = pymunk.pygame_util.DrawOptions(surface)
pymunk.pygame_util.positive_y_is_up = False
 
# Создание пространства pymunk с гравитацией и количеством итераций
space = pymunk.Space()
space.gravity = 0, 8000
space.iterations = 250
 
# Параметры для создания шариков
ball_mass, ball_radius = 1, 7
 
# Параметры для сегментов и гвоздиков
segment_thickness = 6
peg_radius_default = 5
 
# Координаты и размеры сегментов и гвоздиков
a, b, c, d = 10, 100, 18, 40
x1, x2, x3, x4 = a, WIDTH // 2 - c, WIDTH // 2 + c, WIDTH - a
y1, y2, y3, y4, y5 = b, HEIGHT // 4 - d, HEIGHT // 4, HEIGHT // 2 - 1.5 * b, HEIGHT - 4 * b
L1, L2, L3, L4 = (x1, -100), (x1, y1), (x2, y2), (x2, y3)
R1, R2, R3, R4 = (x4, -100), (x4, y1), (x3, y2), (x3, y3)
B1, B2 = (0, HEIGHT), (WIDTH, HEIGHT)
 
# Функция для создания шарика в пространстве
def create_ball(space, ball_radius):
    ball_moment = pymunk.moment_for_circle(ball_mass, 0, ball_radius)
    ball_body = pymunk.Body(ball_mass, ball_moment)
    y = 50  # Фиксированная Y-координата (например, в верхней части окна)
    x = 720  # Рандомная позиция по X от левой стенки до правой
    ball_body.position = x, y
    mean_vx, std_dev_vx = 0, 100
    mean_vy, std_dev_vy = 0, 100
    vx = np.random.normal(mean_vx, std_dev_vx)
    vy = np.random.normal(mean_vy, std_dev_vy)
    ball_body.velocity = vx, vy
    ball_shape = pymunk.Circle(ball_body, ball_radius)
    ball_shape.elasticity = 0.1
    ball_shape.friction = 0.1
    space.add(ball_body, ball_shape)
    return ball_body
 
# Функция для создания сегмента
def create_segment(from_, to_, thickness, space, color):
    segment_shape = pymunk.Segment(space.static_body, from_, to_, thickness)
    segment_shape.color = pg.color.THECOLORS[color]
    space.add(segment_shape)
 
# Функция для создания гвоздика (пега)
def create_peg(x, y, space, color, peg_radius):
    circle_shape = pymunk.Circle(space.static_body, radius=peg_radius, offset=(x, y))
    circle_shape.color = pg.color.THECOLORS[color]
    circle_shape.elasticity = 0.1
    circle_shape.friction = 0.5
    space.add(circle_shape)
 
# Интерактивный ввод параметров симуляции
def get_simulation_parameters(ball_radius):
    balls_count_input = input(f"Введите количество шариков [default=1]: ")
    balls_count = int(balls_count_input) if balls_count_input else 1
 
    peg_radius_input = input(f"Введите размер гвоздиков (радиус) [default={peg_radius_default}]: ")
    peg_radius = int(peg_radius_input) if peg_radius_input else peg_radius_default
 
    ball_radius_input = input(f"Введите размер шариков (радиус) [default={ball_radius}]: ")
    ball_radius = int(ball_radius_input) if ball_radius_input else ball_radius
 
    return balls_count, peg_radius, ball_radius
 
# Создание сегментов, гвоздиков и платформ
def create_environment():
    # Создание гвоздиков (пегов)
    peg_y, step = y4, 60
    for i in range(10):
        peg_x = -1.5 * step if i % 2 else -step
        for j in range(WIDTH // step + 2):
            create_peg(peg_x, peg_y, space, 'darkslateblue', peg_radius_default)
            if i == 9:
                create_segment((peg_x, peg_y + 50), (peg_x, HEIGHT), segment_thickness, space, 'darkslategray')
            peg_x += step
        peg_y += 0.5 * step
 
    # Создание платформ
    platforms = (L1, L2), (L2, L3), (L3, L4), (R1, R2), (R2, R3), (R3, R4)
    for platform in platforms:
        create_segment(*platform, segment_thickness, space, 'darkolivegreen')
    create_segment(B1, B2, 20, space, 'darkslategray')
 
# Инициализация симуляции
create_environment()
 
# Интерактивный ввод параметров симуляции
balls_count, peg_radius, ball_radius = get_simulation_parameters(ball_radius)
 
# Инициализация шариков
balls = []
for _ in range(balls_count):
    balls.append(([randrange(256) for i in range(3)], create_ball(space, ball_radius)))
 
# Периодичность создания новых шаров (в кадрах)
spawn_interval = FPS / 20  # Например, новый шар каждые 2 секунды
frame_counter = 0  # Счетчик кадров для отслеживания периодичности создания новых шаров
 
# Главный игровой цикл
running = True
while running:
    surface.fill(pg.Color('black'))
 
    for event in pg.event.get():
        if event.type == pg.QUIT:
            running = False
 
    # Обновление счетчика кадров
    frame_counter += 1
 
    # Создание нового шара через определенные интервалы кадров
    if frame_counter % spawn_interval == 0:
        balls.append(([randrange(256) for i in range(3)], create_ball(space, ball_radius)))
 
    space.step(1 / FPS)
    space.debug_draw(draw_options)
 
    # Отрисовка шаров
    for color, ball in balls:
        pg.draw.circle(surface, color, (int(ball.position[0]), int(ball.position[1])), ball_radius)
 
    pg.display.flip()
    clock.tick(FPS)
 
pg.quit()