Spaces:
Sleeping
Sleeping
File size: 2,705 Bytes
0fadcb9 54af9e3 0fadcb9 54af9e3 0fadcb9 d005419 aa2cc5f 7648566 0fadcb9 d005419 0fadcb9 d005419 54af9e3 d005419 54af9e3 aa2cc5f 54af9e3 d005419 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import datetime
from enum import Enum
from pathlib import Path
from utils.date_utils import get_today_date_as_dd_mm_yyyy
class Questions(Enum):
FULL_NAME = 0
WORK_TO_DO = 1
COMMUNITY = 2
BUILDING = 3
UNIT_APT_NUMBER = 4
OWNER_OR_TENANT = 5
START_DATE = 6
END_DATE = 7
CONTACT_NUMBER = 8
COMPANY_NAME = 9
COMPANY_EMAIL = 10
COMPANY_NUMBER = 11
YOUR_EMAIL = 12
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
def verify_user_input_prompt(self, user_prompt) -> str:
return (
f"Using only this information \n {user_prompt} \n answer the following questions, for each question that you cannot 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}: 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"
}
|