formatting

app.py

@@ -85,29 +85,30 @@ def get_papers_for_collection(collection_slug: str):
     collection = group_collection_items(collection_slug)
     collection_datasets = collection.get("datasets", None)
     collection_models = collection.get("models", None)
-    dataset_papers = []
-    model_papers = []
+    dataset_papers = defaultdict(dict)
+    model_papers = defaultdict(dict)
     if collection_datasets is not None:
         for dataset in collection_datasets:
-            if arxiv_id := dataset_to_arxiv_id.get(dataset.item_id, None):
+            if arxiv_ids := dataset_to_arxiv_id.get(dataset.item_id, None):
                 data = {
-                    "model": dataset.item_id,
-                    "arxiv_id": arxiv_id,
-                    "hub_paper_link": f"https://huggingface.co/papers/{arxiv_id}",
+                    "arxiv_ids": arxiv_ids,
+                    "hub_paper_links": [
+                        f"https://huggingface.co/papers/{arxiv_id}"
+                        for arxiv_id in arxiv_ids
+                    ],
                 }
-                dataset_papers.append(arxiv_id)
+                dataset_papers[dataset.item_id] = data
     if collection_models is not None:
         for model in collection.get("models", []):
             if arxiv_ids := models_to_arxiv_id.get(model.item_id, None):
                 data = {
-                    "model": model.item_id,
                     "arxiv_ids": arxiv_ids,
                     "hub_paper_links": [
                         f"https://huggingface.co/papers/{arxiv_id}"
                         for arxiv_id in arxiv_ids
                     ],
                 }
-                model_papers.append(data)
+                model_papers[model.item_id] = data
     return {"datasets": dataset_papers, "models": model_papers}