Update app.py

app.py

@@ -188,8 +188,10 @@ def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str]) -> np.nd
     # Convert plt to numpy array
     plt_image = plt.gcf()
     plt_image.canvas.draw()
-    image_array = np.frombuffer(plt_image.canvas.tostring_rgb(), dtype=np.uint8)
-    image_array = image_array.reshape(plt_image.canvas.get_width_height()[::-1] + (3,))
+    #image_array = np.frombuffer(plt_image.canvas.tostring_rgb(), dtype=np.uint8)
+    #image_array = image_array.reshape(plt_image.canvas.get_width_height()[::-1] + (3,))
+    image_array_rgba = np.asarray(plt_image.canvas.buffer_rgba())
+    image_array = image_array_rgba[:, :, :3]
     plt.close()
 
     return image_array