refactor: streamline APR visualization generation and update graph handling

app.py

@@ -214,30 +214,24 @@ def generate_apr_visualizations():
             yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
         )
         
-        # Save as static files for reference
-        fig.write_html("modius_apr_per_agent_graph.html")
-        fig.write_image("modius_apr_per_agent_graph.png")
+        # Save as static file for reference
         fig.write_html("modius_apr_combined_graph.html")
         fig.write_image("modius_apr_combined_graph.png")
         
         csv_file = None
-        return fig, fig, csv_file
+        return fig, csv_file
     
     # No longer generating dummy data
     # Set global_df for access by other functions
     global_df = df
-    print(global_df.head())
-    print(df.head())
+    
     # Save to CSV before creating visualizations
     csv_file = save_to_csv(df)
     
-    # Create per-agent time series graph (returns figure object)
-    per_agent_fig = create_time_series_graph_per_agent(df)
-    
-    # Create combined time series graph (returns figure object)
+    # Only create combined time series graph
     combined_fig = create_combined_time_series_graph(df)
     
-    return per_agent_fig, combined_fig, csv_file
+    return combined_fig, csv_file
 
 def create_time_series_graph_per_agent(df):
     """Create a time series graph for each agent using Plotly"""
@@ -939,34 +933,32 @@ def dashboard():
             with gr.Column():
                 refresh_btn = gr.Button("Refresh APR Data")
                 
-                # Create containers for plotly figures
-                per_agent_graph = gr.Plot(label="APR Per Agent")
-                combined_graph = gr.Plot(label="Combined APR (All Agents)")
+                # Create container for plotly figure (combined graph only)
+                combined_graph = gr.Plot(label="APR for All Agents")
                 
-                # Function to update both graphs
-                def update_apr_graphs():
-                    # Generate visualizations and get figure objects directly
-                    per_agent_fig, combined_fig, _ = generate_apr_visualizations()
-                    return per_agent_fig, combined_fig
+                # Function to update the graph
+                def update_apr_graph():
+                    # Generate visualization and get figure object directly
+                    combined_fig, _ = generate_apr_visualizations()
+                    return combined_fig
                 
                 # Set up the button click event
                 refresh_btn.click(
-                    fn=update_apr_graphs,
+                    fn=update_apr_graph,
                     inputs=[],
-                    outputs=[per_agent_graph, combined_graph]
+                    outputs=[combined_graph]
                 )
                 
-                # Initialize the graphs on load
-                # We'll use placeholder figures initially
+                # Initialize the graph on load
+                # We'll use placeholder figure initially
                 import plotly.graph_objects as go
                 placeholder_fig = go.Figure()
                 placeholder_fig.add_annotation(
-                    text="Click 'Refresh APR Data' to load APR graphs", 
+                    text="Click 'Refresh APR Data' to load APR graph", 
                     x=0.5, y=0.5, 
                     showarrow=False, 
                     font=dict(size=15)
                 )
-                per_agent_graph.value = placeholder_fig
                 combined_graph.value = placeholder_fig
         
     return demo