added legend visibility dashboard

utils.py

@@ -177,6 +177,7 @@ def create_flow_chart(steps):
         mode='markers+text',
         text=node_text,
         textposition="top center",
+        showlegend=False,
         hovertext=hover_text,
         hoverinfo='text',
         hoverlabel=dict(bgcolor="white", font_size=12, font_family="Arial"),
@@ -191,10 +192,37 @@ def create_flow_chart(steps):
         x=edge_x, y=edge_y,
         line=dict(width=2, color='#888'),
         hoverinfo='none',
+        showlegend=False,
         mode='lines')
+    
+    # Create legend traces
+    legend_traces = []
+    
+    # Color legend
+    for success, color in color_map.items():
+        legend_traces.append(go.Scatter(
+            x=[None], y=[None],
+            mode='markers',
+            marker=dict(size=10, color=color),
+            showlegend=True,
+            name=f"{'Success' if success else 'Issue'}"
+        ))
+    
+    # Shape legend
+    for action, shape in shape_map.items():
+        legend_traces.append(go.Scatter(
+            x=[None], y=[None],
+            mode='markers',
+            marker=dict(size=10, symbol=shape, color='gray'),
+            showlegend=True,
+            name=f"{action.capitalize()}"
+        ))
+
+    # Combine all traces
+    all_traces = [edge_trace, node_trace] + legend_traces
 
     layout = go.Layout(
-        showlegend=False,
+        showlegend=True,
         hovermode='closest',
         margin=dict(b=20,l=5,r=5,t=40),
         xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
@@ -206,27 +234,20 @@ def create_flow_chart(steps):
             orientation='h',  # Vertical orientation
             bgcolor='rgba(255,255,255,0.8)',  # Slightly transparent white background
             color='#777',  # Color of inactive tools
-        )
+        ),
+        legend=dict(
+            orientation="h",
+            yanchor="bottom",
+            y=0.02,
+            xanchor="right",
+            x=1,
+            bgcolor='rgba(255,255,255,0.8)',
+            bordercolor='rgba(0,0,0,0.1)',
+            borderwidth=1
+        ),
     )
     
-    fig = go.Figure(data=[edge_trace, node_trace], layout=layout)
-    
-    # Add a legend for colors and shapes
-    for success, color in color_map.items():
-        fig.add_trace(go.Scatter(
-            x=[None], y=[None], mode='markers',
-            marker=dict(size=10, color=color),
-            showlegend=True,
-            name=f"{'Success' if success else 'Challenge'}"
-        ))
-    
-    for action, shape in shape_map.items():
-        fig.add_trace(go.Scatter(
-            x=[None], y=[None], mode='markers',
-            marker=dict(size=10, symbol=shape, color='gray'),
-            showlegend=True,
-            name=f"Action: {action.capitalize()}"
-        ))
+    fig = go.Figure(data=all_traces, layout=layout)
     
     fig.update_layout(legend=dict(
         orientation="h",