I need this code to display the UI in the picture i attached. It needs to look exactly like the picture and im stuck. Im including my code. pygame. 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$ 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]$ # Fix syntax error here

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]*=-1$

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

self.vector$[1]*=-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: # Fix event type check here

# A button to reset ball position

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

# A Button for ball in balls:

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)$:

running $=$ False

# Read joystick axes for wind$/$force application

left$\_$wind $=$ joystick.get$\_$axis$(0)$ if joystick$\_$count $>0$ else $0$

up$\_$force $=$ joystick.get$\_$axis$(1)$ if joystick$\_$count $>0$ else $0$

right$\_$wind $=$ joystick.get$\_$axis$(3)$ if joystick$\_$count $>0$ else $0$

# 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))$

# Clarify the intention of this line and adjust accordingly

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

for ball in balls:

ball.update$()$

ball.draw$($screen$)$

pygame.display.flip$()$

clock.tick$(60)$