alien = Actor('alien', topleft=(20, 30))  #Load alien sprite
WIDTH = 900  # Set window size
HEIGHT = 600

RED = 200, 0, 0  # Set rectangle colour
BOX = Rect((450, 60), (50, 100))  # Set rectangle position and size

def draw():
      screen.fill((3, 70, 26))  # Background colour
      screen.draw.rect(BOX, RED)  # Draw rectangle
      alien.draw()  # Draw alien

def update():
    alien.angle -= 9  # Updates the position of the alien, in this case spins clockwise at a set speed

from pygame.transform import flip
alien = Actor('alien', topleft=(20, 30))                             # Load alien sprites
alien2 = Actor('alien', topright=(880, 30))
alien3 = Actor('alien', bottomleft=(20, 570))
alien4 = Actor('alien', bottomright=(880, 570))

WIDTH = 900                                                          # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

def draw():
    screen.fill((3, 70, 26))                                         # Set window background colour
    #screen.draw.rect(BOX, RED)                                       # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (100, 100)), 'gold')    # Draw squares with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (100, 100)), 'cyan')
    alien.draw()                                                     # Draws alien sprites
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)  # Mirrors/flips the alien image
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)

def update():
    alien.angle = 45

from pygame.transform import flip
alien = Actor('alien', topleft=(20, 30))                             # Load alien sprites
alien2 = Actor('alien', topright=(880, 30))
alien3 = Actor('alien', bottomleft=(20, 570))
alien4 = Actor('alien', bottomright=(880, 570))

WIDTH = 900                                                          # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

XY = (100, 100)

def draw():
    screen.fill((3, 70, 26))                                         # Set window background colour
    #screen.draw.rect(BOX, RED)                                       # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (XY)), 'gold')    # Draws rectangle with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (XY)), 'cyan')
    alien.draw()                                                     # Draws alien sprite
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)

alien = Actor("alien", topleft=(20, 30))
WIDTH = 900
HEIGHT = 600

RED = 200, 0, 0
BOX = Rect((450, 60), (50, 100))


def draw():
    screen.fill((3, 70, 26))
    screen.draw.rect(BOX, RED)
    alien.draw()


def update():
    #alien.angle = 45
    alien.angle -= 9

from pygame.transform import flip

alien = Actor("alien", topleft=(20, 30))                               # Load alien sprites
alien2 = Actor("alien", topright=(880, 30))
alien3 = Actor("alien", bottomleft=(20, 570))
alien4 = Actor("alien", bottomright=(880, 570))

WIDTH = 900                                                            # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

XY = (100, 100)

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    #screen.draw.rect(BOX, RED)                                       # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (XY)), "gold")             # Draws rectangle with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (XY)), "cyan")
    alien.draw()                                                       # Draws alien sprite
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)       # Flip and reposition alien sprite
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)
    #if on_mouse_down(pos):

def on_mouse_down(pos):
      screen.draw.text("hello", (300, 300))

from pygame.transform import flip

alien = Actor("alien", topleft=(20, 30))                               # Load alien sprites
alien2 = Actor("alien", topright=(880, 30))
alien3 = Actor("alien", bottomleft=(20, 570))
alien4 = Actor("alien", bottomright=(880, 570))

WIDTH = 900                                                            # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

XY = (100, 100)


def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    #screen.draw.rect(BOX, RED)                                       # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (XY)), "gold")             # Draws rectangle with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (XY)), "cyan")
    alien.draw()                                                       # Draws alien sprite
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)       # Flip and reposition alien sprite
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)


def on_mouse_down(pos):
    set_joke_display()

def set_joke_display():
    screen.draw.text("hello", (XY))

from pygame.transform import flip
import pygame
pygame.font.init

alien = Actor("alien", topleft=(20, 30))                               # Load alien sprites
alien2 = Actor("alien", topright=(880, 30))
alien3 = Actor("alien", bottomleft=(20, 570))
alien4 = Actor("alien", bottomright=(880, 570))

BLUE = (0, 0, 200)
font = pygame.font.SysFont(None, 24)
joke = font.render('hello', True, BLUE)

WIDTH = 900                                                            # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

XY = (100, 100)

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    #screen.draw.rect(BOX, RED)                                       # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (XY)), "gold")             # Draws rectangle with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (XY)), "cyan")
    alien.draw()                                                       # Draws alien sprite
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)       # Flip and reposition alien sprite
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)

def on_mouse_down(pos):
    screen.blit(joke, (300, 300))

#def set_joke_display():

from pygame.transform import flip
import pygame
pygame.font.init

alien = Actor("alien", topleft=(20, 30))                               # Load alien sprites
alien2 = Actor("alien", topright=(880, 30))
alien3 = Actor("alien", bottomleft=(20, 570))
alien4 = Actor("alien", bottomright=(880, 570))

BLUE = (0, 0, 200)

font = pygame.font.SysFont(None, 24)
joke = font.render('hello', True, BLUE)

WIDTH = 900                                                            # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

XY = (100, 100)

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    #screen.draw.rect(BOX, RED)                                       # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (XY)), "gold")             # Draws rectangle with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (XY)), "cyan")
    alien.draw()                                                       # Draws alien sprite
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)       # Flip and reposition alien sprite
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)

    if joke():
        screen.blit(joke, (300, 300))

def on_mouse_down():
    joke = True

#def on_mouse_down(pos):

#def set_joke_display():

from pygame.transform import flip

alien = Actor("alien", topleft=(20, 30))                               # Load alien sprites
alien2 = Actor("alien", topright=(880, 30))
alien3 = Actor("alien", bottomleft=(20, 570))
alien4 = Actor("alien", bottomright=(880, 570))

WIDTH = 900                                                            # Window size
HEIGHT = 600

# RED = 200, 0, 0                                                      # Rectangle colour
# BOX = Rect((450, 60), (50, 100))                                     # Recangle position and size

XY = (100, 100)

