Update app.py

app.py

@@ -64,23 +64,26 @@ def add_to_graph(
     return search_string, fig
 
 def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
-    pos: Dict[Any, Tuple[float, float]] = nx.spring_layout(graph)
+    pos: Dict[Any, Tuple[float, float]] = nx.spring_layout(graph, seed=42)  # Adding seed for consistent layout
     
-    # Edge trace initialization
+    # Edge trace initialization with descriptive labels
     edge_x, edge_y = [], []
+    edge_text = []
 
-    for edge in graph.edges():
+    for edge in graph.edges(data=True):
         x0, y0 = pos[edge[0]]
         x1, y1 = pos[edge[1]]
         edge_x.extend([x0, x1, None])
         edge_y.extend([y0, y1, None])
+        edge_text.append(f"{edge[0]} -> {edge[1]}: {edge[2]['relationship']}")  # Adding edge relationship label
 
-    # Create edge trace with improved style
+    # Create edge trace with hover text for relationship descriptions
     edge_trace = go.Scatter(
         x=edge_x,
         y=edge_y,
         line=dict(width=1.0, color='#888'),
-        hoverinfo='none',
+        hoverinfo='text',
+        text=edge_text,
         mode='lines',
         name="Edges"
     )
@@ -97,32 +100,32 @@ def create_plotly_graph(graph: nx.DiGraph) -> go.Figure:
         )
         node_color.append(color_map[graph.nodes[node]['type']])
 
-    # Create node trace for hover interactivity
+    # Create node trace with enhanced marker styles and hover info
     node_trace = go.Scatter(
         x=node_x,
         y=node_y,
         text=node_text,
-        mode='markers',  # Changed from 'markers+text' to 'markers' for hover only
+        mode='markers',
         hoverinfo='text',
         marker=dict(
             showscale=True,
-            colorscale='Viridis',  # Updated to 'Viridis' for a more vibrant color scheme
+            colorscale='Viridis',
             reversescale=True,
             color=node_color,
-            size=15,  # Increased size for better visibility
-            line_width=2
+            size=20,  # Increased size for better visibility
+            line=dict(width=3, color='DarkSlateGrey')  # Added border for better contrast
         ),
         name="Nodes"
     )
 
-    # Create the figure and add traces with improved layout and style
+    # Create the figure and add traces with enhanced layout and better spacing
     fig = go.Figure()
     fig.add_trace(edge_trace)
     fig.add_trace(node_trace)
     fig.update_layout(
         title='Your Educational Landscape',
         titlefont_size=20,
-        showlegend=True,  # Show legend
+        showlegend=True,
         hovermode='closest',
         margin=dict(b=20, l=5, r=5, t=40),
         annotations=[dict(
@@ -133,9 +136,19 @@ 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),
-        plot_bgcolor='rgba(240, 240, 240, 0.95)'  # Set a light background for better contrast
+        plot_bgcolor='rgba(240, 240, 240, 0.95)'
     )
 
+    # Add legend groups
+    for node_type, color in color_map.items():
+        fig.add_trace(go.Scatter(
+            x=[],
+            y=[],
+            mode='markers',
+            marker=dict(size=10, color=color),
+            name=node_type
+        ))
+
     return fig
 
 def clear_graph() -> str: