Separate graphs for Pearl and QS markets

app.py

@@ -631,10 +631,18 @@ with demo:
                     active_traders, market_creator="quickstart"
                 )
             with gr.Row():
-                gr.Markdown(" # Daily active agents")
+                gr.Markdown(" # Daily active agents in Pearl markets")
             with gr.Row():
                 rolling_avg_plot = plot_rolling_average(
                     active_traders,
+                    market_creator="pearl",
+                )
+            with gr.Row():
+                gr.Markdown(" # Daily active agents in QS markets")
+            with gr.Row():
+                rolling_avg_plot = plot_rolling_average(
+                    active_traders,
+                    market_creator="quickstart",
                 )
 
         with gr.TabItem("📉 Markets Kullback–Leibler divergence"):

tabs/trader_plots.py

@@ -380,13 +380,14 @@ def plot_rolling_average(
     market_creator: str = None,
 ) -> gr.Plot:
     """Function to plot the rolling average of active traders for the different categories and markets"""
-
-    rolling_avg_df = get_sevenday_rolling_average(active_traders_df)
-    print(rolling_avg_df.head())
     if market_creator is not None:
-        rolling_avg_df = rolling_avg_df.loc[
-            rolling_avg_df["market_creator"] == market_creator
+        filtered_traders_df = active_traders_df.loc[
+            active_traders_df["market_creator"] == market_creator
         ]
+        rolling_avg_df = get_sevenday_rolling_average(filtered_traders_df)
+    else:
+        rolling_avg_df = get_sevenday_rolling_average(active_traders_df)
+    print(rolling_avg_df.head())
 
     # Ensure 'creation_date' is a column, not an index
     if "creation_date" not in rolling_avg_df.columns: