Update plot_graph.py

plot_graph.py

@@ -8,17 +8,12 @@ from scipy.ndimage import gaussian_filter1d
 
 snapshot_download("Weyaxi/followers-leaderboard", local_dir="followers-leaderboard", repo_type="dataset", max_workers=32)
 
-
 global_dataset_dfs = {}
-COLORS = ['#1f77b4', '#ff7f0e', '#2ca02c', '#d62728', '#9467bd',
-          '#8c564b', '#e377c2', '#7f7f7f', '#bcbd22', '#17becf']
-
 
-def plot_follower_comparison(author_names, dataset_path="followers-leaderboard", smooth_sigma=1.5):
+def gr_plot_follower_comparison(author_names, dataset_path="followers-leaderboard", smooth_sigma=1.5):
     if isinstance(author_names, str):
         author_names = [author_names]
 
-    print(author_names)
     # Load or retrieve cached DataFrames
     if dataset_path not in global_dataset_dfs:
         dfs = {}
@@ -37,10 +32,9 @@ def plot_follower_comparison(author_names, dataset_path="followers-leaderboard",
     else:
         dfs = global_dataset_dfs[dataset_path]
 
-    fig, ax = plt.subplots(figsize=(14, 8))  # Create a new figure
-    valid_authors = []
-
-    for idx, author in enumerate(author_names):
+    data = []
+    
+    for author in author_names:
         date_count = {}
         for date, df in dfs.items():
             if author in df['Author'].values:
@@ -55,40 +49,10 @@ def plot_follower_comparison(author_names, dataset_path="followers-leaderboard",
             if smooth_sigma > 0 and len(counts) > 1:
                 counts = gaussian_filter1d(counts, sigma=smooth_sigma)
 
-            # Plot styling
-            color = COLORS[idx % len(COLORS)]
-            style = ['-', '--', '-.', ':'][idx % 4]
-            marker = ['o', 's', 'D', '^', 'v'][idx % 5] if idx < 5 else None
-
-            ax.plot(dates, counts,
-                    linestyle=style,
-                    color=color,
-                    marker=marker,
-                    markersize=7 if marker else 0,
-                    linewidth=2.5,
-                    markevery=0.1,
-                    alpha=0.9,
-                    label=author)
-            valid_authors.append(author)
-
-    if not valid_authors:
-        print("No valid data found for any authors.")
-        return None
-
-    # Plot configuration
-    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')
+            for d, c in zip(dates, counts):
+                data.append({"x": d, "y": c, "author": author})
 
-    # Legend and layout
-    ax.legend(loc='upper left', bbox_to_anchor=(1, 1), framealpha=0.9)
-    plt.tight_layout()
+    if not data:
+        return pd.DataFrame(columns=["x", "y", "author"])
 
-    return fig  # Return the figure instead of plt
-                
+    return pd.DataFrame(data)