Update app.py

app.py

@@ -1,8 +1,8 @@
-from typing import Tuple, Dict, Any, List
-import gradio as gr
 import networkx as nx
 import plotly.graph_objs as go
 from plotly.subplots import make_subplots
+from typing import Dict, Tuple, List, Any
+import gradio as gr
 import plotly.io as pio
 
 # Initialize the lesson plan graph as a directed graph
@@ -77,41 +77,37 @@ def add_to_graph(
 def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
     pos: Dict[Any, Tuple[float, float]] = nx.spring_layout(graph)
     
-    edge_trace: go.Scatter = go.Scatter(
-        x=[],
-        y=[],
-        line=dict(width=0.5, color='#888'),
-        hoverinfo='none',
-        mode='lines')
+    # Edge trace initialization
+    edge_x, edge_y = [], []
 
     for edge in graph.edges():
         x0, y0 = pos[edge[0]]
         x1, y1 = pos[edge[1]]
-        edge_trace['x'] += tuple([x0, x1, None])
-        edge_trace['y'] += tuple([y0, y1, None])
+        edge_x.extend([x0, x1, None])
+        edge_y.extend([y0, y1, None])
 
-    node_trace: go.Scatter = go.Scatter(
-        x=[],
-        y=[],
-        text=[],
-        mode='markers+text',
-        hoverinfo='text',  # Ensure hover text displays
-        marker=dict(
-            showscale=False,
-            colorscale='YlGnBu',
-            reversescale=True,
-            color=[],
-            size=10,
-            line_width=2))
+    # Create edge trace
+    edge_trace = go.Scatter(
+        x=edge_x,
+        y=edge_y,
+        line=dict(width=0.5, color='#888'),
+        hoverinfo='none',
+        mode='lines'
+    )
+
+    # Node trace initialization
+    node_x, node_y, node_text, node_color = [], [], [], []
 
     for node in graph.nodes():
         x, y = pos[node]
-        node_trace['x'].append(x)  # Changed from += tuple([x])
-        node_trace['y'].append(y)  # Changed from += tuple([y])
-        node_trace['marker']['color'].append(color_map[graph.nodes[node]['type']])  # Changed from += tuple()
+        node_x.append(x)
+        node_y.append(y)
+        node_text.append(node)
+        node_color.append(color_map[graph.nodes[node]['type']])
 
-    node_adjacencies: List[int] = []
-    node_hover_texts: List[str] = []
+    # Setting hover texts for each node
+    node_adjacencies = []
+    node_hover_texts = []
     for node, adjacencies in enumerate(graph.adjacency()):
         node_adjacencies.append(len(adjacencies[1]))
         node_info = list(graph.nodes())[node]
@@ -119,10 +115,25 @@ def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
         hover_text = f"Node: {node_info}<br>Type: {node_data['type']}<br>{node_data.get('description', '')}"
         node_hover_texts.append(hover_text)
 
-    node_trace.marker.color = node_adjacencies
-    node_trace.text = node_hover_texts  # Assign the hover text list
+    # Create node trace
+    node_trace = go.Scatter(
+        x=node_x,
+        y=node_y,
+        text=node_hover_texts,
+        mode='markers+text',
+        hoverinfo='text',
+        marker=dict(
+            showscale=False,
+            colorscale='YlGnBu',
+            reversescale=True,
+            color=node_adjacencies,
+            size=10,
+            line_width=2
+        )
+    )
 
-    fig: go.Figure = make_subplots()
+    # Create the figure and add traces
+    fig = make_subplots()
     fig.add_trace(edge_trace)
     fig.add_trace(node_trace)
     fig.update_layout(
@@ -130,7 +141,7 @@ def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
         titlefont_size=16,
         showlegend=False,
         hovermode='closest',
-        margin=dict(b=20,l=5,r=5,t=40),
+        margin=dict(b=20, l=5, r=5, t=40),
         annotations=[dict(
             text="",
             showarrow=False,
@@ -140,7 +151,7 @@ def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
         xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
         yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
     )
-    
+
     return fig
 
 def clear_graph() -> str:
@@ -189,4 +200,4 @@ with demo:
 
 # Launch the EduScape app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()