import json

from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from pydantic import BaseModel, Field

from app.model.KnowledgeDoc import KnowledgeDocServiceWithCreator, KnowledgeDocModel
from app.model.ReimburseModel import ReimburseModel
from app.utils.db_utils import AsyncSessionDep
from app.utils.llm_utils import create_llm


class LlmApproveResult(BaseModel):
  llm_flag: bool = Field(..., description='AI审核是否通过')
  llm_reason: str | None = Field(default=None, description="审核不通过时的原因")


async def get_llm_approve_result(session: AsyncSessionDep, reimburse_cls: ReimburseModel):
  try:
    reimburse_document: KnowledgeDocModel = await KnowledgeDocServiceWithCreator.query_item(session, {"code": "REIMBURSE"})
  except Exception as e:
    raise Exception(f"获取报销规范文档失败: {e}")

  chain = ChatPromptTemplate.from_template("""
          你是一名专业的报销单审核专家，你需要根据报销规范文档审核报销单信息，报销规范如<document/>标签中的内容所示：
          <document>
          {document_content}
          </document>
          报销单信息如下<reimburse/>标签中的内容所示：
          <reimburse>
          {reimburse_content}
          </reimburse>
          你需要返回一个json对象，类型为：{response_description}，其中llm_flag表示审核是否通过，为布尔值；llm_reason为审核不通过的原因，审核通过reason为空；
          """) | create_llm() | StrOutputParser()

  print(":::::::::::::::报销规则:::::::::::::::::::")
  print(reimburse_document.content)

  print(":::::::::::::::报销单描述::::::::::::::::::::::::")
  reimburse_description = get_reimburse_description(reimburse_cls)
  print(reimburse_description)

  llm_response_content = await chain.ainvoke({
    "document_content": reimburse_document.content,
    "reimburse_content": reimburse_description,
    "response_description": '{"llm_flag":true, "llm_reason":""}'
  })

  print("::::::::::::::LLM审核结果:::::::::::::::::::::::")
  print(llm_response_content)

  llm_response_dict = json.loads(llm_response_content)
  return llm_response_dict


def get_reimburse_description(reimrbuse_cls: ReimburseModel):
  if not reimrbuse_cls.project:
    raise Exception("报销单关联项目不存在")

  reimbruse_content = f"""
      报销申请人：{reimrbuse_cls.user.full_name}
      报销单标题：{str(reimrbuse_cls.title)}
      报销单金额：{str(reimrbuse_cls.amount)}
      报销单备注信息：{str(reimrbuse_cls.remarks)}
      报销项目：{str(reimrbuse_cls.project.name)}
    """

  for index, travel_item in enumerate(reimrbuse_cls.travel_list):
    reimbruse_content += f"""

      第{index + 1}条差旅费用信息：
      差旅费用名称：{travel_item.title}
      差旅费用类型：{travel_item.type}
      差旅出发时间：{travel_item.depart_time}
      差旅出发地点：{travel_item.depart_city}
      差旅到达时间：{travel_item.arrive_time}
      差旅到达地点：{travel_item.arrive_city}
      差旅报销金额：{travel_item.amount}
      """

  for index, other_item in enumerate(reimrbuse_cls.other_list):
    reimbruse_content += f"""

      第{index + 1}条费用信息：
      费用名称：{other_item.title}
      费用类型：{other_item.type}
      报销金额：{other_item.amount}
      票据类型：{other_item.recipe_type}
      """
  return reimbruse_content