import mpu6050 as mpu
from oled_plot import plot
from machine import I2C, Pin

mpu_i2c = I2C(0, scl=Pin(1), sda=Pin(0))
mpu.init_mpu6050(mpu_i2c)

data_filter = [0] * 5
def get_moving_avg(input_data):
    global data_filter
    data_filter = data_filter[1:] + [input_data]
    return sum(data_filter)/len(data_filter)

while True:
    raw_data = mpu.get_mpu6050_data(mpu_i2c)
    avg_data = get_moving_avg(raw_data['accel']['x'])
    plot(raw_data, avg_data)