#Traffic light LEDs using an array

from machine import Pin
import time

#######################    ADD BREAKOUT BUTTON #############################
# Configure the button on GPIO16
# PULL_UP means the pin reads HIGH (1) until the button connects it to GND
button = Pin(16, Pin.IN, Pin.PULL_UP)
#######################    ADD BREAKOUT BUTTON #############################


# Define GPIO pins for the LEDs
RED = Pin(2, Pin.OUT)
AMBER = Pin(6, Pin.OUT)
GREEN = Pin(10, Pin.OUT)

# Put them in an array
lights = [RED, AMBER, GREEN]

# Helper function to turn all lights off
def all_off():
    for light in lights:
        light.off()

while True:

    # Red
    all_off()
    RED.on()
    time.sleep(3)

    # Red + Amber
    AMBER.on()
    time.sleep(1)

    # Green
    all_off()
    GREEN.on()
    time.sleep(3)

    # Amber
    all_off()
    AMBER.on()
    time.sleep(1)

####################### CHECK BREAKOUT BUTTON ##########################
    # Read the button
    if button.value() == 0:   # Button pressed (LOW)
        print("Button pressed — stopping program.")
        all_off()
        break                 # Exit the loop and end main.py
####################### CHECK BREAKOUT BUTTON ##########################

# Code End