from machine import Pin, I2C
import ssd1306
import time

# OLED setup
i2c = I2C(0, scl=Pin(5), sda=Pin(4))
oled = ssd1306.SSD1306_I2C(128, 64, i2c)

# Button and LED setup
buttons = [Pin(p, Pin.IN, Pin.PULL_UP) for p in [13, 12, 14, 27, 26]]
leds = [Pin(p, Pin.OUT) for p in [23, 22, 21, 19, 18]]

winner = None

while True:
    for i, btn in enumerate(buttons):
        if btn.value() == 0 and winner is None:
            winner = i + 1
            leds[i].on()
            oled.fill(0)
            oled.text("Winner: Player", 0, 20)
            oled.text(str(winner), 80, 20)
            oled.show()
            print("Player", winner, "pressed first!")
    time.sleep(0.1)