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)

red_led = Pin(12, Pin.OUT)
yellow_led = Pin(13, Pin.OUT)
green_led = Pin(14, Pin.OUT)

def show_on_oled(message):
    oled.fill(0) 
    oled.text(message, 10, 10)  
    oled.show()

def led_r():
    red_led.on()
    show_on_oled("STOP!")
    sleep(5)
    red_led.off()

def led_y():
    yellow_led.on()
    show_on_oled("READY")
    sleep(3)
    yellow_led.off()


def led_g():
    green_led.on()
    show_on_oled("GO")
    sleep(5)
    green_led.off()

while True:
    led_r()
    led_y()
    led_g()