from machine import Pin, I2C
import ssd1306
import time

i2c = I2C(
    0,
    scl=Pin(1),
    sda=Pin(0),
    freq=400000
)

oled = ssd1306.SSD1306_I2C(128, 64, i2c)

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

counter = 0
last_state = 1

def show_display(value):
    oled.fill(0)
    oled.text("SSD1306 Demo", 0, 0)
    oled.text("Button Count:", 0, 10)
    oled.text(str(value), 0, 40)
    oled.show()

show_display(counter)

while True:
    current_state = button.value()

    if last_state == 1 and current_state == 0:
        counter += 1
        show_display(counter)
        time.sleep(0.2)

    last_state = current_state
    time.sleep(0.01)