Update app.py

app.py

@@ -174,14 +174,16 @@ def compute_bfi_scores(*args):
                         ha='center', va='bottom')
         
         plt.tight_layout()
+        
         # Save the plot to a PNG image in memory
-        buf = BytesIO()
-        plt.savefig(buf, format='png')
-        buf.seek(0)
-        plt.close(fig)
-        # Convert bytes to PIL Image
-        image = PILImage.open(buf)
-        buf.close()
+        with BytesIO() as buf:
+            plt.savefig(buf, format='png')
+            buf.seek(0)
+            # Convert bytes to PIL Image and ensure data is loaded
+            image = PILImage.open(buf).copy()
+            # No need to close the buffer explicitly; 'with' statement handles it
+        
+        plt.close(fig)  # Close the figure to free memory
 
         markdown_output += "### **Trait Scores Visualization**\n\n"