prompts/prompts_manager.py

import datetime
from pathlib import Path

from utils.date_utils import get_today_date_as_dd_mm_yyyy
from form.form import work_categories as wc


class PromptsManager:
    def __init__(self, work_categories: dict[str, str] = None):

        if not work_categories:
            self.work_categories = wc
        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

    def verify_user_input_prompt(self, user_prompt) -> str:
        return (f"Using only this information \n {user_prompt} \n answer the following questions, if the answer is not present or you are not sure about the answer just answer null. "
                f"Put each answer in a new line, keep the answer brief "
                f"and maintain the order in which the questions are asked. Do not add any preamble: "
                f"{self.verification_prompt}")

    def get_work_category(self, work_description: str) -> str:
        return (f"The work to do is {work_description}. Given the following categories {', '.join(self.work_categories.values())} "
                f"which ones are the most relevant? Only return the categories, separated by a semicolon")

    def ingest_user_answers(self, user_prompt: str) -> str:
        return f"Ingest the following information: {user_prompt}"