Files
erp-ai-latest/test.py
T
2024-09-10 01:00:08 +01:00

43 lines
1.3 KiB
Python

from src.services.sop_generator import SopGeneratorExecutive
from src.utils.document_loader import load_document
from src.services.document_parser import DocumentParser
from src.services.sop_generator import SopGeneratorExecutive
file_path = r"C:\Users\User\Desktop\Blessing_AI\MKD\test_erp_ai\erp_ai\test\erp_ai\data\raw\document.doc"
docs = load_document(file_path)
if __name__ == "__main__":
SOP = DocumentParser()
so = SopGeneratorExecutive()
referencs_roles = ["AR Director "]
workers_list = [
{
"position": "AR dIRECTOR ",
"role": "Developer",
"department": "IT"
},
{
"name": "Jane Smith",
"position": "Project Manager",
"role": "Manager",
"department": "IT"
}
]
departments_and_roles = SOP.extract_sops_for_workers_by_department(docs,workers_list)
# Prepare extracted roles (only managers)
'''extracted_managers = []
for department in departments_and_roles['departments']:
extracted_managers.extend([
{
'name': manager['name'],
'position': manager.get('position', 'Unknown Position'),
'role': manager.get('role', 'Unknown Role') # PRP or SRP classification
}
for manager in department['managers']
])'''
print(departments_and_roles)