Update ui_generators.py

ui_generators.py

@@ -147,23 +147,25 @@ def run_mentions_tab_display(token_state):
         html_parts.append(mentions_df_display[display_columns].head(20).to_html(escape=False, index=False, classes="table table-sm"))
 
     mentions_html_output = "\n".join(html_parts)
-    fig = None # Initialize fig to None
-    fig_plot_local = None # Use a local variable for the figure object
+    fig = None 
+    fig_plot_local = None 
     if not mentions_df.empty and "sentiment_label" in mentions_df.columns:
         try:
             fig_plot_local, ax = plt.subplots(figsize=(6,4))
             sentiment_counts = mentions_df["sentiment_label"].value_counts()
             sentiment_counts.plot(kind='bar', ax=ax, color=['#4CAF50', '#FFC107', '#F44336', '#9E9E9E', '#2196F3'])
-            ax.set_title("Mention Sentiment Distribution", y=1.08) # MODIFIED: Increased y parameter
+            ax.set_title("Mention Sentiment Distribution", y=1.03) # Adjusted y for Matplotlib title
             ax.set_ylabel("Count")
             plt.xticks(rotation=45, ha='right')
-            # fig_plot_local.subplots_adjust(top=0.88) # Alternative: Adjust top margin if y isn't enough
-            plt.tight_layout() # Ensure tight_layout is called
-            fig = fig_plot_local # Assign to fig for returning
+            
+            plt.tight_layout() 
+            fig_plot_local.subplots_adjust(top=0.90) # MODIFIED: Add space at the top of the figure
+                                                  # This pushes plot content down from Gradio's label
+            fig = fig_plot_local 
             logging.info("Mentions tab: Sentiment distribution plot generated.")
         except Exception as e:
             logging.error(f"Error generating mentions plot: {e}", exc_info=True)
-            fig = None # Ensure fig is None on error
+            fig = None 
         finally:
             if fig_plot_local and fig_plot_local is not plt:
                  plt.close(fig_plot_local)
@@ -221,14 +223,15 @@ def run_follower_stats_tab_display(token_state):
             fig_gains_local, ax_gains = plt.subplots(figsize=(10,5))
             ax_gains.plot(plot_data['_plot_month'], plot_data['organic'], marker='o', linestyle='-', label='Organic Gain')
             ax_gains.plot(plot_data['_plot_month'], plot_data['paid'], marker='x', linestyle='--', label='Paid Gain')
-            ax_gains.set_title("Monthly Follower Gains Over Time", y=1.08) # MODIFIED
+            ax_gains.set_title("Monthly Follower Gains Over Time", y=1.03) # Adjusted y for Matplotlib title
             ax_gains.set_ylabel("Follower Count")
             ax_gains.set_xlabel("Month (YYYY-MM)")
             plt.xticks(rotation=45, ha='right')
             ax_gains.legend()
             plt.grid(True, linestyle='--', alpha=0.7)
-            # fig_gains_local.subplots_adjust(top=0.88) # Alternative adjustment
+            
             plt.tight_layout()
+            fig_gains_local.subplots_adjust(top=0.90) # MODIFIED: Add space at the top of the figure
             plot_monthly_gains = fig_gains_local
             logging.info("Follower stats tab: Monthly gains plot generated.")
         else:
@@ -259,12 +262,13 @@ def run_follower_stats_tab_display(token_state):
             fig_seniority_local, ax_seniority = plt.subplots(figsize=(8,5))
             top_n_seniority = seniority_df_sorted.nlargest(10, FOLLOWER_STATS_ORGANIC_COLUMN)
             ax_seniority.bar(top_n_seniority[FOLLOWER_STATS_CATEGORY_COLUMN], top_n_seniority[FOLLOWER_STATS_ORGANIC_COLUMN], color='skyblue')
