Update title casing in app and optimize ID extraction logic in arvix

app.py

@@ -14,7 +14,7 @@ theme = gr.themes.Soft(
     ],
 )
 
-with gr.Blocks(theme=theme, title="fetch Research Paper IDS") as app:
+with gr.Blocks(theme=theme, title="Fetch Research Paper IDS") as app:
     with gr.Row():
         with gr.Column():
             user_access_key = gr.Textbox(label="Access Key", placeholder="Enter your access key", type="password")

arvix.py

@@ -73,7 +73,6 @@ def extract_data(category):
                 tools.write_data_to_file(id, 'arxiv.txt')
                 sanitized_data.append(id)
     random.shuffle(sanitized_data)
-    print(len(sanitized_data))
     return sanitized_data
 
 def extract_arxiv_data():
@@ -102,13 +101,21 @@ def extract_arxiv_data():
     for category, subcategories in categories.items():
         category_data = {}
         all_ids = []
+        temp_id_storage = []
         for subcategory in subcategories:
             ids = extract_data(subcategory)
-            for id in ids:
-                all_ids.append(id)
+            if len(ids) == 3:
+                for id in ids:
+                    temp_id_storage.append(id)
+            else:
+                for id in ids:
+                    all_ids.append(id)
+        for temp_id in temp_id_storage:
+            all_ids.append(temp_id)
+        random.shuffle(all_ids)
+        print(len(all_ids))
         if len(all_ids) > 12:
             print(f"Found more than 12 papers for {category}. Randomly selecting 12 papers.")
-            random.shuffle(all_ids)
             all_ids = all_ids[:12]
         category_data['count'] = len(all_ids)
         category_data['ids'] = all_ids