TELL_JOKE = False                                                      # Flag the output (Joke) as being not true

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    #screen.draw.rect(BOX, RED)                                        # Draws rectangle
    screen.draw.filled_rect(Rect((450, 60), (XY)), "gold")             # Draws rectangle with position size and colour
    screen.draw.filled_rect(Rect((250, 150), (XY)), "cyan")
    alien.draw()                                                       # Draws alien sprite
    screen.blit(flip(alien2._surf, True, False), alien2.topleft)       # Flip and reposition alien sprite
    alien3.draw()
    screen.blit(flip(alien4._surf, True, False), alien4.topleft)

    if TELL_JOKE:                                                      # Once the Joke state has changed to true in on_mouse_down
        screen.draw.text("hello", (300, 300))                          # Display the joke on screen

def on_mouse_down():                                               # When the mouse is clicked
    global TELL_JOKE                                                   # Make the operations happening to the Joke globally available
    TELL_JOKE = True                                                   # And changes the state of the joke from false to true

"""
The "You're Having a Laugh" joke generator
"""
from pygame.transform import flip
import random

laugh = Actor("haha", topleft=(50, 30))                                # Load images/buttons and set their positions
animal_button = Actor("button_animals", topleft=(100, 100))
food_button = Actor("button_food", topright=(800, 100))
silly_button = Actor("button_silly", bottomleft=(100, 500))
school_button = Actor("button_school", bottomright=(800, 500))

WIDTH = 900                                                            # Window size
HEIGHT = 600

print_joke = (450, 300)                                                # Joke screen position

AUDIENCE_LAUGH = sounds.laugh1                                         # Load laughter

#List of animal jokes
ANIMAL_JOKES = ["How do you tell the difference \
between a bull and a cow? \n\nIt is either one or \
the udder!", "Why do hummingbirds hum? \n\nBecause \
they don’t know the words!", "What kind of dog does \
a magician have? \n\nA Labracadabrador!", "What do \
frogs order at fast-food restaurants? \n\nFrench \
flies!"]

#KNOCK_KNOCK_JOKES = ["Knock, knock.\nWho’s there? \
#Dozen. \nDozen who? \nDozen anyone want to let me in?"]

#List of food jokes
FOOD_JOKES = ["Which vegetable do sailors hate the \
\n\nLeeks!", "What do you call a cheese that’s not \
yours? \n\nNacho cheese!", "What kind of key opens \
a banana? \n\nA mon-key!", "Why do bees have sticky \
haie? \n\nBecause they use honeycombs!"]

#List of silly jokes
SILLY_JOKES = ["If you’re English in the kitchen \
and English in the living room, what are you in the \
bathroom? \n\nEuropean!", "What are mummies \
favourite lunches? \n\nWraps!", "What gives you the \
power to walk through a wall? \n\nA door!", "What \
happens when you wear a snow suit inside? \n\nIt \
melts all over the carpet."]

#List of school jokes
SCHOOL_JOKES = ["Why did the music teacher need a \
ladder? \n\nTo reach the high notes.", "What do \
elves learn in school? \n\nThe elf-abet!", "Why did \
the nose not want to go to school? \n\nHe was tired \
of getting picked on!", "Why did the student eat \
his homework? \n\nBecause his teacher told him it \
would be a piece of cake!"]

rect1 = (100, 100)                                                     # Background shape sizes
rect2 = (65, 125)
rect3 = (35, 30)

