feat: Add separate sliders for all and selected repositories in the PapersWithCode tasks tab

app.py

@@ -119,12 +119,19 @@ with gr.Blocks() as demo:
         gr.Markdown(task_counts_description)
 
         with gr.Row():
-            min_task_counts_slider = gr.Slider(
+            min_task_counts_slider_all = gr.Slider(
                 minimum=10,
                 maximum=1000,
                 value=100,
                 step=10,
-                label="Minimum Task Count",
+                label="Minimum Task Count (All Repositories)",
+            )
+            min_task_counts_slider_selected = gr.Slider(
+                minimum=10,
+                maximum=1000,
+                value=100,
+                step=10,
+                label="Minimum Task Count (Selected Repositories)",
             )
             update_button = gr.Button("Update Plots")
 
@@ -134,7 +141,7 @@ with gr.Blocks() as demo:
 
         update_button.click(
             fn=task_visualizations.get_tasks_sunbursts,
-            inputs=[min_task_counts_slider],
+            inputs=[min_task_counts_slider_all, min_task_counts_slider_selected],
             outputs=[all_repos_tasks_plot, selected_repos_tasks_plot],
         )
 

task_visualizations.py

@@ -57,16 +57,22 @@ class TaskVisualizations:
         )
         return displayed_tasks_with_areas_df
 
-    def get_tasks_sunburst(self, min_task_count, which_df="selected"):
-        if which_df == "selected":
-            df = self.selected_tasks_with_areas_df
-        else:
-            df = self.tasks_with_areas_df
+    def get_tasks_sunbursts(self, min_task_count_all, min_task_count_selected):
+        all_df = self.tasks_with_areas_df
+        selected_df = self.selected_tasks_with_areas_df
 
-        displayed_tasks_with_areas_df = self.get_displayed_tasks_with_areas_df(
-            df, min_task_count
+        displayed_tasks_all_df = self.get_displayed_tasks_with_areas_df(
+            all_df, min_task_count_all
+        )
+        displayed_tasks_selected_df = self.get_displayed_tasks_with_areas_df(
+            selected_df, min_task_count_selected
         )
 
-        return px.sunburst(
-            displayed_tasks_with_areas_df, path=["area", "task"], values="count"
+        all_sunburst = px.sunburst(
+            displayed_tasks_all_df, path=["area", "task"], values="count"
+        )
+        selected_sunburst = px.sunburst(
+            displayed_tasks_selected_df, path=["area", "task"], values="count"
         )
+
+        return all_sunburst, selected_sunburst