''' 
Mission 8 -- Answer Bot
With 2 extensions and 3 challenges
'''


from codex import *
import random
from time import sleep


answers = ["The odds are yes",
           "It is not in the stars",
           "Go for the gold!",
           "Today is your day",
           "Stay home and read",
           "Have an adventure", 
           pics.HEART,      # Challenge #2 - adding pics
           pics.TIARA, 
           pics.PLANE,
           pics.HAPPY ]


# Challenge #3 -- adding JPG images list
images_JPG = ["pics/teacherBear.jpg", 
              "pics/doggie.jpg", 
              "pics/goldfish.jpg"]
  
# Extension #2 -- function for pixels
def display_pixels():
    color = random.choice(COLOR_LIST)
    pixels.set(0, color)
    color = random.choice(COLOR_LIST)
    pixels.set(1, color)
    color = random.choice(COLOR_LIST)
    pixels.set(2, color)
    color = random.choice(COLOR_LIST)
    pixels.set(3, color)


# Challenge #1 -- using a loop for pixels
def display_pixels2():
    counter = 0
    while counter < 3:
        color = random.choice(COLOR_LIST)
        pixels.set(counter, color)
        counter = counter + 1   




while True:
    # Call function from extension 2 or challenge 1
    display_pixels2()  


    if buttons.was_pressed(BTN_A):
        display.clear()
        x = random.choice(answers)
        # Condition for Challenge 2
        if type(x) == str:
            display.print(x)
        else:
            display.show(x)


    sleep(2)


    # If statement for Challenge 3
    if buttons.was_pressed(BTN_B):
        x = random.choice(images_JPG)
        display.draw_jpg(x)


    # Extension #1 -- kill switch
    if buttons.was_pressed(BTN_U):
        break


display.fill(BLACK)