added sops generation for personal assessment

This commit is contained in:
2024-09-05 02:59:01 +00:00
parent 6f781cb90a
commit 67d30fbc6a
6 changed files with 368 additions and 71 deletions
+26 -10
View File
@@ -1,21 +1,37 @@
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/test_sop.pdf"
file_path = "/root/ds_erp_ai/data/raw/document.doc"
docs = load_document(file_path)
sop = SopGenerator()
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 = sop.get_roles(docs)["roles"]
print(f"Roles {roles}")
sop_status = sop.check_role_sop(roles=roles, docs=docs)
print(sop_status)
roles = ["cloud engineer"]
sops = sop.generate_sops(roles=roles,docs=docs)
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}")