Update app.py

app.py

@@ -5,6 +5,8 @@ import matplotlib.pyplot as plt
 import networkx as nx
 from matplotlib.colors import to_hex
 import threading
+from io import BytesIO
+from PIL import Image
 
 lock = threading.Lock()
 
@@ -66,8 +68,15 @@ class Network:
             nx.draw(G, pos, with_labels=False, node_size=sizes, node_color=colors, edge_color="gray")
             plt.axis('off')
             plt.tight_layout()
+
+            # Save the plot to an in-memory buffer
+            buf = BytesIO()
+            plt.savefig(buf, format='png')
+            buf.seek(0)
             plt.close()
-        return "network.png"
+
+            # Return a PIL image
+            return Image.open(buf)
 
     @staticmethod
     def get_color(value):
@@ -95,7 +104,7 @@ def reset_network():
     return network.render()
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Compassion Network (Docker Example)")
+    gr.Markdown("# Compassion Network (In-Memory Image Example)")
     gr.Markdown("Add nodes and watch them grow and connect. Click Reset to start over.")
     with gr.Row():
         with gr.Column():
@@ -105,15 +114,15 @@ with gr.Blocks() as demo:
             random_button = gr.Button("Add Random Node")
             reset_button = gr.Button("Reset Network")
         with gr.Column():
-            graph_output = gr.Image("network.png", interactive=False)
+            graph_output = gr.Image(type="pil", interactive=False)
 
-    # Render an initial blank network image
-    network.render()
+    # Initialize the graph with an empty image
+    graph_output.update(value=network.render())
 
     add_button.click(interact_network, inputs=[x_input, y_input], outputs=graph_output)
     random_button.click(random_node, inputs=None, outputs=graph_output)
     reset_button.click(reset_network, inputs=None, outputs=graph_output)
 
 if __name__ == "__main__":
-    # Run on port 7860; required for Spaces
+    # Required for Hugging Face Spaces Docker
     demo.launch(server_name="0.0.0.0", server_port=7860)