main.py -- put your code here!

from pyb import OLED, delay, Pin

Create OLED Object

oled = OLED()

Create Button Objects

k1 = Pin('K1', Pin.IN, Pin.PULL_UP) k2 = Pin('K2', Pin.IN, Pin.PULL_UP)

Clear OLED Cache

oled.fill(0)

Update OLED Display

oled.show()

Create Coordinate Variables

x = 8 y = 0

Create Variables for wave selection and movement

wave_type = 'sin' move_counter = 0

while True: # Update OLED Content oled.fill(0) if wave_type == 'sin': oled.text("Sin Wave", x, y, 1) elif wave_type == 'cos': oled.text("Cos Wave", x, y, 1) oled.show()

# Move wave every 500ms
if move_counter == 5:
    y += 1
    if y == 32:
        y = -8
    move_counter = 0

# Check button press to toggle wave type
if not k1.value():
    if wave_type == 'sin':
        wave_type = 'cos'
    else:
        wave_type = 'sin'
    delay(200)  # Delay to avoid multiple button press detection

# Check button press to exit program
if not k2.value():
    break

delay(100)
move_counter += 1

Clear OLED Cache

oled.fill(0) oled.show(