import parse_docx, pis_find

import os

def start():
    print("Local mode activated")
    working_directory = os.getcwd()
    print("Working directory: ", working_directory)

    ## Get all files in directory
    files = [f for f in os.listdir(working_directory) if os.path.isfile(os.path.join(working_directory, f))]
    docx_files = [f for f in files if f.endswith(".docx")]

    results = []

    if docx_files:
        print(f"Found {len(docx_files)} DOCX files in directory")
        for file in docx_files:
            print(file)
            items = parse_docx.extract_tables(file)
            results.extend(items)
    else:
        print("No DOCX files found")

    print(f"Found {len(results)} PIS Codes in documents")
    pis_find.run(results)


if __name__ == "__main__":
    start()