from llama_index.core.prompts import PromptTemplate

CITATION_QA_TEMPLATE = PromptTemplate(
    "Будь ласка, надайте відповідь, базуючись виключно на наданих правових позиціях Верховного Суду. "
    "При посиланні на інформацію з джерела цитуйте відповідне джерело в кінці тексту, використовуючи його номер у квадратних дужках: [lp_id]. "
    "Кожна відповідь повинна містити щонайменше одне посилання на джерело. "
    "Цитуйте джерело лише тоді, коли ви явно на нього посилаєтеся. "
    "Якщо жодне з джерел не є корисним, зазначте це. "
    "Приклад:\n"
    "Source 1:\n"
    "Суд постановив, що договірні зобов’язання мають виконуватись, навіть якщо умови є нечіткими.\n"
    "Source 2:\n"
    "У випадку непередбачених обставин, договори можуть бути анульовані за певних умов.\n"
    "Запит: Чи може договір бути анульований через непередбачені обставини?\n"
    "Відповідь: Договір може бути анульований за певних умов у разі настання непередбачених обставин [2].\n"
    "Тепер ваша черга. Нижче наведено кілька пронумерованих правових позицій:"
    "\n------\n"
    "{context_str}"
    "\n------\n"
    "Запит: {query_str}\n"
    "Відповідь: "
)



CITATION_REFINE_TEMPLATE = PromptTemplate(
    "Будь ласка, надайте відповідь, базуючись виключно на наданих правових позиціях Верховного Суду. "
    "При посиланні на інформацію з джерела, цитуйте відповідне джерело, використовуючи його номер у квадратних дужках: [lp_id]. "
    "Кожна відповідь повинна містити щонайменше одне посилання на джерело. "
    "Цитуйте джерело лише тоді, коли ви явно на нього посилаєтеся. "
    "Якщо жодне з джерел не є корисним, зазначте це. "
    "Приклад:\n"
    "Source 1:\n"
    "Суд постановив, що договірні зобов’язання мають виконуватись, навіть якщо умови є нечіткими.\n"
    "Source 2:\n"
    "У випадку непередбачених обставин, договори можуть бути анульовані за певних умов.\n"
    "Запит: Чи може договір бути анульований через непередбачені обставини?\n"
    "Існуюча відповідь: Договір може бути анульований за певних умов у разі настання непередбачених обставин [2].\n"
    "Тепер ваша черга. "
    "Ми надали існуючу відповідь: {existing_answer}"
    "Нижче наведено кілька пронумерованих правових позицій. "
    "Використовуйте їх для уточнення існуючої відповіді. "
    "Якщо надані правові джерела не є корисними, повторіть існуючу відповідь."
    "\nПочинайте уточнення!"
    "\n------\n"
    "{context_msg}"
    "\n------\n"
    "Запит: {query_str}\n"
    "Відповідь: "
)



# CITATION_QA_TEMPLATE = PromptTemplate(
#     "You are a qualified lawyer. "
#     "Please provide a legal analysis based solely on the provided legal precedents and positions. "
#     "When referencing a legal precedent or position, "
#     "cite the appropriate source(s) using their corresponding numbers. "
#     "Every answer should include at least one legal source citation relevant to the query. "
#     "Only cite a source when you are explicitly referencing it. "
#     "If none of the legal precedents or positions are helpful, you should indicate that. "
#     "Use only the Ukrainian language to answer. "
#     "For example:\n"
#     "Source 1:\n"
#     "The court ruled that contractual obligations must be fulfilled even if the terms are vague.\n"
#     "Source 2:\n"
#     "In case of unforeseen circumstances, contracts may be voided under certain conditions.\n"
#     "Query: Can a contract be voided due to unforeseen circumstances?\n"
#     "Answer: A contract may be voided under certain conditions if unforeseen circumstances arise [2]. "
#     "Now it's your turn. Below are several numbered legal sources and precedents (legal positions):"
#     "\n------\n"
#     "{context_str}"
#     "\n------\n"
#     "New legal issue: {query_str}\n"
#     "User question: {question}\n"
#     "Answer: "
# )


# CITATION_REFINE_TEMPLATE = PromptTemplate(
#     "You are a qualified lawyer. "
#     "Please refine the legal analysis based solely on the provided legal precedents and positions. "
#     "When referencing a legal precedent or position, "
#     "cite the appropriate source(s) using their corresponding numbers. "
#     "Every refined answer should include at least one relevant legal source citation. "
#     "Only cite a source when you are explicitly referencing it. "
#     "If none of the legal precedents or positions are helpful, you should repeat the existing answer. "
#     "Use only the Ukrainian language to answer."
#     "For example:\n"
#     "Source 1:\n"
#     "The court ruled that contractual obligations must be fulfilled even if the terms are vague.\n"
#     "Source 2:\n"
#     "In case of unforeseen circumstances, contracts may be voided under certain conditions.\n"
#     "Query: Can a contract be voided due to unforeseen circumstances?\n"
#     "Existing answer: A contract may be voided under certain conditions if unforeseen circumstances arise [2].\n"
#     "Now it's your turn. "
#     "We have provided an existing legal analysis: {existing_answer}"
#     "Below are several numbered legal sources and precedents. "
#     "Use them to refine the existing legal answer. "
#     "If the provided legal sources are not helpful, you will repeat the existing answer."
#     "\nBegin refining!"
#     "\n------\n"
#     "{context_msg}"
#     "\n------\n"
#     "New legal issue: {query_str}\n"
#     "User question: {question}\n"
#     "Answer: "
# )