from machine import Pin
from time import sleep

led1_pin = Pin(25, Pin.OUT)
led2_pin = Pin(14, Pin.OUT) 
pb1_pin = Pin(22, Pin.IN)
pb2_pin = Pin(17, Pin.IN)
 

def main():
    while True:
        sub_led1_off()
        sub_led2_off() 
        pb1_state = pb1_pin.value()
        pb2_state = pb2_pin.value()
 
        if pb1_state == 1:
            sub_led1_on()
            sub_led2_off()
        elif pb2_state == 1:
            sub_led2_on()
            sub_led1_off()
          
def sub_led1_on():
    led1_pin.on()
    sleep(0.5)
    
def sub_led1_off():
    led1_pin.off()
    sleep(0.5)
     
def sub_led2_on():
    led2_pin.on()
    sleep(0.5)
    
def sub_led2_off():
    led2_pin.off()
    sleep(0.5)


if __name__ == '__main__':
 main()