from codex import *
from time import sleep

# declare variable delay
delay = 1.0

# declare image list variable
a_list = [
    pics.HAPPY,
    "Player A",
    pics.SAD,
    pics.HEART,
    pics.HEART_SMALL,
    BLUE,
    pics.ASLEEP,
    pics.SURPRISED
]

b_list = [
    pics.TIARA,
    pics.TSHIRT,
    "Player B",
    pics.PLANE,
    pics.HOUSE,
    pics.MUSIC,
    GREEN,
    pics.TARGET
]

# declare variable choice
choice = 0
set_list = "a"
LAST_INDEX = len(a_list) - 1

# Intro display message
display.show("Player A or Player B")
sleep(2)

while True:
    if buttons.was_pressed(BTN_A):
        set_list = "a"
        leds.set(LED_A, True)
        leds.set(LED_B, False)
        pixels.set(0, BLUE)
        pixels.set(1, YELLOW)
        pixels.set(2, DARK_GREEN)
        pixels.set(3, WHITE)
    
    if buttons.was_pressed(BTN_B):
        set_list = "b" 
        leds.set(LED_B, True)
        leds.set(LED_A, False)
        pixels.set(2, GREEN)
        pixels.set(3, GREEN)
        pixels.set(0, RED)
        pixels.set(1, RED)

    if set_list == "a":
        my_image = a_list[choice]
    else:
        my_image = b_list[choice]

    if type(my_image) == tuple:
        display.fill(my_image)
    else:
        display.show(my_image)

    if buttons.was_pressed(BTN_R):
        choice = choice + 1
        if choice > LAST_INDEX:
            choice = 0

    if buttons.was_pressed(BTN_L):
        choice = choice - 1
        if choice < 0:
            choice = LAST_INDEX
            
    if buttons.was_pressed(BTN_D):
        break

display.fill(BLACK)
display.show("Good-bye")