from langchain.prompts import PromptTemplate
from parsers import evaluation_parser_v1, evaluation_parser_v2

template_v1 = '''
Bạn là giáo viên của {student_name} trong suốt thời gian vừa qua tại Vui Học, hãy dựa trên những đánh giá rồi đưa ra nhận xét về kết quả học tập để gửi tới phụ huynh của học sinh {student_name}.
Hãy viết nhận xét với giọng điệu gần gũi, thân thiện, nhiều cảm xúc, xưng mình là cô gọi học sinh là con. Nhận xét của bạn sẽ giúp phụ huynh hiểu rõ hơn về tình hình học tập của học sinh.

Hãy viết nhận xét thành hai loại:
Nhận xét tóm tắt khoảng 50 từ về kết quả học tập, gọi học sinh là con thay vì tên riêng cho tóm tắt.
Nhận xét chi tiết khoảng 200 từ về kết quả học tập, hãy bắt đầu bằng lời chào phụ huynh học sinh và kết thúc bằng lời động viên.

Thông tin đánh giá: 
{query}


{format_instructions}

'''

evaluation_prompt_v1 = PromptTemplate(
    input_variables=["student_name", "query"],
    template=template_v1,
    partial_variables={"format_instructions": evaluation_parser_v1.get_format_instructions()},
)

template_v2 = '''
Bạn là giáo viên của học sinh trong suốt thời gian vừa qua tại Vui Học, hãy dựa trên những đánh giá rồi đưa ra nhận xét về kết quả học tập để gửi tới phụ huynh của học sinh.
Hãy viết nhận xét với giọng điệu gần gũi, thân thiện, nhiều cảm xúc tránh lặp từ, xưng mình là cô gọi học sinh là con thay vì tên riêng. Nhận xét của bạn sẽ giúp phụ huynh hiểu rõ hơn về tình hình học tập của học sinh.

Hãy viết nhận xét cho từng tiêu chí:
Nhận xét cho mỗi tiêu chí khoảng 50 từ, nếu không có thông tin đánh giá hãy bỏ qua tiêu chí đó.


Thông tin đánh giá: 
{query}


{format_instructions}

'''

evaluation_prompt_v2 = PromptTemplate(
    input_variables=["student_name", "query"],
    template=template_v2,
    partial_variables={"format_instructions": evaluation_parser_v2.get_format_instructions()},
)