Update system_stats.py

system_stats.py

@@ -34,19 +34,24 @@ def get_stats(file_path):
 
 
 
-pages = ['doc/page_1.txt',
-         'doc/page_1.txt',
-         'doc/page_2.table',
-         'doc/page3_img.table',
-         'doc/page3_img2.table',
-         'doc/page3_img3.table']
-
 page_dict = {}
 
-for page in pages:
-    page_number = page.split('/')[-1].split('_')[0].replace('page', '')
+page_files = [
+    'doc/344_page_1_txt.png',
+    'doc/344_page_1_table.png',
+    'doc/5433_page_2_table.png',
+    'doc/54_page3_img.png',
+    'doc/32_page3_txt.png',
+    'doc/32_page3_table.png'
+]
+
+for file in page_files:
+    # Extracting page number from file name
+    page_number = file.split('_')[1].replace('page', '')
+    
     if page_number not in page_dict:
         page_dict[page_number] = []
-    page_dict[page_number].append(page)
+    
+    page_dict[page_number].append(file)
 
 print(page_dict)