Update pilot_graph

pilot_graph

@@ -1,6 +1,7 @@
 from PIL import Image
 import io
 import matplotlib.pyplot as plt
+import networkx as nx
 
 def plot_graph(self, G):
     plt.figure(figsize=(3, 2))
@@ -13,6 +14,9 @@ def plot_graph(self, G):
     plt.savefig(buf, format="png")
     plt.close()
 
+    # Move to the beginning of the buffer before reading
+    buf.seek(0)
+
     # Load the image from the buffer into a PIL Image
     image = Image.open(buf)
-    return image
+    return image