from machine import Pin 
from time import sleep

# buttons 
exit_but = Pin(14, Pin.IN , Pin.PULL_DOWN)
enter_but = Pin(27 , Pin.IN , Pin.PULL_DOWN)

# LEDs
green = Pin(2 , Pin.OUT)
red = Pin(4 , Pin.OUT)
yellow = Pin(5 , Pin.OUT)
blue = Pin(18 , Pin.OUT)

count = 0
max_capacity = 15

def update_leds():
    if count == max_capacity:
        red.on()
        green.off()
    else:
        red.off()
        green.on()

def blink(led):
    led.on()
    sleep(0.3)
    led.off()

while True:

    
    if enter_but.value() ==1:
        if count < max_capacity:
            count += 1
            blink(yellow)
        
        update_leds()
        sleep(0.3)

    if exit_but.value() == 1:
        if count > 0:
            count -= 1 
            blink(blue)

        update_leds()
        sleep(0.3)