Update app.py

app.py

@@ -75,60 +75,54 @@ def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
         edge_x.extend([x0, x1, None])
         edge_y.extend([y0, y1, None])
 
-    # Create edge trace
+    # Create edge trace with improved style
     edge_trace = go.Scatter(
         x=edge_x,
         y=edge_y,
-        line=dict(width=0.5, color='#888'),
+        line=dict(width=1.0, color='#888'),
         hoverinfo='none',
-        mode='lines'
+        mode='lines',
+        name="Edges"
     )
 
-    # Node trace initialization
+    # Node trace initialization with improved style
     node_x, node_y, node_text, node_color = [], [], [], []
 
     for node in graph.nodes():
         x, y = pos[node]
         node_x.append(x)
         node_y.append(y)
-        node_text.append(node)
+        node_text.append(
+            f"Node: {node}<br>Type: {graph.nodes[node]['type']}<br>{graph.nodes[node].get('description', '')}"
+        )
         node_color.append(color_map[graph.nodes[node]['type']])
 
-    # 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]
-        node_data = graph.nodes[node_info]
-        hover_text = f"Node: {node_info}<br>Type: {node_data['type']}<br>{node_data.get('description', '')}"
-        node_hover_texts.append(hover_text)
-
-    # Create node trace
+    # Create node trace for hover interactivity
     node_trace = go.Scatter(
         x=node_x,
         y=node_y,
-        text=node_hover_texts,
-        mode='markers+text',
+        text=node_text,
+        mode='markers',  # Changed from 'markers+text' to 'markers' for hover only
         hoverinfo='text',
         marker=dict(
-            showscale=False,
-            colorscale='YlGnBu',
+            showscale=True,
+            colorscale='Viridis',  # Updated to 'Viridis' for a more vibrant color scheme
             reversescale=True,
-            color=node_adjacencies,
-            size=10,
+            color=node_color,
+            size=15,  # Increased size for better visibility
             line_width=2
-        )
+        ),
+        name="Nodes"
     )
 
-    # Create the figure and add traces
+    # Create the figure and add traces with improved layout and style
     fig = go.Figure()
     fig.add_trace(edge_trace)
     fig.add_trace(node_trace)
     fig.update_layout(
         title='Your Educational Landscape',
-        titlefont_size=16,
-        showlegend=False,
+        titlefont_size=20,
+        showlegend=True,  # Show legend
         hovermode='closest',
         margin=dict(b=20, l=5, r=5, t=40),
         annotations=[dict(
@@ -138,7 +132,8 @@ def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
             x=0.005, y=-0.002
         )],
         xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-        yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
+        yaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
+        plot_bgcolor='rgba(240, 240, 240, 0.95)'  # Set a light background for better contrast
     )
 
     return fig