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}")