Add colors to plot

utils.py

@@ -32,6 +32,7 @@ def dataframe_to_pianoroll_img(df):
     ax.scatter(df.start_time, df.pitch, c="white")
     n_colors = len(df.instrument.unique())
     colorscale = px.colors.sample_colorscale("viridis", [n/(n_colors) for n in range(n_colors)])
+    print("colorscale", colorscale)
     colordict = {f:colorscale[i] for i, f in enumerate(df.instrument.unique())}
     for _, row in df.iterrows():
         ax.add_patch(Rectangle((row["start_time"],