Update ui/analytics_plot_generator.py

ui/analytics_plot_generator.py

@@ -341,8 +341,8 @@ def generate_content_topic_breakdown_plot(df, topics_col='li_eb_labels', **kwarg
 
 def generate_impressions_lineplot_component_data(df, post_details_map_ref):
     """
-    Generates a DataFrame for the Gradio LinePlot and updates the post details map.
-    This function will be called by the main update logic.
+    Generates a DataFrame for the Gradio LinePlot, updates the post details map,
+    and formats columns for a rich tooltip.
     """
     plot_df, post_map = generate_impressions_scatter_plot_data(
         df, 
@@ -357,11 +357,21 @@ def generate_impressions_lineplot_component_data(df, post_details_map_ref):
     # If the df is empty, return an empty one to clear the plot while preserving the component type
     if plot_df.empty:
         logging.info("Returning an empty DataFrame to clear the line plot.")
-        return pd.DataFrame(columns=['date', 'impressions'])
+        # Return all columns the component expects to avoid errors
+        return pd.DataFrame(columns=['date', 'impressions', 'label', 'tooltip_date', 'tooltip_impressions'])
 
-    logging.info(f"Generated LinePlot data with {len(plot_df)} points.")
+    # Format data for a better tooltip
+    # Ensure 'date' is datetime for correct plotting
+    plot_df['date'] = pd.to_datetime(plot_df['date'])
+    
+    # Create a formatted date string for the tooltip
+    plot_df['tooltip_date'] = plot_df['date'].dt.strftime('%B %d, %Y')
+    
+    # Create a formatted impressions string for the tooltip
+    plot_df['tooltip_impressions'] = 'Impressions: ' + plot_df['impressions'].astype(str)
+
+    logging.info(f"Generated LinePlot data with {len(plot_df)} points and formatted tooltip.")
     
-    # Return the DataFrame itself, not a gr.update() object
     return plot_df
 
 def update_analytics_plots_figures(token_state_value, date_filter_option, custom_start_date, custom_end_date, current_plot_configs):