from django.shortcuts import get_object_or_404, redirect, render

from agent_core.orchestrator import run_agent
from agent_core.results import AgentResult
from apps.audit.services import create_audit_log
from apps.documents.models import SubmissionBatch, UploadedDocument
from apps.scenarios.services import get_scenario

from .forms import ChatForm
from .models import Conversation


def index(request):
    conversations = Conversation.objects.all()
    if conversations.exists():
        return redirect("chat:detail", conversation_id=conversations.first().conversation_id)
    return render(
        request,
        "chat/index.html",
        {
            "conversation": None,
            "conversations": [],
            "form": ChatForm(),
            "documents": [],
            "result": None,
            "audit_log": None,
            "node_results": [],
            "active_node": None,
        },
    )


def detail(request, conversation_id: str):
    conversation = get_object_or_404(Conversation, conversation_id=conversation_id)
    batch = SubmissionBatch.objects.filter(batch_id=conversation.batch_id).first()
    documents = UploadedDocument.objects.filter(batch=batch)
    form = ChatForm(request.POST or None, documents=documents)
    result = None
    audit_log = None
    active_node = None
    task_modes = [
        {"name": "目录汇总", "description": "汇总文件、页数、章节点和目录型文档。"},
        {"name": "完整性检查", "description": "对照法规模板检查齐套性、缺失项和错放项。"},
        {"name": "字段抽取", "description": "抽取产品名称、规格、适用范围、储存条件等核心字段。"},
        {"name": "一致性核查", "description": "比较申请表、说明书和产品列表的字段一致性。"},
        {"name": "综合风险报告", "description": "形成高优先级问题、建议动作和责任人通知。"},
    ]
    if request.method == "POST" and form.is_valid():
        scenario = get_scenario("document_review")
        message = form.cleaned_data["message"]
        try:
            result = run_agent(
                scenario,
                message,
                options={
                    "conversation_id": conversation.conversation_id,
                    "batch_id": conversation.batch_id,
                    "product_name": conversation.product_name,
                    "document_ids": form.cleaned_data["document_ids"],
                },
            )
        except Exception as exc:
            result = AgentResult(status="failed", error=str(exc), answer="")
        audit_log = create_audit_log(
            "document_review",
            "注册审核智能体",
            message,
            result,
            batch_id=conversation.batch_id,
            conversation_id=conversation.conversation_id,
            product_name=conversation.product_name,
        )
        active_node = "risk"

    return render(
        request,
        "chat/index.html",
        {
            "conversation": conversation,
            "conversations": Conversation.objects.all(),
            "batch": batch,
            "form": form,
            "documents": documents,
            "document_count": documents.count(),
            "result": result,
            "audit_log": audit_log,
            "task_modes": task_modes,
            "node_results": conversation.node_results,
            "active_node": active_node,
        },
    )