-            ax_seniority.set_title("Follower Distribution by Seniority (Top 10 Organic)", y=1.08) # MODIFIED
+            ax_seniority.set_title("Follower Distribution by Seniority (Top 10 Organic)", y=1.03) # Adjusted y
             ax_seniority.set_ylabel("Organic Follower Count")
             plt.xticks(rotation=45, ha='right')
             plt.grid(axis='y', linestyle='--', alpha=0.7)
-            # fig_seniority_local.subplots_adjust(top=0.85) # Alternative adjustment
+
             plt.tight_layout()
+            fig_seniority_local.subplots_adjust(top=0.88) # MODIFIED: Add space. Experiment with this value.
             plot_seniority_dist = fig_seniority_local
             logging.info("Follower stats tab: Seniority distribution plot generated.")
         else:
@@ -294,12 +298,13 @@ def run_follower_stats_tab_display(token_state):
             fig_industry_local, ax_industry = plt.subplots(figsize=(8,5))
             top_n_industry = industry_df_sorted.nlargest(10, FOLLOWER_STATS_ORGANIC_COLUMN)
             ax_industry.bar(top_n_industry[FOLLOWER_STATS_CATEGORY_COLUMN], top_n_industry[FOLLOWER_STATS_ORGANIC_COLUMN], color='lightcoral')
-            ax_industry.set_title("Follower Distribution by Industry (Top 10 Organic)", y=1.08) # MODIFIED
+            ax_industry.set_title("Follower Distribution by Industry (Top 10 Organic)", y=1.03) # Adjusted y
             ax_industry.set_ylabel("Organic Follower Count")
             plt.xticks(rotation=45, ha='right')
             plt.grid(axis='y', linestyle='--', alpha=0.7)
-            # fig_industry_local.subplots_adjust(top=0.85) # Alternative adjustment
+            
             plt.tight_layout()
+            fig_industry_local.subplots_adjust(top=0.88) # MODIFIED: Add space. Experiment with this value.
             plot_industry_dist = fig_industry_local
             logging.info("Follower stats tab: Industry distribution plot generated.")
         else:
@@ -330,9 +335,11 @@ def create_analytics_plot_panel(label, plot_id_str):
         explore_button (gr.Button): Explore/Zoom button.
         formula_button (gr.Button): Formula button.
     """
-    with gr.Column() as panel_col: # This will be the component that gets hidden/shown by explore
+    with gr.Column() as panel_col: 
         with gr.Row(equal_height=False, variant="panel"): 
-            plot_component = gr.Plot(label=label, scale=8) # The label here is for Gradio, not the Matplotlib title
+            # The gr.Plot's label is what you're referring to.
+            # We will adjust spacing using Matplotlib's subplots_adjust in the plot generation functions.
+            plot_component = gr.Plot(label=label, scale=8) 
             with gr.Column(scale=2, min_width=100): 
                 bomb_button = gr.Button(BOMB_ICON, variant="secondary", size="sm", elem_id=f"bomb_{plot_id_str}")
                 explore_button = gr.Button(EXPLORE_ICON, variant="secondary", size="sm", elem_id=f"explore_{plot_id_str}")
@@ -355,11 +362,9 @@ def build_analytics_tab_plot_area(plot_configs):
         # Start a new section if necessary
         if config1["section"] != current_section_title:
             current_section_title = config1["section"]
-            gr.Markdown(f"### {current_section_title}") # Section title
+            gr.Markdown(f"### {current_section_title}") 
 
-        # Create a new row for each pair of plots or a single plot
         with gr.Row(equal_height=False): 
-            # Process the first plot of the potential pair
             panel_col1, plot_comp1, bomb_btn1, explore_btn1, formula_btn1 = \
                 create_analytics_plot_panel(config1["label"], config1["id"])
             plot_ui_objects[config1["id"]] = {
@@ -370,7 +375,6 @@ def build_analytics_tab_plot_area(plot_configs):
             logging.debug(f"Created UI panel for plot_id: {config1['id']}")
             i += 1
 
-            # Check if there's a second plot for this row and if it's in the same section
             if i < len(plot_configs):
                 config2 = plot_configs[i]
                 if config2["section"] == current_section_title: