Code Python help

Bonjour,

J'ai besoin de quelqu'un qui s'y connait en codage python et qui pourrait m'aider s'il vous plaît ; je vous explique.
J'ai ce code :

Menu :

r = 255
def setup():
  size(640, 480)
  noStroke()
  background(175, 213, 255)
 
def mouseClicked(): 
    global r
    if r == 255:
        r = 0
    else:
        r = 255

def draw():
  fill(r, 50, 0)
  rect(100, 190, 440, 100)
  fill(0)
  text("Cliquez pour jouer", 320, 240)
  textAlign(CENTER)

Code principal :

r = 0  # Variables définies pour plus tard
v = 1
k = 320
score = 0

def setup(): #Setup
    size(640, 480) #Taille de la fenêtre
    fill(0, 4, 110) #Couleur du carré
    noStroke() #Pas de bordure
    frameRate(20) #Rafraîchissement
     
def draw(): 
    global r, v
    background(175, 213, 255) #Couleur du fond
    rect(k, r, 20, 20) #Carré de longueur 20 et de position k,r
    if r >= 460: #Empêche le carré de sortir par le bas
        v = 0
    r = r + v #Le carré descend de 1 (20 fois/secondes)
     
def keyPressed(): #Touches
    global k #Variable k
    if key == CODED:
        if keyCode == LEFT: #Le carré se déplace de 20 sur -x
            k = k - 20
        if keyCode == RIGHT: #Le carré se déplace de 20 sur x
            k = k + 20
    if k < 0: #Si le carré sors du cadre à gauche il se déplace à droite
        k = 620
    elif k > 620: #Si le carré sors du cadre à droite il se déplace à gauche
        k = 0

Et je veux que le premier code qui est le menu, quand je clique au lieu de changer mon rectangle, ça passe sur l'autre code. En gros, je veux que quand je clique le deuxièmes code démarre. Faudrait-t'il le mettre sous forme de fonction (sauf que je ne sais pas faire) ?

Pouvez-vous m'aider s'il vous plaît ?
Merci