Update ui_generators.py

ui_generators.py

@@ -147,28 +147,26 @@ 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
+    fig = None # Initialize fig to None
+    fig_plot_local = None # Use a local variable for the figure object
     if not mentions_df.empty and "sentiment_label" in mentions_df.columns:
         try:
-            fig_plot, ax = plt.subplots(figsize=(6,4))
+            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.03) # MODIFIED: Added y parameter
+            ax.set_title("Mention Sentiment Distribution", y=1.08) # MODIFIED: Increased y parameter
             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
+            fig = fig_plot_local # Assign to fig for returning
             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
         finally:
-            # Only close if fig_plot was successfully created and is not the global plt context
-            if fig_plot and fig_plot is not plt: # fig_plot is the figure object from subplots
-                 plt.close(fig_plot)
-            # Fallback if somehow a global figure was left open without being assigned to fig_plot
-            # elif fig is None and plt.get_fignums():
-            #     plt.close('all') # This might be too aggressive if other plots are intended to be open
+            if fig_plot_local and fig_plot_local is not plt:
+                 plt.close(fig_plot_local)
     return mentions_html_output, fig
 
 
@@ -192,7 +190,7 @@ def run_follower_stats_tab_display(token_state):
     plot_industry_dist = None
 
     # Monthly Gains Plot
-    fig_gains_local = None # Use local var for figure to manage closing
+    fig_gains_local = None
     try:
         monthly_gains_df = follower_stats_df[
             (follower_stats_df[FOLLOWER_STATS_TYPE_COLUMN] == 'follower_gains_monthly') &
@@ -216,7 +214,6 @@ def run_follower_stats_tab_display(token_state):
                 organic=(FOLLOWER_STATS_ORGANIC_COLUMN, 'sum'),
                 paid=(FOLLOWER_STATS_PAID_COLUMN, 'sum')
             ).reset_index()
-            # Ensure months are sorted correctly for plotting after groupby
             plot_data['_plot_month_dt'] = pd.to_datetime(plot_data['_plot_month'], format=UI_MONTH_FORMAT)
             plot_data = plot_data.sort_values(by='_plot_month_dt')
 
@@ -224,23 +221,24 @@ 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.03) # MODIFIED
+            ax_gains.set_title("Monthly Follower Gains Over Time", y=1.08) # MODIFIED
             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()
-            plot_monthly_gains = fig_gains_local # Assign to the return variable
+            plot_monthly_gains = fig_gains_local
             logging.info("Follower stats tab: Monthly gains plot generated.")
         else:
             html_parts.append("<p>No monthly follower gain data available or required columns missing.</p>")
     except Exception as e:
         logging.error(f"Error processing or plotting monthly gains: {e}", exc_info=True)
         html_parts.append("<p>Error displaying monthly follower gain data.</p>")
-        plot_monthly_gains = None # Ensure it's None on error
+        plot_monthly_gains = None
     finally:
-        if fig_gains_local and fig_gains_local is not plt: # Check local var
+        if fig_gains_local and fig_gains_local is not plt:
              plt.close(fig_gains_local)
     html_parts.append("<hr/>")
 
@@ -261,10 +259,11 @@ 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.03) # MODIFIED
+            ax_seniority.set_title("Follower Distribution by Seniority (Top 10 Organic)", y=1.08) # MODIFIED
             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()
             plot_seniority_dist = fig_seniority_local
             logging.info("Follower stats tab: Seniority distribution plot generated.")
@@ -295,10 +294,11 @@ 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.03) # MODIFIED
+            ax_industry.set_title("Follower Distribution by Industry (Top 10 Organic)", y=1.08) # MODIFIED
             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()
             plot_industry_dist = fig_industry_local
             logging.info("Follower stats tab: Industry distribution plot generated.")
@@ -331,9 +331,9 @@ def create_analytics_plot_panel(label, plot_id_str):
         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.Row(equal_height=False, variant="panel"): # Removed variant="compact" as it might affect spacing
+        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
-            with gr.Column(scale=2, min_width=100): # min_width for button column
+            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}")
                 formula_button = gr.Button(FORMULA_ICON, variant="secondary", size="sm", elem_id=f"formula_{plot_id_str}")
@@ -358,14 +358,14 @@ def build_analytics_tab_plot_area(plot_configs):
             gr.Markdown(f"### {current_section_title}") # Section title
 
         # Create a new row for each pair of plots or a single plot
-        with gr.Row(equal_height=False): # A new row for one or two plots
+        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"]] = {
                 "plot_component": plot_comp1, "bomb_button": bomb_btn1,
                 "explore_button": explore_btn1, "formula_button": formula_btn1,
-                "label": config1["label"], "panel_component": panel_col1 # panel_col1 is the gr.Column from create_analytics_plot_panel
+                "label": config1["label"], "panel_component": panel_col1 
             }
             logging.debug(f"Created UI panel for plot_id: {config1['id']}")
             i += 1
@@ -383,10 +383,7 @@ def build_analytics_tab_plot_area(plot_configs):
                     }
                     logging.debug(f"Created UI panel for plot_id: {config2['id']} in same row")
                     i += 1
-                # If config2 is in a new section, the row ends with config1.
-                # config2 will be processed as the first plot in a new row in the next iteration.
-            # Row ends here. Next iteration will start a new gr.Row if needed.
-
+                
     logging.info(f"Finished building plot area. Total plot objects: {len(plot_ui_objects)}")
     if len(plot_ui_objects) != len(plot_configs):
         logging.error(f"MISMATCH: Expected {len(plot_configs)} plot objects, but created {len(plot_ui_objects)}.")