20 lines
615 B
Python
20 lines
615 B
Python
import os
|
|
from pathlib import Path
|
|
SRC_DIR = Path(__file__).parent
|
|
|
|
def load_prompt():
|
|
"""Load the prompt from data/prompt.txt"""
|
|
try:
|
|
# get directory of this file:
|
|
file_dir = Path(os.path.dirname(os.path.realpath(__file__)))
|
|
data_dir = file_dir / "data"
|
|
prompt_file = data_dir / "prompt.txt"
|
|
# Load the promt from data/prompt.txt
|
|
with open(SRC_DIR/ "data/prompt.txt", "r") as prompt_file:
|
|
prompt = prompt_file.read()
|
|
|
|
return prompt
|
|
except FileNotFoundError:
|
|
print("Error: Prompt file not found", flush=True)
|
|
return ""
|