Files
erp-ai-latest/test.py
T

38 lines
1.1 KiB
Python

from src.services.sop_generator import SopGenerator
from src.utils.document_loader import load_document
from src.services.sop_generator import SopPersonalAssessment
file_path = "/root/ds_erp_ai/data/raw/document.doc"
docs = load_document(file_path)
from src.services.sop_generator import SopGeneratorDocument
sop = SopPersonalAssessment()
if __name__ == "__main__":
# Assuming 'sop' is an instance of SopGenerator and 'docs' is the loaded document content.
# Step 1: Get the roles from the document
roles = [
{
"role": "Content Marketing Specialist",
"sop_types": ["will", "shall"],
"areas": ["communication", "development"]
},
{
"role": "Digital Marketing Specialist",
"sop_types": ["must"],
"areas": ["finance", "project management"]
},
{
"role": "Information Technology Officer",
"sop_types": ["shall", "must"],
"areas": ["development", "communication", "operations"]
}
]
sops = sop.generate_sops_by_role_and_area(roles)
print(f"sops:{sops}")