visualizations update

gradio_tabs.py

@@ -87,7 +87,7 @@ def setup_repository_representations_tab(repos_df, repos, representation_types):
         )
         tasks_display = tasks if tasks is not None else "No tasks available"
         return (
-            "## Repository PapersWithCode tasks:\n" + tasks_display,
+            "## Repository PapersWithCode tasks:\n" + " ".join(tasks_display),
             f"### Representation 1: {representation1}\n\n{text1_content}",
             f"### Representation 2: {representation2}\n\n{text2_content}",
         )

text_visualization.py

@@ -84,21 +84,21 @@ class EmbeddingVisualizer(BaseModel):
 
         display_df = display_df.sort_values("representation", ascending=False)
         readme_df = display_df[
-            display_df["representation"].isin(
-                ["readme", "code2doc_generated_readme", "task"]
+            display_df["representation"].apply(
+                lambda r: "readme" in r.lower() or r == "task"
             )
         ]
         raw_df = display_df[
             display_df["representation"].isin(
-                ["dependency_signature", "selected_code", "task"]
+                ["dependency signature", "selected code", "task"]
             )
         ]
         dependency_df = display_df[
             display_df["representation"].isin(
                 [
-                    "repository_signature",
-                    "dependency_signature",
-                    "generated_tasks",
+                    "repository signature",
+                    "dependency signature",
+                    "generated tasks",
                     "task",
                 ]
             )