Update app.py

app.py

@@ -24,6 +24,7 @@ import seaborn as sns
 import gradio as gr
 #https://gradio.app/docs/#i_slider
 import matplotlib.pyplot as plt
+import matplotlib.patches as mpatches
 import time
 import numpy as np
 
@@ -167,6 +168,13 @@ def gradio_music_graph(client_id, client_secret, artist_name): #total_albums
   plt.tight_layout(rect=[1, 2.5, 1, 0.45]) #(left, bottom, right, top)
 
   sns.boxplot(x=df["album_name"], y=df["popularity"], ax=ax)
+  
+  leg_handles = []
+  for label, artist in zip(labels,ax.artists):
+    handle = mpatches.Patch(facecolor=artist.get_facecolor(),label=label)
+    leg_handles.append(handle)
+  ax.legend(handles=leg_handles,bbox_to_anchor=(0., 1.02, 1., .102), loc=3, ncol=1, mode="expand", borderaxespad=0.)
+  
   plt.show()
   
   return fig