from machine import Pin
import time

seg = [0, 1, 2, 3, 4, 5, 6]
segs = [Pin(pin, Pin.OUT) for pin in seg]
digits = [
    [0, 0, 0, 0, 0, 0, 1],
    [1, 0, 0, 1, 1, 1, 1],
    [0, 0, 1, 0, 0, 1, 0],
    [0, 0, 0, 0, 1, 1, 0],
    [1, 0, 0, 1, 1, 0, 0],
    [0, 1, 0, 0, 1, 0, 0],
    [0, 1, 0, 0, 0, 0, 0],
    [0, 0, 0, 1, 1, 1, 1],
    [0, 0, 0, 0, 0, 0, 0],
    [0, 0, 0, 0, 1, 0, 0]
]

seg1 = [18, 17, 16, 15, 14, 13, 12]
segs1 = [Pin(pin, Pin.OUT) for pin in seg1]
digits1 = [
    [0, 0, 0, 1, 0, 0, 0],
    [1, 1, 0, 0, 0, 0, 0],
    [0, 1, 1, 0, 0, 0, 1],
    [1, 0, 0, 0, 0, 1, 0],
    [0, 1, 1, 0, 0, 0, 0],
    [0, 1, 1, 1, 0, 0, 0]
]

seg2 = [7, 8, 9, 10, 11, 26, 27]
segs2 = [Pin(pin, Pin.OUT) for pin in seg2]
digits2 = [
    [0, 1, 1, 1, 0, 0, 0],
    [0, 0, 0, 1, 0, 0, 0],
    [1, 0, 0, 0, 0, 0, 1]
]

def display_number(num):
    if 0 <= num <= 9:
        pattern = digits[num]
        print(num)
        for i in range(7):
            segs[i].value(pattern[i])

def display_number1(num):
    if 0 <= num <= 5:
        l = ['A', 'B', 'C', 'D', 'E', 'F']
        pattern = digits1[num]
        print(l[num])
        for i in range(7):
            segs1[i].value(pattern[i])

def display_number2(num):
    if 0 <= num <= 2:
        l = ['F', 'A', 'U']
        pattern = digits2[num]
        print(l[num])
        for i in range(7):
            segs2[i].value(pattern[i])

while True:
    for i in range(10):
        display_number(i)
        time.sleep(0.5)
    for i in range(8, -1, -1):
        display_number(i)
        time.sleep(0.5)

    for i in range(6):
        display_number1(i)
        time.sleep(0.5)
    for i in range(5, -1, -1):
        display_number1(i)
        time.sleep(0.5)

    for i in range(3):
        display_number2(i)
        time.sleep(0.5)
    for i in range(2, -1, -1):
        display_number2(i)
        time.sleep(0.5)