File size: 3,042 Bytes
b692870 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import os
import abc_1
import shutil
def get_subfolder_names(folder_path):
try:
subfolders = [f for f in os.listdir(folder_path) if os.path.isdir(os.path.join(folder_path, f))]
return subfolders
except FileNotFoundError:
print(f"Error: Folder not found: {folder_path}")
return []
def create_folder(folder_path):
"""Creates a folder if it doesn't exist."""
if not os.path.exists(folder_path):
try:
os.makedirs(folder_path) # Create the folder and any missing parent directories
except OSError as e:
print(f"Error creating folder {folder_path}: {e}")
def move_file(source_path, destination_path):
"""Moves a file from the source to the destination."""
try:
os.rename(source_path, destination_path)
except OSError as e:
print(f"Error moving file {source_path} to {destination_path}: {e}")
def process_file(folder_path, name):
"""Processes a single file, performing OCR, categorization, and moving."""
text = abc_1.use_ocr(os.path.join(folder_path, name))
category = abc_1.categorize_text_chunk(text, compiled_keywords)
category_folder = os.path.join(folder_output, category)
create_folder(category_folder)
has_pdf, has_docx= check_file_existence(folder_path,name)
if has_pdf:
source_file = os.path.join(folder_path, name + '.pdf')
destination_file = os.path.join(category_folder, name + '.pdf')
move_file(source_file, destination_file)
print(f"File '{name}' categorized as '{category}' and moved to '{category_folder}'.")
if has_docx:
source_file = os.path.join(folder_path, name + '.docx')
destination_file = os.path.join(category_folder, name + '.docx')
move_file(source_file, destination_file)
print(f"File '{name}' categorized as '{category}' and moved to '{category_folder}'.")
def check_file_existence(folder_path, filename):
has_pdf = False
has_docx = False
for filename_in_folder in os.listdir(folder_path):
base_filename, ext = os.path.splitext(filename_in_folder)
if base_filename == filename:
if ext == '.pdf':
has_pdf = True
elif ext == '.docx':
has_docx = True
return has_pdf, has_docx
def runOCR(subfolder_names):
for name in subfolder_names:
process_file(folder_path, name)
if os.path.exists(folder_path+'/'+name): # buffer folder delete
shutil.rmtree(folder_path+'/'+name)
if __name__ == '__main__':
categories_keywords_dict = {
'AI': ['Artificial', 'Intelligence'],
'Automata': ['finite', 'state', 'machines'],
'DT': ['game', 'theory']
}
folder_path = 'input' #output folder ni pdftoimage toh
folder_output = 'output' # Fixed typo
compiled_keywords = abc_1.compile_keywords(categories_keywords_dict)
subfolder_names = get_subfolder_names(folder_path)
runOCR(subfolder_names)
|