switching from graphviz to networkx

app.py

@@ -19,6 +19,7 @@ from PIL import Image, ImageDraw, ImageFont
 from io import BytesIO
 import functools
 import logging
+import tempfile
 
 
 # Set up custom logger
@@ -344,14 +345,12 @@ def create_tree_diagram(data, config, max_weight, min_weight, trim_cutoff=0):
     labels = nx.get_node_attributes(G, 'label')
     nx.draw(G, pos, labels=labels, with_labels=True, node_size=5000, node_color="lightblue", font_size=10, font_weight="bold", edge_color="gray", arrows=False)
 
-    # Save the image to a BytesIO object
-    output = BytesIO()
-    plt.savefig(output, format='png')
+    # Save the image to a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+    plt.savefig(temp_file.name)
     plt.close()
-    output.seek(0)
-
-    return output
 
+    return temp_file.name