improved displaying representations

gradio_tabs.py

@@ -7,6 +7,7 @@ import gradio as gr
 
 def display_representations(repos_df, repo, representation1, representation2):
     repo_data = repos_df[repos_df["repo_name"] == repo]
+    tasks = repo_data["tasks"].iloc[0]
     logging.info(f"repo_data: {repo_data}")
     text1 = (
         repo_data[repo_data["representation"] == representation1]["text"].iloc[0]
@@ -19,7 +20,7 @@ def display_representations(repos_df, repo, representation1, representation2):
         else "No data available"
     )
 
-    return text1, text2
+    return tasks, text1, text2
 
 
 def get_representation_wordclouds(representations, repos_df):
@@ -60,6 +61,7 @@ def setup_repository_representations_tab(repos_df, repos, representation_types):
             value="code2doc_generated_readme",
         )
 
+    displayed_tasks = gr.Markdown(elem_id="tasks")
     with gr.Row():
         with gr.Column(
             elem_id="column1",
@@ -77,24 +79,25 @@ def setup_repository_representations_tab(repos_df, repos, representation_types):
             text2 = gr.Markdown()
 
     def update_representations(repo, representation1, representation2):
-        text1_content, text2_content = display_representations(
+        tasks, text1_content, text2_content = display_representations(
             repos_df, repo, representation1, representation2
         )
         return (
+            "## Repository PapersWithCode tasks:\n" + ", ".join(tasks),
             f"### Representation 1: {representation1}\n\n{text1_content}",
             f"### Representation 2: {representation2}\n\n{text2_content}",
         )
 
     # Initial call to populate textboxes with default values
-    text1.value, text2.value = update_representations(
-        repos[0], "readme", "generated_readme"
+    displayed_tasks.value, text1.value, text2.value = update_representations(
+        repos[0], "readme", "code2doc_generated_readme"
     )
 
     for component in [repo, representation1, representation2]:
         component.change(
             fn=update_representations,
             inputs=[repo, representation1, representation2],
-            outputs=[text1, text2],
+            outputs=[displayed_tasks, text1, text2],
         )