import pypianoroll as roll
import matplotlib.pyplot as plt
import numpy as np
import os
import settings

instruments = np.load(settings.generated_sample_path)['arr_0'][0]

instruments = instruments.reshape(96,128)
# instruments = instruments>0.5
instruments = instruments*255

i = roll.Track(instruments, program=0)
generated_midi = roll.Multitrack(tracks=[i], tempo=120.0, downbeat=[0, 96, 192, 288], beat_resolution=24)
roll.write(generated_midi, settings.generated_midi_path)

plt.imshow(instruments.T, cmap='gray')
plt.savefig(settings.generated_pianoroll_path)