from machine import Pin
from time import sleep

# Road A LEDs
red1 = Pin(2, Pin.OUT)
yellow1 = Pin(3, Pin.OUT)
green1 = Pin(4, Pin.OUT)

# Road B LEDs
red2 = Pin(5, Pin.OUT)
yellow2 = Pin(6, Pin.OUT)
green2 = Pin(7, Pin.OUT)

def road_a_green():
    red1.off()
    yellow1.off()
    green1.on()

    red2.on()
    yellow2.off()
    green2.off()

def road_a_yellow():
    red1.off()
    yellow1.on()
    green1.off()

    red2.on()
    yellow2.off()
    green2.off()

def road_b_green():
    red1.on()
    yellow1.off()
    green1.off()

    red2.off()
    yellow2.off()
    green2.on()

def road_b_yellow():
    red1.on()
    yellow1.off()
    green1.off()

    red2.off()
    yellow2.on()
    green2.off()

while True:
    # Road A Green, Road B Red
    road_a_green()
    sleep(5)

    # Road A Yellow
    road_a_yellow()
    sleep(2)

    # Road B Green, Road A Red
    road_b_green()
    sleep(5)

    # Road B Yellow
    road_b_yellow()
    sleep(2)