from machine import Pin, I2C
from time import sleep import ssd1306
i2c = I2C(0, scl=Pin(22), sda=Pin(21))
oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)

led1 = Pin(2,Pin.OUT)
led2 = Pin(18,Pin.OUT)
sw1 = Pin(4,Pin.IN)
sw2 = Pin(5,Pin.IN)
a = 0
b = 0
while True:
    if sw1.value() == 0:
        a + = 1 
        b = 0
        sleep(0.5)      
    if a == 2:
        a = 0
        b = 1
    if sw2.value() == 0:
        b + = 1  
        b = 0 
        a = 0
        sleep(0.5)      
    if b == 2:
        b = 0
        a = 0
    led1.value(a)
    led2.value(b)