Update analytics_fetch_and_rendering.py

analytics_fetch_and_rendering.py

@@ -220,6 +220,75 @@ def plot_follower_gains(follower_data):
         plt.tight_layout()
         return fig
 
+# Add this function below `plot_follower_gains` in your code
+
+def plot_growth_rate(follower_data, total_follower_count):
+    """
+    Generates a line chart for Follower Growth Rate (%) on a monthly basis.
+    """
+    print(f"Plotting follower growth rate from data: {follower_data} and total followers: {total_follower_count}")
+    if not follower_data:
+        fig, ax = plt.subplots(figsize=(10, 5))
+        ax.text(0.5, 0.5, 'No data available to calculate growth rate.',
+                horizontalalignment='center', verticalalignment='center',
+                transform=ax.transAxes, fontsize=12, color='grey')
+        ax.set_title('Monthly Follower Growth Rate (%)')
+        ax.set_xticks([])
+        ax.set_yticks([])
+        plt.tight_layout()
+        return fig
+
+    try:
+        follower_data.sort(key=lambda x: x['date'])
+        dates = [entry['date'] for entry in follower_data]
+        total_gains = [entry['organic'] + entry['paid'] for entry in follower_data]
+
+        # Reconstruct monthly total followers backwards from the current total
+        followers_at_end_of_month = []
+        current_total = total_follower_count
+        for gain in reversed(total_gains):
+            followers_at_end_of_month.insert(0, current_total)
+            current_total -= gain
+
+        # followers_at_end_of_month now contains follower totals at end of each month
+        growth_rates = []
+        for i in range(1, len(followers_at_end_of_month)):
+            start = followers_at_end_of_month[i - 1]
+            end = followers_at_end_of_month[i]
+            rate = ((end - start) / start * 100) if start > 0 else 0
+            growth_rates.append(rate)
+
+        # Trim the first date (because we start calculating growth from second month)
+        rate_dates = dates[1:]
+
+        fig, ax = plt.subplots(figsize=(12, 6))
+        ax.plot(rate_dates, growth_rates, label='Growth Rate (%)', marker='o', linestyle='-', color='green')
+        ax.set_title('Monthly Follower Growth Rate (%)')
+        ax.set_xlabel('Month (YYYY-MM)')
+        ax.set_ylabel('Growth Rate (%)')
+        ax.grid(True, linestyle='--', alpha=0.6)
+        tick_frequency = max(1, len(rate_dates) // 10)
+        ax.set_xticks(rate_dates[::tick_frequency])
+        ax.tick_params(axis='x', rotation=45, labelsize=9)
+        ax.legend()
+
+        plt.tight_layout()
+        print("Successfully generated growth rate plot.")
+        return fig
+
+    except Exception as e:
+        print(f"ERROR generating growth rate plot: {e}")
+        tb = traceback.format_exc()
+        print(tb)
+        fig, ax = plt.subplots(figsize=(10, 5))
+        ax.text(0.5, 0.5, f'Error generating growth rate plot: {e}',
+                horizontalalignment='center', verticalalignment='center',
+                transform=ax.transAxes, fontsize=12, color='red', wrap=True)
+        ax.set_title('Follower Growth Rate Plot Error')
+        plt.tight_layout()
+        return fig
+
+
 
 def fetch_and_render_analytics(comm_client_id, comm_token):
     """Fetches analytics data and prepares updates for Gradio UI."""
@@ -254,8 +323,11 @@ def fetch_and_render_analytics(comm_client_id, comm_token):
         # If plot generation failed, plot_fig might contain an error message plot
         plot_output = gr.update(value=plot_fig, visible=True)
 
-        print("Analytics data fetched and processed successfully.")
-        return count_output, plot_output
+        # Generate follower growth rate plot
+        growth_rate_fig = plot_growth_rate(follower_gains_list)
+        growth_output = gr.update(value=growth_rate_fig, visible=True)
+
+        return count_output, plot_fig, growth_output
 
     except (ValueError, requests.exceptions.RequestException) as api_ve:
         # Catch specific API or configuration errors from fetch_analytics_data