import board, time, neopixel, digitalio
from digitalio import DigitalInOut

# Define Strip
strip_pin = board.A1
strip_num_of_lights = 30
strip = neopixel.NeoPixel(strip_pin, strip_num_of_lights, brightness=0.5, auto_write=True)

# Define Buttons & Locations
button_A = digitalio.DigitalInOut(board.BUTTON_A)
button_A.switch_to_input(pull = digitalio.Pull.DOWN)
button_B = digitalio.DigitalInOut(board.BUTTON_B)
button_B.switch_to_input(pull = digitalio.Pull.DOWN)

lights = False

# While True loop
while True:
    if lights = False:
        if button_A.value:
            while button_A.value:
                pass
            strip.fill(255, 194, 0)
            lights = True
        elif button_B.value:
            while button_B.value:
                pass
            strip.fill(255, 255, 255)
            lights = True
    elif lights = True:
        if button_A.value:
            strip.fill(0, 0, 0)
        elif button_B.value:
            strip.fill(0, 0, 0)