from machine import Pin
from time import sleep_ms

a = Pin(2, Pin.OUT)
b = Pin(3, Pin.OUT)
c = Pin(4, Pin.OUT)
d = Pin(5, Pin.OUT)
e = Pin(6, Pin.OUT)
f = Pin(7, Pin.OUT)
g = Pin(8, Pin.OUT)


button = Pin(9, Pin.IN, Pin.PULL_UP)

n = 0

def display(num):
    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]   
    ]

    a.value(digits[num][0])
    b.value(digits[num][1])
    c.value(digits[num][2])
    d.value(digits[num][3])
    e.value(digits[num][4])
    f.value(digits[num][5])
    g.value(digits[num][6])


display(0)

while True:
    if button.value() == 0:
        n += 1

        if n == 10:
            n = 0

        display(n)

        
        sleep_ms(300)