Update arvix.py

arvix.py

@@ -54,7 +54,7 @@ def extract_data(category):
     if category in ["hep-ex", "hep-lat", "hep-ph", "hep-th"]:
         category_list = []
         for id in data:
-            if len(category_list) >= 3:
+            if len(category_list) >= 1:
                 break
             if tools.check_data_in_file(id, 'arxiv.txt'):
                 continue
@@ -65,7 +65,7 @@ def extract_data(category):
             tools.write_data_to_file(id, 'arxiv.txt')
     else:
         for id in data:
-            if len(sanitized_data) >= 12:
+            if len(sanitized_data) >= 4:
                 break
             if tools.check_data_in_file(id, 'arxiv.txt'):
                 continue
@@ -104,7 +104,7 @@ def extract_arxiv_data():
         temp_id_storage = []
         for subcategory in subcategories:
             ids = extract_data(subcategory)
-            if len(ids) == 12:
+            if len(ids) == 4:
                 for id in ids:
                     temp_id_storage.append(id)
             else:
@@ -113,8 +113,8 @@ def extract_arxiv_data():
         for temp_id in temp_id_storage:
             all_ids.append(temp_id)
         random.shuffle(all_ids)
-        if len(all_ids) > 12:
-            print(f"Found more than 12 papers for {category}.")
+        if len(all_ids) > 4:
+            print(f"Found more than 4 papers for {category}.")
             all_ids = all_ids
         category_data['count'] = len(all_ids)
         category_data['ids'] = all_ids