from utils.config_handler import prompts_conf
from utils.path_tool import get_abs_path
from utils.logger_handler import logger

def load_system_prompts():
    try:
        system_prompt_path = get_abs_path(prompts_conf["main_prompt_path"])
    except KeyError as e:
        logger.error(f"load_system_prompts: 配置文件中缺少键: {e}")
        raise e

    try:
        return open(system_prompt_path, 'r', encoding='utf-8').read()
    except Exception as e:
        logger.error(f"load_system_prompts: 读取文件时出错: {system_prompt_path}, 错误信息: {str(e)}")
        raise e


def load_rag_prompts():
    try:
        rag_prompt_path = get_abs_path(prompts_conf["rag_summarize_prompt_path"])
    except KeyError as e:
        logger.error(f"load_rag_prompts: 配置文件中缺少键: {e}")
        raise e

    try:
        return open(rag_prompt_path, 'r', encoding='utf-8').read()
    except Exception as e:
        logger.error(f"load_rag_prompts: 读取文件时出错: {rag_prompt_path}, 错误信息: {str(e)}")
        raise e


def load_report_prompts():
    try:
        report_prompt_path = get_abs_path(prompts_conf["report_prompt_path"])
    except KeyError as e:
        logger.error(f"load_report_prompts: 配置文件中缺少键: {e}")
        raise e

    try:
        return open(report_prompt_path, 'r', encoding='utf-8').read()
    except Exception as e:
        logger.error(f"load_report_prompts: 读取文件时出错: {report_prompt_path}, 错误信息: {str(e)}")
        raise e

if __name__ == '__main__':
    print(load_report_prompts())
    print(load_system_prompts())
    print(load_rag_prompts())