feat: add markers for all agents and toggle for top agents

app.py

@@ -713,8 +713,35 @@ def create_combined_time_series_graph(df):
         
         logger.info(f"Showing {len(top_agents)} agents by default out of {len(unique_agents)} total agents")
         
-        # No longer showing individual agent data points as requested
-        logger.info("Skipping individual agent data points as requested")
+        # Add data points for each agent, but only make top agents visible by default
+        for agent_name in unique_agents:
+            agent_data = apr_data[apr_data['agent_name'] == agent_name]
+            
+            # Explicitly convert to Python lists
+            x_values = agent_data['timestamp'].tolist()
+            y_values = agent_data['apr'].tolist()
+            
+            # Determine if this agent should be visible by default
+            is_visible = agent_name in top_agents
+            
+            # Add data points as markers
+            fig.add_trace(
+                go.Scatter(
+                    x=x_values,
+                    y=y_values,
+                    mode='markers',  # Only markers for original data
+                    marker=dict(
+                        color=color_map[agent_name],
+                        symbol='circle',
+                        size=10,
+                        line=dict(width=1, color='black')
+                    ),
+                    name=f'Agent: {agent_name}',
+                    hovertemplate='Time: %{x}<br>APR: %{y:.2f}<br>Agent: ' + agent_name + '<extra></extra>',
+                    visible=is_visible  # Only top agents visible by default
+                )
+            )
+            logger.info(f"Added data points for agent {agent_name} with {len(x_values)} points (visible: {is_visible})")
         
         # Add 2-hour moving average as a smooth line
         x_values_ma = avg_apr_data_with_ma['timestamp'].tolist()
@@ -785,7 +812,16 @@ def create_combined_time_series_graph(df):
         hovermode="closest"
     )
     
-    # No longer need the note about hidden agents since we're not showing individual agents
+    # Add a note about hidden agents if there are more than MAX_VISIBLE_AGENTS
+    if len(unique_agents) > MAX_VISIBLE_AGENTS:
+        fig.add_annotation(
+            text=f"Note: Only showing top {MAX_VISIBLE_AGENTS} agents by default. Toggle others in legend.",
+            xref="paper", yref="paper",
+            x=0.5, y=1.05,
+            showarrow=False,
+            font=dict(size=12, color="gray"),
+            align="center"
+        )
     
     # FORCE FIXED Y-AXIS RANGE
     fig.update_yaxes(
@@ -887,8 +923,39 @@ def create_combined_time_series_graph(df):
                 else:
                     avg_apr_data_with_ma.at[i, 'infinite_avg'] = row['apr']
             
-            # No longer showing individual agent data points in fallback graph as requested
-            logger.info("Skipping individual agent data points in fallback graph as requested")
+            # Add data points for each agent, but only make top agents visible by default
+            unique_agents = apr_data['agent_name'].unique()
+            colors = px.colors.qualitative.Plotly[:len(unique_agents)]
+            color_map = {agent: colors[i % len(colors)] for i, agent in enumerate(unique_agents)}
+            
+            # Calculate the total number of data points per agent
+            agent_counts = apr_data['agent_name'].value_counts()
+            
+            # Determine how many agents to show individually (limit to top 5 most active)
+            MAX_VISIBLE_AGENTS = 5
+            top_agents = agent_counts.nlargest(min(MAX_VISIBLE_AGENTS, len(agent_counts))).index.tolist()
+            
+            for agent_name in unique_agents:
+                agent_data = apr_data[apr_data['agent_name'] == agent_name]
+                
+                # Determine if this agent should be visible by default
+                is_visible = agent_name in top_agents
+                
+                # Add data points as markers
+                simple_fig.add_trace(
+                    go.Scatter(
+                        x=agent_data['timestamp'],
+                        y=agent_data['apr'],
+                        mode='markers',
+                        name=f'Agent: {agent_name}',
+                        marker=dict(
+                            size=10, 
+                            color=color_map[agent_name]
+                        ),
+                        hovertemplate='Time: %{x}<br>APR: %{y:.2f}<br>Agent: ' + agent_name + '<extra></extra>',
+                        visible=is_visible  # Only top agents visible by default
+                    )
+                )
             
             # Add 2-hour moving average as a line
             simple_fig.add_trace(
@@ -922,7 +989,16 @@ def create_combined_time_series_graph(df):
             width=1000
         )
         
-        # No longer need the note about hidden agents since we're not showing individual agents
+        # Add a note about hidden agents if there are more than MAX_VISIBLE_AGENTS
+        if len(unique_agents) > MAX_VISIBLE_AGENTS:
+            simple_fig.add_annotation(
+                text=f"Note: Only showing top {MAX_VISIBLE_AGENTS} agents by default. Toggle others in legend.",
+                xref="paper", yref="paper",
+                x=0.5, y=1.05,
+                showarrow=False,
+                font=dict(size=12, color="gray"),
+                align="center"
+            )
         
         # Return the simple figure
         return simple_fig