ANIMAL_JOKE = False                                                    # Setting flags
FOOD_JOKE = False
SILLY_JOKE = False
SCHOOL_JOKE = False                                                    

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    screen.draw.filled_rect(Rect((50, 60), rect1), "gold")             # Draw rectangles
    screen.draw.filled_rect(Rect((40, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 145), rect3), "red")
    screen.draw.filled_rect(Rect((50, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((40, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 425), rect3), "red")
    #screen.blit(flip(alien2._surf, True, False), alien2.topleft)
    animal_button.draw()                                               # Draw pushbuttons
    food_button.draw()
    silly_button.draw()
    school_button.draw()

    if ANIMAL_JOKE:                                                      # Once the Animal Joke flag has changed to true in on_mouse_down
        laugh.draw()                                                     # Display the joke and laugh image on screen, play laughter
        screen.draw.text(random.choice(ANIMAL_JOKES), center=(print_joke))      
        AUDIENCE_LAUGH.play()

    elif FOOD_JOKE:                                                      # As per Animal flag
        laugh.draw()                                                   
        screen.draw.text(random.choice(FOOD_JOKES), center=(print_joke))      
        AUDIENCE_LAUGH.play()

    elif SILLY_JOKE:                                                      # As above
        laugh.draw()                                                    
        screen.draw.text(random.choice(SILLY_JOKES), center=(print_joke))      
        AUDIENCE_LAUGH.play()

    elif SCHOOL_JOKE:                                                     # As above
        laugh.draw()
        screen.draw.text(random.choice(SCHOOL_JOKES), center=(print_joke))
        AUDIENCE_LAUGH.play()


def on_mouse_down(pos, button):                                            # When the left mouse button is clicked on the animal button
    if button == mouse.LEFT and animal_button.collidepoint(pos):
        global ANIMAL_JOKE                                                 # Make the operations happening to the Joke flag available globally
        ANIMAL_JOKE = True                                                 # Set the animal joke flag to true

    elif button == mouse.LEFT and food_button.collidepoint(pos):           # Sets the Food joke flag to true
        global FOOD_JOKE
        FOOD_JOKE = True

    elif button == mouse.LEFT and silly_button.collidepoint(pos):          # Sets the silly joke flag to true
        global SILLY_JOKE
        SILLY_JOKE = True

    elif button == mouse.LEFT and school_button.collidepoint(pos):         # Sets the school joke glag to true
        global SCHOOL_JOKE
        SCHOOL_JOKE = True

"""
The "You're Having a Laugh" joke generator
"""
from pygame.transform import flip
import random

laugh = Actor("haha", topleft=(50, 30))                                # Load images/buttons and set their positions
animal_button = Actor("button_animals", topleft=(100, 85))
food_button = Actor("button_food", topright=(800, 85))
silly_button = Actor("button_silly", bottomleft=(100, 515))
school_button = Actor("button_school", bottomright=(800, 515))
reset_button = Actor("button_reset", midbottom=(450, 575))

WIDTH = 900                                                            # Window size
HEIGHT = 600

print_joke = (450, 300)                                                # Joke screen position

AUDIENCE_LAUGH = sounds.laugh1                                         # Load laughter

#List of animal jokes
ANIMAL_JOKES = ["How do you tell the difference \
between a bull and a cow? \n\nIt is either one or \
the udder!", "Why do hummingbirds hum? \n\nBecause \
they don’t know the words!", "What kind of dog does \
a magician have? \n\nA Labracadabrador!", "What do \
frogs order at fast-food restaurants? \n\nFrench \
flies!"]

#KNOCK_KNOCK_JOKES = ["Knock, knock.\nWho’s there? \
#Dozen. \nDozen who? \nDozen anyone want to let me in?"]

#List of food jokes
FOOD_JOKES = ["Which vegetable do sailors hate the \
\n\nLeeks!", "What do you call a cheese that’s not \
yours? \n\nNacho cheese!", "What kind of key opens \
a banana? \n\nA mon-key!", "Why do bees have sticky \
hair? \n\nBecause they use honeycombs!"]

#List of silly jokes
SILLY_JOKES = ["If you’re English in the kitchen \
and English in the living room, what are you in the \
bathroom? \n\nEuropean!", "What are mummies \
favourite lunches? \n\nWraps!", "What gives you the \
power to walk through a wall? \n\nA door!", "What \
happens when you wear a snow suit inside? \n\nIt \
melts all over the carpet."]

#List of school jokes
SCHOOL_JOKES = ["Why did the music teacher need a \
ladder? \n\nTo reach the high notes.", "What do \
elves learn in school? \n\nThe elf-abet!", "Why did \
the nose not want to go to school? \n\nHe was tired \
of getting picked on!", "Why did the student eat \
his homework? \n\nBecause his teacher told him it \
would be a piece of cake!"]

rect1 = (100, 100)                                                     # Background shape sizes
rect2 = (65, 125)
rect3 = (35, 30)

ANIMAL_JOKE = False                                                    # Setting flags
FOOD_JOKE = False
SILLY_JOKE = False
SCHOOL_JOKE = False

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    screen.draw.filled_rect(Rect((50, 60), rect1), "gold")             # Draw rectangles
    screen.draw.filled_rect(Rect((40, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 145), rect3), "red")
    screen.draw.filled_rect(Rect((50, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((40, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 425), rect3), "red")
    screen.draw.filled_rect(Rect((750, 60), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 145), rect3), "red")
    screen.draw.filled_rect(Rect((750, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 425), rect3), "red")
    #screen.blit(flip(alien2._surf, True, False), alien2.topleft)
    animal_button.draw()                                               # Draw pushbuttons
    food_button.draw()
    silly_button.draw()
    school_button.draw()

    if ANIMAL_JOKE:                                                      # Once the Animal Joke flag has changed to true in on_mouse_down
        category_selection()                                                     # Display the joke and laugh image on screen, play laughter

    elif FOOD_JOKE:                                                      # As per Animal flag
        category_selection()

    elif SILLY_JOKE:                                                      # As above
        category_selection()

    elif SCHOOL_JOKE:                                                     # As above
        category_selection()

    reset_button.draw()

def on_mouse_down(pos, button):
    global JOKE, SILLY_JOKE, ANIMAL_JOKE, FOOD_JOKE, SCHOOL_JOKE  # When the left mouse button is clicked on the animal button
    if button == mouse.LEFT and animal_button.collidepoint(pos):
        ANIMAL_JOKE = True
        JOKE = ANIMAL_JOKES  # Set the animal joke flag to true

    elif button == mouse.LEFT and food_button.collidepoint(pos):           # Sets the Food joke flag to true
        FOOD_JOKE = True
        JOKE = FOOD_JOKES

    elif button == mouse.LEFT and silly_button.collidepoint(pos):          # Sets the silly joke flag to true
        SILLY_JOKE = True
        JOKE = SILLY_JOKES

    elif button == mouse.LEFT and school_button.collidepoint(pos):         # Sets the school joke glag to true
        SCHOOL_JOKE = True
        JOKE = SCHOOL_JOKES

def category_selection():
    laugh.draw()
    screen.draw.text(random.choice(JOKE), center=(print_joke))
    AUDIENCE_LAUGH.play()

WIDTH = 900
HEIGHT = 600
"""
The "You're Having a Laugh" joke generator
"""
#from pygame.transform import flip
import random

laugh = Actor("haha", topleft=(50, 30))                                # Load images/buttons and set their positions
animal_button = Actor("button_animals", topleft=(100, 85))
food_button = Actor("button_food", topright=(800, 85))
silly_button = Actor("button_silly", bottomleft=(100, 515))
school_button = Actor("button_school", bottomright=(800, 515))
reset_button = Actor("button_reset", midbottom=(450, 575))

WIDTH = 900                                                            # Window size
HEIGHT = 600

print_joke = (450, 300)                                                # Joke screen position

audience_laugh = sounds.laugh1
#transparent = (0, 0, 0, 0)# Load laughter

#List of animal jokes
animal_jokes = ["How do you tell the difference \
between a bull and a cow? \n\nIt is either one or \
the udder!", "Why do hummingbirds hum? \n\nBecause \
they don’t know the words!", "What kind of dog does \
a magician have? \n\nA Labracadabrador!", "What do \
frogs order at fast-food restaurants? \n\nFrench \
flies!"]

#KNOCK_KNOCK_JOKES = ["Knock, knock.\nWho’s there? \
#Dozen. \nDozen who? \nDozen anyone want to let me in?"]

#List of food jokes
food_jokes = ["Which vegetable do sailors hate the \
\n\nLeeks!", "What do you call a cheese that’s not \
yours? \n\nNacho cheese!", "What kind of key opens \
a banana? \n\nA mon-key!", "Why do bees have sticky \
hair? \n\nBecause they use honeycombs!"]

#List of silly jokes
silly_jokes = ["If you’re English in the kitchen \
and English in the living room, what are you in the \
bathroom? \n\nEuropean!", "What are mummies \
favourite lunches? \n\nWraps!", "What gives you the \
power to walk through a wall? \n\nA door!", "What \
happens when you wear a snow suit inside? \n\nIt \
melts all over the carpet."]

#List of school jokes
school_jokes = ["Why did the music teacher need a \
ladder? \n\nTo reach the high notes.", "What do \
elves learn in school? \n\nThe elf-abet!", "Why did \
the nose not want to go to school? \n\nHe was tired \
of getting picked on!", "Why did the student eat \
his homework? \n\nBecause his teacher told him it \
would be a piece of cake!"]

rect1 = (100, 100)                                                     # Background shape sizes
rect2 = (65, 125)
rect3 = (35, 30)

animal_joke = False                                                    # Setting flags
food_joke = False
silly_joke = False
school_joke = False
start = True

def tell_a_joke():
    start = False
    draw()

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    screen.draw.filled_rect(Rect((50, 60), rect1), "gold")             # Draw rectangles
    screen.draw.filled_rect(Rect((40, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 145), rect3), "red")
    screen.draw.filled_rect(Rect((50, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((40, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 425), rect3), "red")
    screen.draw.filled_rect(Rect((750, 60), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 145), rect3), "red")
    screen.draw.filled_rect(Rect((750, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 425), rect3), "red")
    #screen.blit(flip(alien2._surf, True, False), alien2.topleft)
    animal_button.draw()                                               # Draw pushbuttons
    food_button.draw()
    silly_button.draw()
    school_button.draw()
    Tell_Joke()

def on_mouse_down(button, pos):
    global JOKE, animal_joke, food_joke, silly_joke, school_joke # When the left mouse button is clicked on the animal button
    if button == mouse.LEFT and animal_button.collidepoint(pos):
        animal_joke = True
        JOKE = animal_jokes  # Set the animal joke flag to true

    elif button == mouse.LEFT and food_button.collidepoint(pos):           # Sets the Food joke flag to true
        food_joke = True
        JOKE = food_jokes

    elif button == mouse.LEFT and silly_button.collidepoint(pos):          # Sets the silly joke flag to true
        silly_joke = True
        JOKE = silly_jokes

    elif button == mouse.LEFT and school_button.collidepoint(pos):         # Sets the school joke flag to true
        school_joke = True
        JOKE = school_jokes

    elif button == mouse.LEFT and reset_button.collidepoint(pos):
        restart()

def Tell_Joke():
    if animal_joke:                                                      # Once the Animal Joke flag has changed to true in on_mouse_down
        selected_joke()                                                     # Display the joke and laugh image on screen, play laughter

    elif food_joke:                                                      # As per Animal flag
        selected_joke()

    elif silly_joke:                                                      # As above
        selected_joke()

    elif school_joke:                                                     # As above
        selected_joke()


def selected_joke():
    laugh.draw()
    screen.draw.text(random.choice(JOKE), center=(print_joke))
    audience_laugh.play()
    reset_button.draw()
    restart()

def restart():
    global start
    start = True

"""
The "You're Having a Laugh" joke generator
"""
#from pygame.transform import flip
import random

laugh = Actor("haha", topleft=(50, 30))                                # Load images/buttons and set their positions
animal_button = Actor("button_animals", topleft=(100, 85))
food_button = Actor("button_food", topright=(800, 85))
silly_button = Actor("button_silly", bottomleft=(100, 515))
school_button = Actor("button_school", bottomright=(800, 515))
reset_button = Actor("button_reset", midbottom=(450, 575))

WIDTH = 900                                                            # Window size
HEIGHT = 600

print_joke = (450, 300)                                                # Joke screen position

audience_laugh = sounds.laugh1                                         # Load laughter

#List of animal jokes
animal_jokes = ["How do you tell the difference \
between a bull and a cow? \n\nIt is either one or \
the udder!", "Why do hummingbirds hum? \n\nBecause \
they don’t know the words!", "What kind of dog does \
a magician have? \n\nA Labracadabrador!", "What do \
frogs order at fast-food restaurants? \n\nFrench \
flies!"]

#KNOCK_KNOCK_JOKES = ["Knock, knock.\nWho’s there? \
#Dozen. \nDozen who? \nDozen anyone want to let me in?"]

#List of food jokes
food_jokes = ["Which vegetable do sailors hate the \
\n\nLeeks!", "What do you call a cheese that’s not \
yours? \n\nNacho cheese!", "What kind of key opens \
a banana? \n\nA mon-key!", "Why do bees have sticky \
hair? \n\nBecause they use honeycombs!"]

#List of silly jokes
silly_jokes = ["If you’re English in the kitchen \
and English in the living room, what are you in the \
bathroom? \n\nEuropean!", "What are mummies \
favourite lunches? \n\nWraps!", "What gives you the \
power to walk through a wall? \n\nA door!", "What \
happens when you wear a snow suit inside? \n\nIt \
melts all over the carpet."]

#List of school jokes
school_jokes = ["Why did the music teacher need a \
ladder? \n\nTo reach the high notes.", "What do \
elves learn in school? \n\nThe elf-abet!", "Why did \
the nose not want to go to school? \n\nHe was tired \
of getting picked on!", "Why did the student eat \
his homework? \n\nBecause his teacher told him it \
would be a piece of cake!"]

rect1 = (100, 100)                                                     # Background shape sizes
rect2 = (65, 125)
rect3 = (35, 30)

animal_joke = False                                                    # Setting flags
food_joke = False
silly_joke = False
school_joke = False

def draw():
    screen.fill((3, 70, 26))                                           # Set window background colour
    screen.draw.filled_rect(Rect((50, 60), rect1), "gold")             # Draw rectangles for background
    screen.draw.filled_rect(Rect((40, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 145), rect3), "red")
    screen.draw.filled_rect(Rect((50, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((40, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 425), rect3), "red")
    screen.draw.filled_rect(Rect((750, 60), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 145), rect3), "red")
    screen.draw.filled_rect(Rect((750, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 425), rect3), "red")
    #screen.blit(flip(alien2._surf, True, False), alien2.topleft)
    animal_button.draw()                                               # Draw pushbuttons
    food_button.draw()
    silly_button.draw()
    school_button.draw()

    if animal_joke:                                                    # Once the Animal Joke flag has changed to true in on_mouse_down
        category_selection()                                           # Run the category_selection function

    elif food_joke:                                                    # As per Animal flag
        category_selection()

    elif silly_joke:                                                   # As above
        category_selection()

    elif school_joke:                                                  # As above
        category_selection()

def on_mouse_down(pos, button):
    global JOKE, animal_joke, food_joke, silly_joke, school_joke       # When the left mouse button is clicked on the animal button
    if button == mouse.LEFT and animal_button.collidepoint(pos):       # Set the animal joke flag to true
        animal_joke = True
        JOKE = animal_jokes  

    elif button == mouse.LEFT and food_button.collidepoint(pos):       # Sets the Food joke flag to true
        food_joke = True
        JOKE = food_jokes

    elif button == mouse.LEFT and silly_button.collidepoint(pos):      # Sets the silly joke flag to true
        silly_joke = True
        JOKE = silly_jokes

    elif button == mouse.LEFT and school_button.collidepoint(pos):     # Sets the school joke flag to true
        school_joke = True
        JOKE = school_jokes

def category_selection():                                              # Tell the joke
    laugh.draw()
    screen.draw.text(random.choice(JOKE), center=(print_joke))         # Pics a random joke from the category selected by the user 
    audience_laugh.play()
    reset_button.draw()                                                # To be used for starting over

def my_function():
    """
    This is where I write my comments for the function.

    It can be over several lines.
    """
    return "Hello"

my_jokes = [
    ["setup 1", "punchline 1"],
    ["setup 2", "punchline 2"],
    # ... etc ...
]

my_other_jokes = [
    ["setup 3", "punchline 3"],
    ["setup 4", "punchline 4"],
    # ... etc ...
]

the_joke = "\n\n".join(random.choice(my_jokes))

if animal_joke:
    category_selection()

elif food_joke:
    category_selection()

elif silly_joke:
    category_selection()

elif school_joke:
    category_selection()

if tell_joke:
        laugh.draw()
        bean.draw()
        screen.draw.text("\n\n".join(random.choice(JOKE)), center=(print_joke), fontname="rabbit_on_the_moon", fontsize=36,
        color="#AAFF00")  #, gcolor="#66AA00", owidth=1.5, ocolor="black", alpha=0.8)
        audience_laugh.play()
        reset_button.draw()

pygame 2.1.2 (SDL 2.0.18, Python 3.8.12)
Hello from the pygame community. https://www.pygame.org/contribute.html
Traceback (most recent call last):
  File "C:\Local\Programs\MUEDIT~1\Python\lib\runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "C:\Local\Programs\MUEDIT~1\Python\lib\runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\__main__.py", line 3, in <module>
    main()
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\runner.py", line 93, in main
    run_mod(mod)
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\runner.py", line 113, in run_mod
    PGZeroGame(mod).run()
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\game.py", line 217, in run
    self.mainloop()
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\game.py", line 256, in mainloop
    draw()
  File "paul honey grade 1 - joke generator.py", line 97, in draw
    screen.draw.text("\n\n".join(random.choice(JOKE)), center=(print_joke), fontname="rabbit_on_the_moon", fontsize=36,
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\screen.py", line 61, in text
    ptext.draw(*args, surf=self._surf, **kwargs)
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\ptext.py", line 439, in draw
    tsurf = getsurf(text, fontname, fontsize, sysfontname, bold, italic, underline, width, widthem,
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\ptext.py", line 280, in getsurf
    surf0 = getsurf(text, fontname, fontsize, sysfontname, bold, italic, underline,
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\ptext.py", line 311, in getsurf
    surf0 = getsurf(text, fontname, fontsize, sysfontname, bold, italic, underline,
  File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\ptext.py", line 349, in getsurf
    (1.0 - m) * array[:, :, j] + m * gcolor[j]).astype(array.dtype)
ValueError: operands could not be broadcast together with shapes (50,) (0,49) 

"""
The "You're Having a Laugh" joke generator
"""
# from pygame.transform import flip
import random

laugh = Actor("haha", topleft=(50, 30))  # Load images/buttons and set their positions
bean = Actor("bean", bottomright=(900, 600))
animal_button = Actor("button_animals", topleft=(100, 85))
food_button = Actor("button_food", topright=(800, 85))
silly_button = Actor("button_silly", bottomleft=(100, 515))
school_button = Actor("button_school", bottomright=(800, 515))
reset_button = Actor("button_reset", midbottom=(450, 575))

WIDTH = 900  # Window size
HEIGHT = 600

print_joke = (450, 300)  # Joke screen position

audience_laugh = sounds.laugh1  # Load laughter

animal_jokes = [
    [
        "How do you tell the difference between a bull and a cow?",
        "It is either one or the udder!",
    ],
    ["Why do hummingbirds hum?", "Because they don’t know the words!"],
    ["What kind of dog does a magician have?", "A Labracadabrador!"],
    ["What do frogs order at fast-food restaurants", "French flies!"],
]

# KNOCK_KNOCK_JOKES = ["Knock, knock.\nWho’s there? \
# Dozen. \nDozen who? \nDozen anyone want to let me in?"]

food_jokes = [
    ["Which vegetable do sailors hate the most?", "Leeks!"],
    ["What do you call a cheese that’s not yours?" "Nacho cheese!"],
    ["What kind of key opens a banana?", "A mon-key!"],
    ["Why do bees have sticky hair?, Because they use honeycombs!"],
]

silly_jokes = [
    [
        "If you’re English in the kitchen and English in the living room, \n\nwhat are you in the bathroom?",
        "European!",
    ],
    ["What are mummies favourite lunches?", "Wraps!"],
    ["What gives you the power to walk through a wall?", "A door!"],
    ["What happens when you wear a snow suit inside?", "It melts all over the carpet."],
]

# List of school jokes
school_jokes = [
    ["Why did the music teacher need a ladder?", "To reach the high notes."],
    ["What do elves learn in school?", "The elf-abet!"],
    [
        "Why did the nose not want to go to school?",
        "He was tired of getting picked on!",
    ],
    [
        "Why did the student eat his homework?",
        "Because his teacher told him it would be a piece of cake!",
    ],
]

rect1 = (100, 100)  # Background shape sizes
rect2 = (65, 125)
rect3 = (35, 30)

tell_joke = False

def draw():
    screen.fill((3, 70, 26))  # Set window background colour
    screen.draw.filled_rect(
        Rect((50, 60), rect1), "gold"
    )  # Draw rectangles for background
    screen.draw.filled_rect(Rect((40, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 145), rect3), "red")
    screen.draw.filled_rect(Rect((50, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((40, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((95, 425), rect3), "red")
    screen.draw.filled_rect(Rect((750, 60), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 75), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 145), rect3), "red")
    screen.draw.filled_rect(Rect((750, 440), rect1), "gold")
    screen.draw.filled_rect(Rect((795, 400), rect2), "cyan")
    screen.draw.filled_rect(Rect((770, 425), rect3), "red")
    # screen.blit(flip(alien2._surf, True, False), alien2.topleft)
    animal_button.draw()  # Draw pushbuttons
    food_button.draw()
    silly_button.draw()
    school_button.draw()

    if tell_joke:
        laugh.draw()
        bean.draw()
        screen.draw.text("\n\n".join(random.choice(JOKE)), center=(print_joke), fontname="rabbit_on_the_moon", fontsize=36,
        color="#AAFF00", owidth=1.5, ocolor="black", alpha=0.8)
        audience_laugh.play()
        reset_button.draw()  # Run the category_selection function


def on_mouse_down(pos, button):
    """
    click a button to select a joke category
    """
    global JOKE, tell_joke  # food_joke, silly_joke, school_joke
    if button == mouse.LEFT and animal_button.collidepoint(pos):
        tell_joke = True
        JOKE = animal_jokes

    elif button == mouse.LEFT and food_button.collidepoint(pos):
        tell_joke = True
        JOKE = food_jokes

    elif button == mouse.LEFT and silly_button.collidepoint(pos):
        tell_joke = True
        JOKE = silly_jokes

    elif button == mouse.LEFT and school_button.collidepoint(pos):
        tell_joke = True
        JOKE = school_jokes

    elif button == mouse.LEFT and reset_button.collidepoint(pos):
        tell_joke = ""

KeyError: "No image found like 'bean'. Are you sure the image exists?"

 File "C:\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\ptext.py", line 349, in getsurf
    (1.0 - m) * array[:, :, j] + m * gcolor[j]).astype(array.dtype)
ValueError: operands could not be broadcast together with shapes (50,) (0,49)

"""
The "You're Having a Laugh" Joke Generator
"""

import random

WIDTH = 900  # Window size
HEIGHT = 600

stage = Actor("stage", midtop=(450, 0))     # Background images
neon_mic = Actor("neon_mic", topleft=(0, 0))
mic = Actor("micstand", center=(430, 375))

laugh = Actor("haha", center=(450, 300))    # Randomly generated images
bean = Actor("bean", bottomright=(900, 600))
lol = Actor("lol", center=(375, 300))
rofl = Actor("rofl", center=(430, 400))
funny_image = [laugh, bean, lol, rofl]

audience_laugh = sounds.laugh1      # Randomly generated sounds
girl_laugh = sounds.laugh2
badum_tish = sounds.drum
slow_clap = sounds.slow_clap
funny_sound = [audience_laugh, girl_laugh, badum_tish, slow_clap]

"""
Section 1 - Joke Lists
Add your own!
"""
animal_jokes = [
    [
        "How do you tell the difference between a bull and a cow?",
        "It is either one or the udder!"
    ],
    [
        "Why do hummingbirds hum?",
        "Because they don't know the words!"
    ],
    [
        "What kind of dog does a magician have?",
        "A Labracadabrador!"
    ],
    [
        "What do frogs order at fast-food restaurants",
        "French flies!"
    ],
]

food_jokes = [
    [
        "Which vegetable do sailors hate the most?",
        "Leeks!"
    ],
    [
        "What do you call a cheese that's not yours?",
        "Nacho cheese!"
    ],
    [
        "What kind of key opens a banana?",
        "A mon-key!"
    ],
    [
        "Why do bees have sticky hair?",
        "Because they use honeycombs!"
    ],
]

silly_jokes = [
    [
        "If you're English in the kitchen and English in the living room,\
        \n\nwhat are you in the bathroom?",
        "European!"
    ],
    [
        "What are mummies favourite lunches?",
        "Wraps!"
    ],
    [
        "What gives you the power to walk through a wall?",
        "A door!"
    ],
    [
        "What happens when you wear a snow suit inside?",
        "It melts all over the carpet."
    ],
]

school_jokes = [
    [
        "Why did the music teacher need a ladder?",
        "To reach the high notes."
    ],
    [
        "What do elves learn in school?",
        "The elf-abet!"
    ],
    [
        "Why did the nose not want to go to school?",
        "He was tired of getting picked on!"
    ],
    [
        "Why did the student eat his homework?",
        "Because his teacher told him it would be a piece of cake!"
    ],
]

"""
Section 2 - Buttons list
Can you think of your own categories?
"""
top_left_button = Actor("button_animals", topleft=(100, 150))
top_right_button = Actor("button_food", topright=(800, 150))
bottom_left_button = Actor("button_silly", bottomleft=(100, 450))
bottom_right_button = Actor("button_school", bottomright=(800, 450))
reset_button = Actor("button_reset", midbottom=(450, 575))
buttons = [top_left_button, top_right_button, bottom_left_button, bottom_right_button]

joke = False

def draw():
    stage.draw()
    neon_mic.draw()
    mic.draw()
    buttons[0].draw()
    buttons[1].draw()
    buttons[2].draw()
    buttons[3].draw()

    if joke:
        """
        After joke category is selected,
        display random joke and images,
        play random sound.
        """
        draw_image = random.choice(funny_image)
        draw_image.draw()
        screen.draw.text(
            "\n\n".join(random.choice(joke)),
            center=(450, 300),
            fontname="rabbit_on_the_moon",
            fontsize=36,
            color="darkorange1",
            owidth=1.5,
            ocolor="black"
        )
        play_sound = random.choice(funny_sound)
        play_sound.play()
        reset_button.draw()

def on_mouse_down(pos, button):
    """
    click a button to select a joke category
    """
    global joke
    if button == mouse.LEFT and buttons[0].collidepoint(pos):
        joke = animal_jokes

    elif button == mouse.LEFT and buttons[1].collidepoint(pos):
        joke = food_jokes

    elif button == mouse.LEFT and buttons[2].collidepoint(pos):
        joke = silly_jokes

    elif button == mouse.LEFT and buttons[3].collidepoint(pos):
        joke = school_jokes

    elif button == mouse.LEFT and reset_button.collidepoint(pos):
        joke = ""      

[
        "How do you tell the difference between a bull and a cow?",
        "It is either one or the udder!"
    ],

[
        "How do you tell the difference between a bull and a cow?",
        "It is either one or the udder!",
    ],
    ["Why do hummingbirds hum?", "Because they don't know the words!"],

pygame 2.1.2 (SDL 2.0.18, Python 3.8.12)
Hello from the pygame community. https://www.pygame.org/contribute.html
Traceback (most recent call last):
  File "C:\Users\406850\AppData\Local\Programs\MUEDIT~1\Python\lib\runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "C:\Users\406850\AppData\Local\Programs\MUEDIT~1\Python\lib\runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "C:\Users\406850\AppData\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\__main__.py", line 3, in <module>
    main()
  File "C:\Users\406850\AppData\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\runner.py", line 93, in main
    run_mod(mod)
  File "C:\Users\406850\AppData\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\runner.py", line 113, in run_mod
    PGZeroGame(mod).run()
  File "C:\Users\406850\AppData\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\game.py", line 217, in run
    self.mainloop()
  File "C:\Users\406850\AppData\Local\python\mu\mu_venv-38-20221122-201713\lib\site-packages\pgzero\game.py", line 256, in mainloop
    draw()
  File "paul honey grade 1 - joke generator.py", line 124, in draw
    buttons[0, 1, 2, 3].draw()
TypeError: list indices must be integers or slices, not tuple


---------- FINISHED ----------
exit code: 1 status: 0

draw_image = random.choice(funny_image)
draw_image.draw()

def draw
      [backgrounds.draw() for backgrounds in background_images]
      [joke_categories.draw() for joke_categories in buttons]

def draw
      [for backgrounds in background_images backgrounds.draw()]

    [for backgrounds in background_images backgrounds.draw()]
     ^
SyntaxError: invalid syntax

"""
The "You're Having a Laugh" Joke Generator
"""

import random

WIDTH = 900  # Window size
HEIGHT = 600

stage = Actor("stage", midtop=(450, 0))     # Background images
neon_mic = Actor("neon_mic", topleft=(0, 195))
mic = Actor("micstand", center=(430, 375))
comedy = Actor("comedy", center=(450, 70))
laugh_emoji = Actor("laugh_emoji", midright=(880, 280))
laugh_emoji.angle = -30
background_images = [stage, comedy, neon_mic, mic, laugh_emoji]

laugh = Actor("hahaha", center=(450, 300))    # Randomly generated images
haha = Actor("haha", center=(400,300))
bean = Actor("bean", bottomright=(900, 600))
lol = Actor("lol", center=(375, 300))
rofl = Actor("rofl", center=(430, 400))
funny_image = [laugh, bean, lol, rofl, haha]

audience_laugh = sounds.laugh1      # Randomly generated sounds
girl_laugh = sounds.laugh2
badum_tish = sounds.drum
slow_clap = sounds.slow_clap
funny_sound = [audience_laugh, girl_laugh, badum_tish, slow_clap]

"""
Section 1 - Joke Lists
Add your own!
"""
animal_jokes = [
    [
        "How do you tell the difference between a bull and a cow?",
        "It is either one or the udder!"
    ],
    [
        "Why do hummingbirds hum?",
        "Because they don't know the words!"
    ],
    [
        "What kind of dog does a magician have?",
        "A Labracadabrador!"
    ],
    [
        "What do frogs order at fast-food restaurants",
        "French flies!"
    ],
]

food_jokes = [
    [
        "Which vegetable do sailors hate the most?",
        "Leeks!"
    ],
    [
        "What do you call a cheese that's not yours?",
        "Nacho cheese!"
    ],
    [
        "What kind of key opens a banana?",
        "A mon-key!"
    ],
    [
        "Why do bees have sticky hair?",
        "Because they use honeycombs!"
    ],
]

silly_jokes = [
    [
        "If you're English in the kitchen and English in the living room,\
        \n\nwhat are you in the bathroom?",
        "European!"
    ],
    [
        "What are mummies favourite lunches?",
        "Wraps!"
    ],
    [
        "What gives you the power to walk through a wall?",
        "A door!"
    ],
    [
        "What happens when you wear a snow suit inside?",
        "It melts all over the carpet."
    ],
]

school_jokes = [
    [
        "Why did the music teacher need a ladder?",
        "To reach the high notes."
    ],
    [
        "What do elves learn in school?",
        "The elf-abet!"
    ],
    [
        "Why did the nose not want to go to school?",
        "He was tired of getting picked on!"
    ],
    [
        "Why did the student eat his homework?",
        "Because his teacher told him it would be a piece of cake!"
    ],
]

"""
Section 2 - Buttons list
Can you think of your own categories?
"""
top_left_button = Actor("button_animals", topleft=(100, 150))
top_right_button = Actor("button_food", topright=(800, 150))
bottom_left_button = Actor("button_silly", bottomleft=(100, 450))
bottom_right_button = Actor("button_school", bottomright=(800, 450))
reset_button = Actor("button_reset", midbottom=(450, 575))
buttons = [top_left_button, top_right_button, bottom_left_button, bottom_right_button]

joke = False

def draw():
    [backgrounds.draw() for backgrounds in background_images]
    [joke_categories.draw() for joke_categories in buttons]

    if joke:
        """
        After joke category is selected,
        display random joke and images,
        play random sound.
        """
        draw_image = random.choice(funny_image)
        draw_image.draw()
        screen.draw.text(
            "\n\n".join(random.choice(joke)),
            center=(450, 300),
            fontname="rabbit_on_the_moon",
            fontsize=36,
            color="darkorange1",
            owidth=1.5,
            ocolor="black"
        )
        play_sound = random.choice(funny_sound)
        play_sound.play()
        reset_button.draw()

def on_mouse_down(pos, button):
    """
    click a button to select a joke category
    """
    global joke
    if button == mouse.LEFT and buttons[0].collidepoint(pos):
        joke = animal_jokes

    elif button == mouse.LEFT and buttons[1].collidepoint(pos):
        joke = food_jokes

    elif button == mouse.LEFT and buttons[2].collidepoint(pos):
        joke = silly_jokes

    elif button == mouse.LEFT and buttons[3].collidepoint(pos):
        joke = school_jokes

    elif button == mouse.LEFT and reset_button.collidepoint(pos):
        joke = ""

new_list = [do_something(x) for x in some_other_list]

"""
The "You're Having a Laugh" Joke Generator
"""

import random

WIDTH = 900  # Window size
HEIGHT = 600

joke = False

"""
Section 1 - Backgrounds, popups and sounds
"""

stage = Actor("stage", midtop=(450, 0))     # Background images
neon_mic = Actor("neon_mic", topleft=(0, 195))
mic = Actor("micstand", center=(430, 375))
comedy = Actor("comedy", center=(450, 70))
laugh_emoji = Actor("laugh_emoji", midright=(880, 280))
laugh_emoji.angle = -30
background_images = [stage, comedy, neon_mic, mic, laugh_emoji]

raspberry = Actor("raspberry", topright=(485, 125))    # Randomly generated images
haha = Actor("haha", center=(400,300))
bean = Actor("bean", bottomright=(900, 600))
lol = Actor("lol", center=(375, 300))
rofl = Actor("rofl", center=(430, 400))
funny_image = [raspberry, haha, bean, lol, rofl]

audience_laugh = sounds.laugh1      # Randomly generated sounds
girl_laugh = sounds.laugh2
badum_tish = sounds.drum
slow_clap = sounds.slow_clap
funny_sound = [audience_laugh, girl_laugh, badum_tish, slow_clap]

"""
Section 2 - Joke Lists
Add your own!
"""
animal_jokes = [
    [
        "How do you tell the difference between a bull and a cow?",
        "It is either one or the udder!"
    ],
    [
        "Why do hummingbirds hum?",
        "Because they don't know the words!"
    ],
    [
        "What kind of dog does a magician have?",
        "A Labracadabrador!"
    ],
    [
        "What do frogs order at fast-food restaurants",
        "French flies!"
    ],
]

food_jokes = [
    [
        "Which vegetable do sailors hate the most?",
        "Leeks!"
    ],
    [
        "What do you call a cheese that's not yours?",
        "Nacho cheese!"
    ],
    [
        "What kind of key opens a banana?",
        "A mon-key!"
    ],
    [
        "Why do bees have sticky hair?",
        "Because they use honeycombs!"
    ],
]

silly_jokes = [
    [
        "If you're English in the kitchen and English in the living room,\
        \n\nwhat are you in the bathroom?",
        "European!"
    ],
    [
        "What are mummies favourite lunches?",
        "Wraps!"
    ],
    [
        "What gives you the power to walk through a wall?",
        "A door!"
    ],
    [
        "What happens when you wear a snow suit inside?",
        "It melts all over the carpet."
    ],
]

school_jokes = [
    [
        "Why did the music teacher need a ladder?",
        "To reach the high notes."
    ],
    [
        "What do elves learn in school?",
        "The elf-abet!"
    ],
    [
        "Why did the nose not want to go to school?",
        "He was tired of getting picked on!"
    ],
    [
        "Why did the student eat his homework?",
        "Because his teacher told him it would be a piece of cake!"
    ],
]

"""
Section 3 - Buttons list
Can you think of your own categories?
"""
top_left_button = Actor("button_animals", topleft=(100, 150))
top_right_button = Actor("button_food", topright=(800, 150))
bottom_left_button = Actor("button_silly", bottomleft=(100, 450))
bottom_right_button = Actor("button_school", bottomright=(800, 450))
reset_button = Actor("button_reset", midbottom=(450, 575))
buttons = [top_left_button, top_right_button, bottom_left_button, bottom_right_button]

"""
Section 4 - The code which tells your jokes
"""

def draw():
    [backgrounds.draw() for backgrounds in background_images]
    [joke_categories.draw() for joke_categories in buttons]

    if joke:
        """
        After joke category is selected,
        display random joke and images,
        play random sound.
        """
        draw_image = random.choice(funny_image)
        draw_image.draw()
        screen.draw.text(
            "\n\n".join(random.choice(joke)),
            center=(450, 300),
            fontname="rabbit_on_the_moon",
            fontsize=36,
            color="darkorange1",
            owidth=1.5,
            ocolor="black"
        )
        play_sound = random.choice(funny_sound)
        play_sound.play()
        reset_button.draw()

def on_mouse_down(pos, button):
    """
    click a button to select a joke category
    """
    global joke
    if button == mouse.LEFT and buttons[0].collidepoint(pos):
        joke = animal_jokes

    elif button == mouse.LEFT and buttons[1].collidepoint(pos):
        joke = food_jokes

    elif button == mouse.LEFT and buttons[2].collidepoint(pos):
        joke = silly_jokes

    elif button == mouse.LEFT and buttons[3].collidepoint(pos):
        joke = school_jokes

    elif button == mouse.LEFT and reset_button.collidepoint(pos):
        joke = ""