re-add draw graph

lesson_graph.py

@@ -155,6 +155,41 @@ class LessonGraph:
     
         return search_string, image
 
+    def draw_graph(self) -> Image.Image:
+        """
+        Visualize the graph using Matplotlib, handling layout, labels, and interactivity.
+        """
+        fig, ax = plt.subplots(figsize=(14, 10))
+        pos = nx.spring_layout(self.graph, k=0.9, iterations=50)
+
+        self._draw_nodes(ax, pos)
+        self._draw_edges(ax, pos)
+        self._add_legend(ax)
+        
+        plt.title("Your Educational Landscape", fontsize=16)
+        plt.axis('off')
+        plt.tight_layout()
+        
+        self._add_interactivity()
+        
+        # Save the plot to a BytesIO object
+        buf = BytesIO()
+        plt.savefig(buf, format="png", dpi=300, bbox_inches="tight")
+        buf.seek(0)
+        plt.close(fig)
+
+        return Image.open(buf)
+
+    def _draw_nodes(self, ax, pos):
+        node_colors = [self.COLOR_MAP[self.graph.nodes[node]['type']] for node in self.graph.nodes()]
+        nx.draw_networkx_nodes(self.graph, pos, node_color=node_colors, node_size=3000, alpha=0.8, ax=ax)
+        nx.draw_networkx_labels(self.graph, pos, font_size=10, font_weight="bold", ax=ax)
+
+    def _draw_edges(self, ax, pos):
+        nx.draw_networkx_edges(self.graph, pos, edge_color='gray', arrows=True, arrowsize=20, ax=ax)
+        edge_labels = nx.get_edge_attributes(self.graph, 'relationship')
+        nx.draw_networkx_edge_labels(self.graph, pos, edge_labels=edge_labels, font_size=8, ax=ax)
+    
     def _add_legend(self, ax):
         legend_elements = [mpatches.Patch(color=color, label=node_type.value) for node_type, color in self.COLOR_MAP.items()]
         ax.legend(handles=legend_elements, loc='upper left', bbox_to_anchor=(1, 1), title="Node Types")