Please help I am trying to get my pygame to display what is in the picture and keep getting the error "joystick is not defined" code is provided. import pygame

import random

# Initialize Pygame

pygame.init$()$

# Screen dimensions

WIDTH, HEIGHT $=800,600$

screen $=$ pygame.display.set$\_$mode$(($WIDTH$,$ HEIGHT$))$

clock $=$ pygame.time.Clock$()$

# Load sound

click$\_$sound $=$ pygame.mixer.Sound$("$collision$.$mp$3")$

# Ball class

class Ball:

def $\_\_$init$\_\_($self$,$ x$,$ y$,$ radius, mass, color$)$:

self.location $=[$x$,$ y$]$

self.vector $=[0,0]$

self.acceleration $=[0,0]$

self.radius $=$ radius

self.mass $=$ mass

self.color $=$ color

def addForce$($self$,$ force$)$:

fx$,$ fy $=$ force

self.acceleration$[0]+=$ fx $/$ self.mass

self.acceleration$[1]+=$ fy $/$ self.mass

def update$($self$)$:

# Update motion

self.vector$[0]+=$ self.acceleration$[0]$

self.vector$[1]+=$ self.acceleration$[1]$

self.location$[0]+=$ self.vector$[0]$

self.location$[1]+=$ self.vector$[1]$

self.acceleration $=[0,0]$ # Reset acceleration

# Boundary collision

if self.location or self.location$[0]>=$ WIDTH:

self.vector$[0]=-$self.vector$[0]$

if self.location or self.location$[1]>=$ HEIGHT:

self.vector$[1]=-$self.vector$[1]$

def draw$($self$,$ screen$)$:

pygame.draw.circle$($screen$,$ self.color, $($int$($self$.$location$[0]),$ int$($self$.$location$[1])),$ self.radius$)$

# Initialize balls

balls $=[]$

for i in range$(10)$:

radius $=$ random.randrange$(10,50)$

x $=$ random.randrange$($radius$,$ WIDTH $-$ radius$)$

y $=$ random.randrange$($radius$,$ HEIGHT $-$ radius$)$

mass $=$ radius $**2$

color $=($random$.$randrange$(0,256),$ random.randrange$(0,256),$ random.randrange$(0,256))$

ball $=$ Ball$($x$,$ y$,$ radius, mass, color$)$

balls.append$($ball$)$

# Initialize joystick

pygame.joystick.init$()$

joystick$\_$count $=$ pygame.joystick.get$\_$count$()$

if joystick$\_$count $>0$:

joystick $=$ pygame.joystick.Joystick$(0)$

joystick.init$()$

print$($f$"$Number of joysticks: $\{$joystick$\_$count$\}")$

print$($f$"$Joystick name: $\{$joystick$.$get$\_$name$()\}")$

print$($f$"$GUID: $\{$joystick$.$get$\_$guid$()\}")$

print$($f$"$Number of axes: $\{$joystick$.$get$\_$numaxes$()\}")$

print$($f$"$Number of buttons: $\{$joystick$.$get$\_$numbuttons$()\}")$

print$($f$"$Number of hats: $\{$joystick$.$get$\_$numhats$()\}")$

# Game loop

running $=$ True

gravity$\_$enabled $=$ True

while running:

for event in pygame.event.get$()$:

if event.type $==$ pygame.QUIT:

running $=$ False

elif event.type $==$ pygame.JOYBUTTONDOWN:

# A button to reset ball position

if joystick.get$\_$button$(0)$: # A Button

for ball in balls:

ball.location $=[$WIDTH $//2,$ HEIGHT $//2]$

ball.vector $=[0,0]$

# LT $+$ Y Button to quit game

if joystick.get$\_$button$(6)$ and joystick.get$\_$button$(3)$: # LT $+$ Y Buttons

running $=$ False

# Read joystick axes for wind$/$force application

left$\_$wind $=$ joystick.get$\_$axis$(0)$

up$\_$force $=$ joystick.get$\_$axis$(1)$

right$\_$wind $=$ joystick.get$\_$axis$(3)$

# Apply forces based on joystick input

for ball in balls:

if left$\_$wind $>0\mathrm{.}1$:

ball.addForce$((-1,0))$

if right$\_$wind :

ball.addForce$((1,0))$

if up$\_$force :

force$\_$up $=-$up$\_$force $*10$ # Adjust multiplier as needed

ball.addForce$((0,-$force$\_$up$))$

screen.fill$((0,0,0))$

for ball in balls:

ball.update$()$

ball.draw$($screen$)$

pygame.display.flip$()$

clock.tick$(60)$Number of joysticks: $1$

Joystick $0$

Joystick name: Nintendo Switch Pro Controller

GUID: $030000007e0500000920000000026800$

Number of axes: $6$

Axis $0$ value: $0\mathrm{.}008$

Axis $1$ value: $-0\mathrm{.}011$

Axis $2$ value: $0\mathrm{.}003$

Axis $3$ value: $-0\mathrm{.}011$

Axis $4$ value: $0\mathrm{.}000$

Axis $5$ value: $0\mathrm{.}000$

Number of buttons: $16$

Button $0$ value: $0$

Button $1$ value: $0$

Button $2$ value: $0$

Button $3$ value: $0$

Button $4$ value: $0$

Button $5$ value: $0$

Button $6$ value: $0$

Button $7$ value: $0$

Button $8$ value: $0$

Button $9$ value: $0$

Button $10$ value: $0$

Button $11$ value: $0$

Button $12$ value: $0$

Button $13$ value: $0$

Button $14$ value: $0$

Button $15$ value: $0$

Number of hats: $0$

ubric:

$0$: UI shown above is implemented and responds to all controlle

$0$ : Apply to your gravity and wind project.

$10$: gravity toggled on$/$off $($you pick$)$

$10$: left wind applied momentarily $($Axis $0,0..-1)$

$10$: right wind applied momentarily $($Axis $3,-1.\mathrm{.}0)$