from machine import Pin
from utime import sleep

ledRed = Pin(5, Pin.OUT)
ledGreen = Pin(6, Pin.OUT)
ledYellow = Pin(7, Pin.OUT)

segments = [
    Pin(8, Pin.OUT),
    Pin(9, Pin.OUT),
    Pin(10, Pin.OUT),
    Pin(11, Pin.OUT),
    Pin(12, Pin.OUT),
    Pin(13, Pin.OUT),
    Pin(14, Pin.OUT)
]

digits = [
    [1,1,1,1,1,1,0],  
    [0,1,1,0,0,0,0],  
    [1,1,0,1,1,0,1],  
    [1,1,1,1,0,0,1],  
    [0,1,1,0,0,1,1],  
    [1,0,1,1,0,1,1],  
    [1,0,1,1,1,1,1],  
    [1,1,1,0,0,0,0],  
    [1,1,1,1,1,1,1],  
    [1,1,1,1,0,1,1]  
]

def show_number(n):
    pattern = digits[n]
    for i in range(7):
        segments[i].value(pattern[i])

def countdown(t, led):
    led.on()
    for i in range(t, 0, -1):
        show_number(i)
        sleep(1)
    led.off()
    show_number(0)
    sleep(1)

while True:
    countdown(5, ledRed)
    countdown(5, ledGreen)
    countdown(2, ledYellow)