feat: Add task visualizations tab to the Gradio app

app.py

@@ -2,13 +2,20 @@ import gradio as gr
 import pandas as pd
 import logging
 import re
-
+from task_visualizations import TaskVisualizations
 
 logging.basicConfig(level=logging.INFO)
 
 
-def load_repo_df(data_path="data/repo_representations.jsonl"):
-    data = pd.read_json(data_path, lines=True, orient="records")
+class AppConfig:
+    repo_representations_path = "data/repo_representations.jsonl"
+    task_counts_path = "data/repos_task_counts.csv"
+    selected_task_counts_path = "data/selected_repos_task_counts.csv"
+    tasks_path = "data/paperswithcode_tasks.csv"
+
+
+def load_repo_df(repo_representations_path):
+    data = pd.read_json(repo_representations_path, lines=True, orient="records")
     return data.assign(
         text=data["text"]
         .str.replace(r"<img.*\/>", "", regex=True)
@@ -34,15 +41,7 @@ def display_representations(repo, representation1, representation2):
     return text1, text2
 
 
-## main
-repos_df = load_repo_df()
-repos = list(repos_df["repo_name"].unique())
-representation_types = list(repos_df["representation"].unique())
-logging.info(f"found {len(repos)} repositories")
-logging.info(f"representation types: {representation_types}")
-
-with gr.Blocks() as demo:
-    gr.Markdown("# Repository Representations Viewer")
+def setup_repository_representations_tab(repos, representation_types):
     gr.Markdown("Select a repository and two representation types to compare them.")
 
     with gr.Row():
@@ -93,4 +92,25 @@ with gr.Blocks() as demo:
             outputs=[text1, text2],
         )
 
+
+## main
+repos_df = load_repo_df()
+repos = list(repos_df["repo_name"].unique())
+representation_types = list(repos_df["representation"].unique())
+logging.info(f"found {len(repos)} repositories")
+logging.info(f"representation types: {representation_types}")
+task_visualizations = TaskVisualizations(
+    AppConfig.task_counts_path,
+    AppConfig.selected_task_counts_path,
+    AppConfig.tasks_path,
+)
+
+with gr.Blocks() as demo:
+    with gr.Tab("Explore Repository Representations"):
+        setup_repository_representations_tab(repos, representation_types)
+    with gr.Tab("Explore PapersWithCode Tasks"):
+        with open("plots/pwc_area_tasks_sunburst.html") as f:
+            sunburst_tasks = f.read()
+        gr.HTML(sunburst_tasks)
+
 demo.launch()