dam_helper/prompts/prompts_manager.py

from pathlib import Path

from enums.enums import Questions
from local_storage.entities import PersonalDetails, ContractorDetails, LocationDetails
from utils.date_utils import get_today_date_as_dd_mm_yyyy


class PromptsManager:
    def __init__(self, work_categories: dict[str, str] = None):
        self.work_categories = work_categories
        base_path = Path(__file__).parent
        with open(Path(base_path, "system_prompt.txt")) as sysprompt_file:
            self.system_prompt: str = sysprompt_file.read()
        with open(Path(base_path, "questions.txt")) as questions_file:
            self.questions: list[str] = questions_file.readlines()
        with open(Path(base_path, "verification_prompt2.txt")) as verification_prompt_file:
            verification_prompt = verification_prompt_file.read()
            todays_date = get_today_date_as_dd_mm_yyyy()
            verification_prompt = verification_prompt.replace("{today}", todays_date)
            self.verification_prompt: str = verification_prompt
            self.verification_prompt_questions = self.verification_prompt.split("\n")

    def verify_user_input_prompt(self, user_prompt, exclude_questions_group: list[int] = None) -> str:
        prompt = f"""
            Using only this information \n {user_prompt} \n answer the following questions, for each question that you cannot answer just answer 'null'.
            Put each answer in a new line, keep the answer brief 
            and maintain the order in which the questions are asked. Do not add any preamble:
        """

        skip_questions = self._get_questions(exclude_questions_group)
        questions = [q for idx, q in enumerate(self.verification_prompt_questions) if idx not in skip_questions]
        return prompt + "\n".join(questions)

    def get_work_category(self, work_description: str) -> str:
        return (
            f"The work to do is {work_description}: choose the most accurate categories among the following:"
            f"{', '.join(self.work_categories.values())}\n"
            f"Only return the categories, separated by a semicolon")

    @staticmethod
    def questions_to_field_labels():
        return {
            Questions.FULL_NAME: "Full name", Questions.WORK_TO_DO: "Work to do", Questions.COMMUNITY: "Community",
            Questions.BUILDING: "Building name", Questions.UNIT_APT_NUMBER: "Unit/apartment number",
            Questions.OWNER_OR_TENANT: "Owner/Tenant", Questions.START_DATE: "Start date",
            Questions.END_DATE: "End date", Questions.CONTACT_NUMBER: "Your contact number",
            Questions.COMPANY_NAME: "Contractor company name", Questions.COMPANY_EMAIL: "Contracting company email",
            Questions.COMPANY_NUMBER: "Contracting company contact number", Questions.YOUR_EMAIL: "Your email"
        }

    def compose_email(self, answers: dict[Questions, str]):
        prompt = f"""
        Compose a formal email directed at Dubai Community Management asking for a minor work permit.
        It should include the following details:
        My name is {answers[Questions.FULL_NAME]} and I am the {answers[Questions.OWNER_OR_TENANT]} of unit
        {answers[Questions.UNIT_APT_NUMBER]} located in {answers[Questions.BUILDING]}, {answers[Questions.COMMUNITY]};
        
        The work involves {answers[Questions.WORK_TO_DO]}, and will be carried out by {answers[Questions.COMPANY_NAME]}
        from {answers[Questions.START_DATE]} to {answers[Questions.END_DATE]};
        
        The work permit request form is attached.
        
        Please return only the email body, without any preamble
        """
        return prompt

    @staticmethod
    def _get_questions(details: list[int]):
        to_skip: list[int] = []
        for d in details:
            if isinstance(d, PersonalDetails):
                to_skip.extend([Questions.FULL_NAME.value, Questions.CONTACT_NUMBER.value, Questions.YOUR_EMAIL.value])
            if isinstance(d, ContractorDetails):
                to_skip.extend([Questions.COMPANY_NAME.value, Questions.COMPANY_NUMBER.value, Questions.COMPANY_EMAIL.value])
            if isinstance(d, LocationDetails):
                to_skip.extend([Questions.OWNER_OR_TENANT.value, Questions.COMMUNITY.value, Questions.BUILDING.value, Questions.UNIT_APT_NUMBER.value])
        return to_skip