Update app.py

app.py

@@ -150,8 +150,13 @@ def generate_wordcloud(entities: List[Dict], color_map: Dict[str, str], file_pat
 
     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,))
+    # Get the RGBA buffer directly as a (height, width, 4) numpy array
+    image_array_rgba = np.asarray(plt_image.canvas.buffer_rgba())
+
+    # Keep only the RGB channels (slice off the Alpha channel)
+    image_array = image_array_rgba[:, :, :3]
     plt.close()
 
     return image_array