from machine import Pin
from time import sleep

button1 = Pin(18, Pin.IN, Pin.PULL_DOWN)
button2 = Pin(19, Pin.IN, Pin.PULL_DOWN)
button3 = Pin(22, Pin.IN, Pin.PULL_DOWN)
button4 = Pin(20, Pin.IN, Pin.PULL_DOWN)
button5 = Pin(21, Pin.IN, Pin.PULL_DOWN)

led1 = Pin(0, Pin.OUT)
led2 = Pin(1, Pin.OUT)
led3 = Pin(2, Pin.OUT)
led4 = Pin(3, Pin.OUT)
led5 = Pin(4, Pin.OUT)
led6 = Pin(5, Pin.OUT)
led7 = Pin(6, Pin.OUT)
led8 = Pin(7, Pin.OUT)

led9 = Pin(8, Pin.OUT)
led10 = Pin(9, Pin.OUT)
led11 = Pin(10, Pin.OUT)
led12 = Pin(11, Pin.OUT)
led13 = Pin(12, Pin.OUT)
led14 = Pin(13, Pin.OUT)
led15 = Pin(14, Pin.OUT)
led16 = Pin(15, Pin.OUT)

led17 = Pin(28, Pin.OUT)
led18 = Pin(27, Pin.OUT)
led19 = Pin(26, Pin.OUT)

s1 = 0
s2 = 0
s3 = 0
s4 = 0
s5 = 0

def set_color(r, g, b):
    led17.value(r)
    led18.value(g)
    led19.value(b)

rgb_step = 0

while True:
    if button1.value() == 1:
        if s1 == 0:
            s1 = 1
        else:
            s1 = 0
        sleep(0.3)
        
    elif button2.value() == 1:
        if s2 == 0:
            s2 = 1
        else:
            s2 = 0
        sleep(0.3)
        
    elif button3.value() == 1:
        if s3 == 0:
            s3 = 1
        else:
            s3 = 0
        sleep(0.3)
        
    elif button4.value() == 1:
        if s4 == 0:
            s4 = 1
        else:
            s4 = 0
        sleep(0.3)
        
    elif button5.value() == 1:
        if s5 == 0:
            s5 = 1
        else:
            s5 = 0
        sleep(0.3)

    if rgb_step == 0:
        set_color(1, 0, 1)
        rgb_step = 1
    elif rgb_step == 1:
        set_color(0, 0, 0)
        rgb_step = 2
    elif rgb_step == 2:
        set_color(0, 1, 1)
        rgb_step = 3
    elif rgb_step == 3:
        set_color(0, 0, 0)
        rgb_step = 4
    elif rgb_step == 4:
        set_color(1, 1, 0)
        rgb_step = 5
    else:
        set_color(0, 0, 0)
        rgb_step = 0

    if s1 == 1:
        print("Sara")
        led9.value(1)
        sleep(0.15)
        led10.value(1)
        sleep(0.15)
        led11.value(1)
        sleep(0.15)
        led12.value(1)
        sleep(0.15)
        led13.value(1)
        sleep(0.15)
        led14.value(1)
        sleep(0.15)
        led15.value(1)
        sleep(0.15)
        led16.value(1)
        sleep(0.15)
        led9.value(0)
        led10.value(0)
        led11.value(0)
        led12.value(0)
        led13.value(0)
        led14.value(0)
        led15.value(0)
        led16.value(0)

    elif s2 == 1:
        print("Sara Alamari")
        led1.value(1)
        sleep(0.15)
        led2.value(1)
        sleep(0.15)
        led3.value(1)
        sleep(0.15)
        led4.value(1)
        sleep(0.15)
        led5.value(1)
        sleep(0.15)
        led6.value(1)
        sleep(0.15)
        led7.value(1)
        sleep(0.15)
        led8.value(1)
        sleep(0.15)
        led1.value(0)
        led2.value(0)
        led3.value(0)
        led4.value(0)
        led5.value(0)
        led6.value(0)
        led7.value(0)
        led8.value(0)

    elif s3 == 1:
        led1.value(1)
        sleep(0.15)
        led2.value(1)
        sleep(0.15)
        led3.value(1)
        sleep(0.15)
        led4.value(1)
        sleep(0.15)
        led9.value(1)
        sleep(0.15)
        led10.value(1)
        sleep(0.15)
        led11.value(1)
        sleep(0.15)
        led12.value(1)
        sleep(0.15)
        led1.value(0)
        led2.value(0)
        led3.value(0)
        led4.value(0)
        led9.value(0)
        led10.value(0)
        led11.value(0)
        led12.value(0)

    elif s4 == 1:
        led5.value(1)
        sleep(0.15)
        led6.value(1)
        sleep(0.15)
        led7.value(1)
        sleep(0.15)
        led8.value(1)
        sleep(0.15)
        led13.value(1)
        sleep(0.15)
        led14.value(1)
        sleep(0.15)
        led15.value(1)
        sleep(0.15)
        led16.value(1)
        sleep(0.15)
        led5.value(0)
        led6.value(0)
        led7.value(0)
        led8.value(0)
        led13.value(0)
        led14.value(0)
        led15.value(0)
        led16.value(0)

    elif s5 == 1:
        led1.value(1)
        led2.value(1)
        led3.value(1)
        led4.value(1)
        led5.value(1)
        led6.value(1)
        led7.value(1)
        led8.value(1)
        led9.value(1)
        led10.value(1)
        led11.value(1)
        led12.value(1)
        led13.value(1)
        led14.value(1)
        led15.value(1)
        led16.value(1)

    else:
        led1.value(0)
        led2.value(0)
        led3.value(0)
        led4.value(0)
        led5.value(0)
        led6.value(0)
        led7.value(0)
        led8.value(0)
        led9.value(0)
        led10.value(0)
        led11.value(0)
        led12.value(0)
        led13.value(0)
        led14.value(0)
        led15.value(0)
        led16.value(0)

    sleep(0.2)