from ssd1306 import SSD1306_I2C
from machine import Pin, I2C
import time
from time import *

width = 128
height = 64

i2c=I2C(0,scl=Pin(1),sda=Pin(0),freq=200000)
oled = SSD1306_I2C(width,height,i2c)

led_green = Pin(2, Pin.OUT)
btn_green = Pin(26, Pin.IN)

led_blue = Pin(15, Pin.OUT)
btn_blue = Pin(22, Pin.IN)

count = 0

Am = Pin(7,Pin.OUT)
Ap = Pin(6,Pin.OUT)
Bn = Pin(5,Pin.OUT)
Bp = Pin(4,Pin.OUT)

Am1 = Pin(9,Pin.OUT)
Ap1 = Pin(10,Pin.OUT)
Bn1 = Pin(11,Pin.OUT)
Bp1 = Pin(12,Pin.OUT)

def stepper_motor1():
    sleep_ms(1)
    Bn.value(0)
    Ap.value(1)
    sleep_ms(1)
    Ap.value(0)
    Bp.value(1)
    sleep_ms(1)
    Bp.value(0)
    Am.value(1)
    sleep_ms(1)
    Am.value(0)
    Bn.value(1)

def stepper_motor2():
    sleep_ms(1)
    Bn1.value(0)
    Ap1.value(1)
    sleep_ms(1)
    Ap1.value(0)
    Bp1.value(1)
    sleep_ms(1)
    Bp1.value(0)
    Am1.value(1)
    sleep_ms(1)
    Am1.value(0)
    Bn1.value(1)

while 1:
    if btn_green.value():
        led_green.toggle()
        if led_green.value():
            stepper_motor1()
            oled.text('btn_green is on', 0, count)
            oled.show()
            count +=10
    else:
        oled.text('btn_green is off', 0, count)
        oled.show()
        sleep(1.5)
        count +=10




    if btn_blue.value():
        led_blue.toggle()
        if led_blue.value():
            stepper_motor2()
            oled.text('btn_blue is on', 0, count)
            count += 10
    else:
        oled.text('btn_blue is off', 0, count)
        oled.show()
        sleep(1.5)
        count += 10

    if count == 60:
        count = 0
        oled.fill(0)