Update services/analytics_handlers.py

services/analytics_handlers.py

@@ -99,7 +99,7 @@ class AnalyticsHandlers:
                                           ):
         logging.info(f"Refreshing analytics graph UI. Filter: {date_filter_val}. Token set: {'yes' if current_token_state_val.get('token') else 'no'}")
         start_time = time.time()
-
+    
         # Call the function that generates plot figures and summaries
         # Ensure update_analytics_plots_figures is adapted to return:
         # status_msg (str), figures_dict (dict: {plot_id: fig}), summaries_dict (dict: {plot_id: summary_text})
@@ -112,22 +112,34 @@ class AnalyticsHandlers:
         )
         # Expected: status_msg, list_of_figures, dict_of_plot_summaries
         # Original: status_msg, gen_figs (list), new_summaries (dict)
-
+    
         status_msg = plot_gen_results[0]
-        gen_figs_list = plot_gen_results[1] # This should be a list of figures in order of plot_configs
+        gen_figs_result = plot_gen_results[1] # This might be a single figure or a list
         new_summaries_dict = plot_gen_results[2] # This should be a dict {plot_id: summary}
-
+    
+        # Handle the case where gen_figs_result might be a single figure or a list
+        if isinstance(gen_figs_result, list):
+            gen_figs_list = gen_figs_result
+        else:
+            # If it's a single figure, wrap it in a list
+            gen_figs_list = [gen_figs_result]
+            logging.warning(f"Expected list of figures but got single figure of type {type(gen_figs_result)}")
+    
         all_updates = [gr.update(value=status_msg)] # For analytics_status_md
-
+    
         # Update plot components with new figures
         if len(gen_figs_list) == len(self.plot_configs):
             for fig in gen_figs_list:
                 all_updates.append(fig) # fig itself is the update for gr.Plot
         else:
             logging.error(f"Figure list length mismatch: got {len(gen_figs_list)}, expected {len(self.plot_configs)}")
-            for _ in self.plot_configs:
-                all_updates.append(create_placeholder_plot("Error", "Figura mancante"))
-
+            # If we have fewer figures than expected, pad with placeholders
+            for i in range(len(self.plot_configs)):
+                if i < len(gen_figs_list):
+                    all_updates.append(gen_figs_list[i])
+                else:
+                    all_updates.append(create_placeholder_plot("Error", "Figura mancante"))
+    
         # Reset action panel UI elements
         all_updates.extend([
             gr.update(visible=False),  # global_actions_column_ui
@@ -159,7 +171,7 @@ class AnalyticsHandlers:
             ])
             
         all_updates.append(None) # explored_plot_id_state (reset)
-
+    
         # Reset section title visibility
         for _ in self.unique_ordered_sections:
             all_updates.append(gr.update(visible=True))
@@ -173,7 +185,7 @@ class AnalyticsHandlers:
             logging.error(f"Output length mismatch in refresh_analytics_graphs_ui. Got {len(all_updates)}, expected {expected_len}")
             # Pad with gr.update() if lengths don't match, to avoid Gradio errors, though this indicates a logic flaw.
             all_updates.extend([gr.update()] * (expected_len - len(all_updates)))
-
+    
         return tuple(all_updates)
 
     def _get_action_panel_outputs_list(self):