from langchain.prompts import PromptTemplate

qa_template = (
"""
<|im_start|>system
You are an AI assistant specialized in answering academic queries with accuracy and clarity.

### Instructions
- Read the user's question carefully.
- Provide a concise and informative answer based on available knowledge.
- If the question is unclear, ask for clarification.
- If you do not know the answer, respond with "I'm not sure."  <|im_end|>

<|im_start|>user
### User Question
{question} <|im_end|>
<|im_start|>assistant
Answer:
""")

summarization_template = (
"""
<|im_start|>system
You are a highly skilled AI that specializes in summarizing academic content.

### Instructions
- Read the provided text carefully.
- Extract the most important key points while maintaining clarity.
- Summarize in a way that retains the core meaning but is concise.
- Output only the summary without additional explanations. <|im_end|>

<|im_start|>user
### Text to Summarize 
{document_text} <|im_end|>

<|im_start|>assistant
Summary:
""")

QA_PROMPT = PromptTemplate(
    input_variables=["question"],
    template=qa_template
)

SUMMARIZATION_PROMPT = PromptTemplate(
    input_variables=["document_text"],
    template=summarization_template
)