import time
import machine

time.sleep(5)

print("Approaching Traffic Light")

LED_PIN1 = 0
BUTTON_PIN = 14
LED_PIN2 = 1
LED_PIN3 = 2

led1 = machine.Pin(LED_PIN1, machine.Pin.OUT)
led2 = machine.Pin(LED_PIN2, machine.Pin.OUT)
led3 = machine.Pin(LED_PIN3, machine.Pin.OUT)
button = machine.Pin(BUTTON_PIN, machine.Pin.IN, machine.Pin.PULL_UP)

while True:
    if not button.value():
        led1.on()
        time.sleep(0.1)
        print('GO')
        led2.off()
        time.sleep(0.5)
        led2.off

        led3.off()
        time.sleep(0.1)
    else:
        led1.off()
        time.sleep(0.1)
        print('STOP')
        led2.on()
        time.sleep(0.5)

        led3.on()
        time.sleep(0.1)