Update plot_graph.py

plot_graph.py

@@ -36,8 +36,7 @@ def plot_follower_comparison(author_names, dataset_path="followers-leaderboard",
     else:
         dfs = global_dataset_dfs[dataset_path]
 
-    plt.figure(figsize=(14, 8))
-    ax = plt.gca()
+    fig, ax = plt.subplots(figsize=(14, 8))  # Create a new figure
     valid_authors = []
 
     for idx, author in enumerate(author_names):
@@ -73,23 +72,22 @@ def plot_follower_comparison(author_names, dataset_path="followers-leaderboard",
 
     if not valid_authors:
         print("No valid data found for any authors.")
-        return
+        return None
 
     # Plot configuration
-    plt.title(f'Follower Trend Comparison ({", ".join(valid_authors)})', pad=20)
-    plt.xlabel('Date', labelpad=15)
-    plt.ylabel('Followers', labelpad=15)
-    plt.ylim(bottom=0)
-    plt.grid(True, alpha=0.25)
+    ax.set_title(f'Follower Trend Comparison ({", ".join(valid_authors)})', pad=20)
+    ax.set_xlabel('Date', labelpad=15)
+    ax.set_ylabel('Followers', labelpad=15)
+    ax.set_ylim(bottom=0)
+    ax.grid(True, alpha=0.25)
 
     # Date formatting
     ax.xaxis.set_major_formatter(plt.matplotlib.dates.DateFormatter('%d-%m-%Y'))
     plt.xticks(rotation=45, ha='right')
 
     # Legend and layout
-    plt.legend(loc='upper left', bbox_to_anchor=(1, 1), framealpha=0.9)
+    ax.legend(loc='upper left', bbox_to_anchor=(1, 1), framealpha=0.9)
     plt.tight_layout()
-    #plt.show()
-    fig = plt.gcf()
-    plt.close(fig) 
-    return fig
+
+    return fig  # Return the figure instead of plt
+