from imu import MPU6050
from time import sleep
from machine import Pin, I2C

# Initialize I@C and MPU6050
i2c = I2C( 1, sda=Pin(2), scl=Pin(3), freq=400000) # Pico
imu = MPU6050

try: 
# Open or create a file for writing data
  file = open("data.txt", "w") 

except OSError:
  file = open("data.txt", "w")

count = 0

while True:
  # Read accelerometer, gyroscope, and temperature data
  ax = round(imu.accel.x,2)
  ay = round(imu.accel.y,2)
  az = round(imu.accel.z,2)
  gx = round(imu.gyro.x)
  gy = round(imu.gyro.y)
  gz = round(imu.gyro.z)
  tem = round(imu.temperature, )

  print("ax", ax, "\tay", ay, "\taz", az, "\tgx", gx, "\tgy", gy, "\tgz", gz, "\temerature", tem)

  file.write(str(count) + "," + str(ax) + "," + str(ay) + "," + str(az) + "," + str(gx) + "," + str(gy) + "," + str(gz) + "," str(tem) + "\n")
  file.flush()
  count += 1

  sleep(1)