import machine
import ssd1306
import time

# 1. Налаштування I2C та екрана
i2c = machine.I2C(0, scl=machine.Pin(22), sda=machine.Pin(21))
oled = ssd1306.SSD1306_I2C(128, 64, i2c)

# 2. Робочий режим (Працюю...)
oled.fill(0)
oled.text("Working...", 0, 20)
oled.show()
time.sleep(2)

# 3. Підготовка до сну (Засинаю...)
oled.fill(0)
oled.text("Sleeping 5s...", 0, 20)
oled.show()
time.sleep(1) # Даємо 1 секунду учню прочитати напис

# Гасимо екран для економії енергії перед сном
oled.poweroff() 

# Переводимо ESP32 в режим легкого сну на 5 секунд
machine.lightsleep(5000)

# 4. Пробудження (Прокинувся!)
oled.poweron() # Вмикаємо екран назад
oled.fill(0)
oled.text("Woke up!", 0, 20)
oled.text("Continuing...", 0, 40)
oled.show()