Ошибка с отталкиванием от стенок

import pygame
import os
import time
 
pygame.init()
 
WIDTH, HEIGHT = 180, 1500
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Monkey Jump")
 
WHITE = (255, 255, 255)
 
FPS = 60
clock = pygame.time.Clock()
 
character_image = pygame.image.load(os.path.join("data", "MONKEY_JUMP.png"))
character_width, character_height = character_image.get_size()
 
new_width = character_width // 2
new_height = character_height // 2
character_image = pygame.transform.scale(character_image, (new_width, new_height))
 
x, y = WIDTH // 2, HEIGHT - new_height
speed_x, speed_y = 0, 0
gravity = 0.5
jump_force = -15
 
on_ground = False
on_wall = False
wall_slide_speed = 1
 
wall_push_speed = 3
wall_push_direction = 0
push_counter = 0
 
 
def draw():
    screen.fill(WHITE)
    screen.blit(character_image, (x, y))
    pygame.display.flip()
 
 
running = True
while running:
    clock.tick(FPS)
 
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
 
        if event.type == pygame.MOUSEBUTTONDOWN:
            if event.button == 1:
                print('Была нажата левая кнопка мыши')
                if on_ground or on_wall:
                    speed_y = jump_force
                    on_ground = False
                    on_wall = False
 
    keys = pygame.key.get_pressed()
 
    speed_y += gravity
    y += speed_y
 
    if y + new_height >= HEIGHT:
        y = HEIGHT - new_height
        speed_y = 0
        on_ground = True
 
    previous_on_wall = on_wall
    on_wall = False
    if x <= 0 or x + new_width >= WIDTH:
        on_wall = True
        speed_y = min(speed_y, wall_slide_speed)
 
    if push_counter == 0:
        if x <= 0:
            on_wall = True
            speed_y = min(speed_y, wall_slide_speed)
            wall_push_direction = 1
 
        elif x + new_width >= WIDTH:
            on_wall = True
            speed_y = min(speed_y, wall_slide_speed)
            wall_push_direction = -1
 
    if on_wall and event.type == pygame.MOUSEBUTTONDOWN:
        if push_counter == 0:
            print(f"Отталкивание от стены! Текущие координаты X: {x}")
            push_counter = 1
 
    if push_counter > 0:
        x += wall_push_speed * wall_push_direction
        push_counter += 1
 
        if push_counter > 20:
            push_counter = 0
 
    if keys[pygame.K_a]:
        x -= 5
    if keys[pygame.K_d]:
        x += 5
 
    if x < 0:
        x = 0
    if x + new_width > WIDTH:
        x = WIDTH - new_width
 
    draw()
 
pygame.quit()