Update arvix.py

arvix.py

@@ -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) >= 2:
+            if len(sanitized_data) >= 3:
                 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) == 2:
+            if len(ids) == 3:
                 for id in ids:
                     temp_id_storage.append(id)
             else:
@@ -113,9 +113,9 @@ def extract_arxiv_data():
         for temp_id in temp_id_storage:
             all_ids.append(temp_id)
         random.shuffle(all_ids)
-        if len(all_ids) > 2:
-            print(f"Found more than 4 papers for {category}.")
-            all_ids = all_ids[:1]
+        if len(all_ids) > 3:
+            print(f"Found more than 3 papers for {category}.")
+            all_ids = all_ids[:2]
         category_data['count'] = len(all_ids)
         category_data['ids'] = all_ids
         data[category] = category_data