chore: Update plot_datasets_growth to include growth rate option

app.py

@@ -44,7 +44,7 @@ def generate_dashboard(data, grouped, framework):
     return dashboard
 
 
-def plot_datasets_growth(data, framework):
+def plot_datasets_growth(data, framework, show_growth_rate=True):
     df = pd.DataFrame(data)
     df["createdAt"] = pd.to_datetime(df["createdAt"])
     df["month"] = df["createdAt"].dt.to_period("M").astype(str)
@@ -60,20 +60,28 @@ def plot_datasets_growth(data, framework):
     fig.update_layout(
         xaxis_title="Month",
         yaxis_title="Cumulative Number of Datasets",
-        yaxis=dict(title=f"Cumulative Number of Datasets ({framework}"),
-        yaxis2=dict(
-            title="Month-over-Month Growth Rate",
-            overlaying="y",
-            side="right",
-            tickformat=",.0%",
-        ),
+        yaxis=dict(title=f"Cumulative Number of Datasets ({framework})"),
         legend=dict(
             title="", orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1
         ),
     )
-    fig.add_scatter(
-        x=df_counts["month"], y=df_counts["growth_rate"], name="Growth Rate", yaxis="y2"
-    )
+
+    if show_growth_rate:
+        fig.update_layout(
+            yaxis2=dict(
+                title="Month-over-Month Growth Rate",
+                overlaying="y",
+                side="right",
+                tickformat=",.0%",
+            )
+        )
+        fig.add_scatter(
+            x=df_counts["month"],
+            y=df_counts["growth_rate"],
+            name="Growth Rate",
+            yaxis="y2",
+        )
+
     fig.update_layout(
         title={
             "text": f"Dataset Growth for {framework} datasets",
@@ -89,7 +97,8 @@ def plot_datasets_growth(data, framework):
                 y=0.85,
                 xref="paper",
                 yref="paper",
-                text="Cumulative number of datasets and month-over-month growth rate",
+                text="Cumulative number of datasets"
+                + (" and month-over-month growth rate" if show_growth_rate else ""),
                 showarrow=False,
                 font=dict(size=14),
             )
@@ -98,10 +107,10 @@ def plot_datasets_growth(data, framework):
     return fig
 
 
-def update_dashboard(framework):
+def update_dashboard(framework, show_growth_rate=True):
     data, grouped = fetch_data(framework)
     dashboard = generate_dashboard(data, grouped, framework)
-    fig = plot_datasets_growth(data, framework)
+    fig = plot_datasets_growth(data, framework, show_growth_rate)
     return fig, dashboard
 
 
@@ -115,8 +124,14 @@ with gr.Blocks() as demo:
         allow_custom_value=True,
         label="Select a framework/tag",
     )
+    show_growth_rate = gr.Checkbox(True, label="Show growth rate")
     plot = gr.Plot(label="Growth of datasets over time")
     markdown = gr.Markdown(label="summary")
-    framework.change(update_dashboard, inputs=[framework], outputs=[plot, markdown])
+    framework.change(
+        update_dashboard, inputs=[framework, show_growth_rate], outputs=[plot, markdown]
+    )
+    show_growth_rate.change(
+        update_dashboard, inputs=[framework, show_growth_rate], outputs=[plot, markdown]
+    )
 
 demo.launch()