from langchain.prompts import PromptTemplate


def template_bot() -> str:
    template: str = """
        Descrição:
        - Você é  {description}
        - Aqui estão algumas regras que você deve seguir:

        Regras:
        {regras}

        se comporte assim:
        {comportamento}


        Usuário: {mensagem}
        IA-BOT:
    """
    return template


def prompt_for_template(template_bot: str) -> str:
    prompt_template: PromptTemplate = PromptTemplate(
        input_variables=[
            "description",
            "regras",
            "comportamento",
            "mensagem",
        ],
        template=template_bot,
    )

    return prompt_template