Add count of category and PMC IDs to data output in extract_arxiv_data and extract_pmc_data functions

arvix.py

@@ -92,7 +92,7 @@ def extract_arxiv_data():
                 break
         while len(category_ids) < 4:
             category_ids.add(random.choice(list(used_ids)))
-        data[category] = {"ids": list(category_ids)}
+        data[category] = {"ids": list(category_ids), "count": len(category_ids)}
     if not utils.upload_datafile('arxiv.txt'):
         raise Exception("Failed to upload datafile")
     return json.dumps(data, indent=4, ensure_ascii=False)

pmc.py

@@ -79,7 +79,7 @@ def extract_pmc_data():
             if not utils.check_data_in_file(pmcid, 'pmc.txt'):
                 utils.write_data_to_file(pmcid, 'pmc.txt')
                 pmc_ids.append(pmcid)
-        pmc_data[topic] = {"ids": pmc_ids}
+        pmc_data[topic] = {"ids": pmc_ids, "count": len(pmc_ids)}
     if not utils.upload_datafile('pmc.txt'):
         raise Exception("Failed to upload datafile")
     return json.dumps(pmc_data, indent=4, ensure_ascii=False)