Update system_stats.py

system_stats.py

@@ -34,23 +34,19 @@ def get_stats(file_path):
 
 
 
-file_list = ['doc/page_1.txt', 'doc/page_1.txt', 'doc/page_2.table', 'doc/page3_img.table', 'doc/page3_img2.table', 'doc/page3_img3.table']
+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 file_path in file_list:
-    # Extracting page number from the file name
-    page_number = file_path.split('/')[-1].split('_')[1].split('.')[0]
-    
-    # Checking if the file is a text file
-    if file_path.endswith('.txt'):
-        # Opening the text file and reading its content
-        with open(file_path, 'r') as file:
-            content = file.read()
-        
-        # Adding the content to the dictionary with page number as key
-        page_dict[page_number] = content
-
-print(page_dict)
-
-  
+for page in pages:
+    page_number = page.split('/')[-1].split('_')[0].replace('page', '')
+    if page_number not in page_dict:
+        page_dict[page_number] = []
+    page_dict[page_number].append(page)
+
+print(page_dict)