AIRegulation-DocAnalysis/backend/app/services/mock_data.py

"""Provide service-layer logic for mock data."""

from datetime import datetime
from typing import Dict, List, Any
import uuid
# Keep service responsibilities explicit so downstream behavior stays predictable.


# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_DOCUMENTS: List[Dict[str, Any]] = [
    {
        "id": "doc-001",
        "name": "道路交通安全法.pdf",
        "chunks": 156,
        "status": "indexed",
        "created_at": datetime(2026, 5, 10, 10, 0, 0),
    },
    {
        "id": "doc-002",
        "name": "机动车登记规定.docx",
        "chunks": 89,
        "status": "indexed",
        "created_at": datetime(2026, 5, 10, 11, 0, 0),
    },
    {
        "id": "doc-003",
        "name": "电动自行车规范.pdf",
        "chunks": 42,
        "status": "indexed",
        "created_at": datetime(2026, 5, 10, 12, 0, 0),
    },
    {
        "id": "doc-004",
        "name": "GB 38031-2020 电动汽车安全要求.pdf",
        "chunks": 128,
        "status": "indexed",
        "created_at": datetime(2026, 5, 10, 13, 0, 0),
    },
    {
        "id": "doc-005",
        "name": "C-NCAP管理规则(2021版).pdf",
        "chunks": 95,
        "status": "indexed",
        "created_at": datetime(2026, 5, 10, 14, 0, 0),
    },
]

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_QUICK_QUESTIONS: List[Dict[str, str]] = [
    {"id": "q1", "question": "电动自行车需要上牌照吗？", "category": "车辆登记"},
    {"id": "q2", "question": "新能源汽车有哪些补贴政策？", "category": "新能源"},
    {"id": "q3", "question": "车辆年检的规定是什么？", "category": "年检"},
    {"id": "q4", "question": "驾驶证过期了怎么处理？", "category": "驾驶证"},
]

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_RETRIEVAL_RESULTS: List[Dict[str, Any]] = [
    {
        "id": "chunk-001",
        "score": 0.95,
        "preview": "根据《道路交通安全法》第十八条规定，电动自行车经公安机关交通管理部门登记后，方可上道路行驶...",
        "doc_name": "道路交通安全法",
        "clause": "第十八条",
        "content": "根据《道路交通安全法》第十八条规定，电动自行车经公安机关交通管理部门登记后，方可上道路行驶。电动自行车应当符合国家标准，最高设计车速不超过二十五公里每小时，整车质量不超过五十五千克。",
    },
    {
        "id": "chunk-002",
        "score": 0.88,
        "preview": "电动自行车需符合GB17761-2018国家标准，包括最高车速、整车质量、脚踏骑行能力等要求...",
        "doc_name": "电动自行车规范",
        "clause": "第4条",
        "content": "电动自行车需符合GB17761-2018国家标准。主要技术要求包括：最高设计车速不超过25km/h，整车质量不超过55kg，具有脚踏骑行能力，蓄电池标称电压不超过48V，电动机额定连续输出功率不超过400W。",
    },
    {
        "id": "chunk-003",
        "score": 0.82,
        "preview": "机动车登记规定：初次申领机动车号牌、行驶证的，机动车所有人应当向住所地的车辆管理所申请注册登记...",
        "doc_name": "机动车登记规定",
        "clause": "第5条",
        "content": "机动车登记规定：初次申领机动车号牌、行驶证的，机动车所有人应当向住所地的车辆管理所申请注册登记。申请注册登记的，应当提交机动车所有人的身份证明、购车发票等机动车来历证明、机动车整车出厂合格证明或者进口机动车进口凭证。",
    },
    {
        "id": "chunk-004",
        "score": 0.75,
        "preview": "驾驶电动自行车上道路行驶，应当佩戴安全头盔，遵守道路交通安全法律法规...",
        "doc_name": "道路交通安全法",
        "clause": "第76条",
        "content": "驾驶电动自行车上道路行驶，应当佩戴安全头盔，遵守道路交通安全法律法规。电动自行车不得逆向行驶，不得在机动车道内行驶，最高车速不得超过规定的限速。",
    },
    {
        "id": "chunk-005",
        "score": 0.68,
        "preview": "电动汽车动力电池安全要求：电池系统发生热失控后，应在5分钟内不起火不爆炸...",
        "doc_name": "GB 38031-2020",
        "clause": "第7条",
        "content": "电动汽车动力电池安全要求（GB 38031-2020）：电池系统发生热失控后，应在5分钟内不起火不爆炸，为乘员预留逃生时间。电池包需通过针刺、过充、短路等安全测试。",
    },
]

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_RAG_ANSWERS: Dict[str, Dict[str, Any]] = {
    "电动自行车": {
        "text": "根据《道路交通安全法》及相关规范，电动自行车上路需满足以下条件：\n\n1. 符合国家标准 GB17761-2018\n2. 经公安机关交通管理部门登记\n3. 最高设计车速不超过 25km/h\n4. 整车质量不超过 55kg\n5. 具有脚踏骑行能力\n6. 蓄电池标称电压不超过 48V\n\n行驶时还需佩戴安全头盔，不得逆向行驶或在机动车道内行驶。",
        "retrieval_ids": ["chunk-001", "chunk-002", "chunk-004"],
    },
    "驾驶证": {
        "text": "驾驶证申请流程如下：\n\n1. 到驾校报名并参加培训\n2. 通过科目一（理论考试）\n3. 通过科目二（场地驾驶技能考试）\n4. 通过科目三（道路驾驶技能考试）\n5. 通过科目四（安全文明驾驶常识考试）\n6. 领取驾驶证\n\n初次申领需到住所地车辆管理所申请注册登记。",
        "retrieval_ids": ["chunk-003"],
    },
    "超速": {
        "text": "超速处罚标准（根据《道路交通安全法》）：\n\n- 超速10%以下：警告\n- 超速10%-20%：罚款50-200元\n- 超速20%-50%：罚款200-500元，记3-6分\n- 超速50%以上：罚款500-2000元，记12分，可吊销驾驶证\n\n机动车驾驶人违反道路交通安全法律、法规将处警告或二十元以上二百元以下罚款。",
        "retrieval_ids": ["chunk-001"],
    },
    "年检": {
        "text": "车辆年检规定：\n\n- 小型私家车：6年内免检（每2年申领标志），6-10年每2年检验，10年以上每年检验\n- 车辆需携带行驶证、交强险保单\n- 检验项目：灯光、制动、排放等\n\n机动车所有人的住所迁出车辆管理所管辖区域的，需在登记证书上签注变更事项。",
        "retrieval_ids": ["chunk-003"],
    },
    "电池": {
        "text": "电动汽车电池安全标准（GB 38031-2020）：\n\n1. 热失控要求：电池系统发生热失控后，应在5分钟内不起火不爆炸，为乘员预留逃生时间\n2. 电池包需通过针刺、过充、短路等安全测试\n3. 充电系统应具备过充保护功能，当电池SOC达到100%时应自动停止充电\n4. 充电接口应符合GB/T 18487.1标准要求\n\n以上要求确保电动汽车的整车安全性。",
        "retrieval_ids": ["chunk-005"],
    },
    "碰撞": {
        "text": "正面碰撞测试要求（C-NCAP管理规则）：\n\n1. 正面100%重叠刚性壁障碰撞试验\n2. 碰撞速度：50km/h\n3. 试验后要求：\n   - 车门应能打开\n   - 燃油系统无泄漏\n   - 座椅及安全带功能正常\n\n此测试用于评估车辆在正面碰撞事故中对乘员的保护能力。",
        "retrieval_ids": [],
    },
    "AEB": {
        "text": "AEB（自动紧急制动系统）测试标准：\n\n1. 系统应在检测到前方障碍物时主动减速或停车\n2. 测试场景分为三种：\n   - 目标车静止\n   - 目标车移动\n   - 目标车制动\n3. AEB功能是C-NCAP评分的重要加分项\n\n该系统对提升车辆主动安全性能具有重要意义。",
        "retrieval_ids": [],
    },
    "高速公路": {
        "text": "高速公路安全距离规定：\n\n1. 车速超过100km/h时，与同车道前车保持100米以上距离\n2. 车速低于100km/h时，距离可适当缩短\n3. 执行紧急任务的警车、消防车、救护车、工程救险车不受行驶速度限制\n\n保持安全距离是预防追尾事故的关键措施。",
        "retrieval_ids": [],
    },
}

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_COMPLIANCE_RESULT: Dict[str, Any] = {
    "task_id": "task-001",
    "dashboard": {
        "score": 78,
        "high_risk_count": 2,
        "medium_risk_count": 1,
        "low_risk_count": 0,
        "need_fix_segments": 3,
        "status": "warning",
        "status_label": "需优化",
    },
    "segments": [
        {
            "id": 1,
            "index": 1,
            "intent": "车身结构设计",
            "start_pos": 45,
            "end_pos": 230,
            "content": "车身采用高强度钢铝混合结构，A柱和B柱使用热成型钢板，厚度2.5mm。车顶结构设计满足GB 26112-2010抗压强度要求，正面碰撞能量吸收区域采用渐进式变形设计，确保碰撞时能量有效分散。",
            "risk_level": "high",
            "regulations": [
                {
                    "id": 1,
                    "name": "GB 26112-2010",
                    "clause": "第4.2条",
                    "score": 0.95,
                    "match_keyword": "车顶抗压强度",
                    "category": "high",
                    "full_content": "车顶结构应能承受相当于车辆整备质量1.5倍的载荷，载荷分布应均匀，试验后车顶变形量不超过规定值。",
                },
                {
                    "id": 2,
                    "name": "C-NCAP管理规则",
                    "clause": "第3.1条",
                    "score": 0.88,
                    "match_keyword": "正面碰撞",
                    "category": "high",
                    "full_content": "正面碰撞试验速度为50km/h，碰撞后车门应能打开，燃油系统无泄漏，座椅及安全带功能正常。",
                },
                {
                    "id": 3,
                    "name": "GB 11551-2014",
                    "clause": "第5条",
                    "score": 0.72,
                    "match_keyword": "碰撞能量吸收",
                    "category": "medium",
                    "full_content": "车辆正面碰撞时应有效保护乘员，碰撞能量应通过车身结构合理分散。",
                },
                {
                    "id": 4,
                    "name": "机动车安全技术条件",
                    "clause": "第12条",
                    "score": 0.58,
                    "match_keyword": "A柱强度",
                    "category": "medium",
                    "full_content": "A柱应具备足够的抗变形能力，材料强度应符合相关标准要求。",
                },
            ],
        },
        {
            "id": 2,
            "index": 2,
            "intent": "动力系统配置",
            "start_pos": 298,
            "end_pos": 425,
            "content": "搭载永磁同步电机，最大功率150kW，峰值扭矩310Nm。电池组采用三元锂离子电池，容量75kWh，能量密度180Wh/kg。充电接口支持快充（30分钟充至80%）和慢充（8小时充满），符合GB/T 18487.1-2015标准。",
            "risk_level": "medium",
            "regulations": [
                {
                    "id": 5,
                    "name": "GB/T 18487.1-2015",
                    "clause": "第6条",
                    "score": 0.94,
                    "match_keyword": "充电接口标准",
                    "category": "high",
                    "full_content": "电动汽车传导充电接口应符合GB/T 18487.1标准要求，充电系统应具备过充保护功能。",
                },
                {
                    "id": 6,
                    "name": "GB/T 31484-2015",
                    "clause": "第4条",
                    "score": 0.85,
                    "match_keyword": "电池能量密度",
                    "category": "high",
                    "full_content": "动力电池能量密度不低于120Wh/kg，电池系统需通过热失控测试。",
                },
                {
                    "id": 7,
                    "name": "新能源汽车生产企业准入",
                    "clause": "第8条",
                    "score": 0.65,
                    "match_keyword": "电机功率",
                    "category": "medium",
                    "full_content": "驱动电机应符合相关技术标准，功率参数应在规定范围内。",
                },
                {
                    "id": 8,
                    "name": "电动汽车安全要求",
                    "clause": "第7条",
                    "score": 0.45,
                    "match_keyword": "充电时间",
                    "category": "low",
                    "full_content": "充电系统应具备过充保护功能，当电池SOC达到100%时应自动停止充电。",
                },
            ],
        },
        {
            "id": 3,
            "index": 3,
            "intent": "安全配置设计",
            "start_pos": 570,
            "end_pos": 725,
            "content": "配备6个安全气囊（前排双气囊、侧气囊、侧气帘），采用预紧式安全带。ABS系统采用博世第9代ESP，具备碰撞预警功能（FCW）和自动紧急制动（AEB）。方向盘集成驾驶员疲劳监测摄像头。",
            "risk_level": "low",
            "regulations": [
                {
                    "id": 9,
                    "name": "GB 27887-2011",
                    "clause": "第5条",
                    "score": 0.92,
                    "match_keyword": "安全气囊",
                    "category": "high",
                    "full_content": "乘用车应配备驾驶员和乘客安全气囊，气囊系统应符合相关技术标准。",
                },
                {
                    "id": 10,
                    "name": "GB/T 26991-2011",
                    "clause": "第3条",
                    "score": 0.78,
                    "match_keyword": "ABS系统",
                    "category": "medium",
                    "full_content": "车辆应配备防抱死制动系统，系统性能应符合相关标准要求。",
                },
                {
                    "id": 11,
                    "name": "C-NCAP管理规则",
                    "clause": "第4.2条",
                    "score": 0.71,
                    "match_keyword": "AEB自动制动",
                    "category": "medium",
                    "full_content": "主动安全配置评分包含AEB功能，AEB系统应能有效检测障碍物并主动减速。",
                },
                {
                    "id": 12,
                    "name": "机动车运行安全技术条件",
                    "clause": "第15条",
                    "score": 0.38,
                    "match_keyword": "疲劳监测",
                    "category": "low",
                    "full_content": "建议配备驾驶员状态监测系统，及时发现驾驶员疲劳或分心状态。",
                },
            ],
        },
    ],
    "priority_actions": [
        {
            "regulation": "GB 26112-2010 第4.2条",
            "issue": "缺少车顶抗压强度测试数据",
            "suggestion": "补充车顶抗压强度具体测试数据，确保满足1.5倍整备质量载荷要求",
            "severity": "high",
        },
        {
            "regulation": "GB/T 31484-2015 第4条",
            "issue": "缺少电池热失控测试报告",
            "suggestion": "补充电池热失控测试报告，验证5分钟内不起火不爆炸",
            "severity": "high",
        },
        {
            "regulation": "C-NCAP管理规则 第3.1条",
            "issue": "缺少碰撞后车门开启性能数据",
            "suggestion": "提供碰撞后车门开启性能测试数据",
            "severity": "medium",
        },
    ],
}

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_COMPLIANCE_CHAT_RESPONSES: Dict[str, Dict[str, str]] = {
    "车身结构设计": {
        "compliance": "根据当前分析，车身结构设计部分存在以下合规问题：\n\n1. GB 26112-2010要求车顶承受1.5倍整备质量载荷，目前设计声明满足要求但缺少测试数据\n2. C-NCAP正面碰撞后车门应能打开，需提供碰撞测试报告\n\n建议补充相关测试数据以提升合规评分。",
        "interpretation": "GB 26112-2010 第4.2条具体要求解读：\n\n车顶抗压强度测试是车辆被动安全的重要指标。该标准要求车顶结构能够承受相当于车辆整备质量1.5倍的均匀分布载荷，试验后车顶变形量不得超过规定限值。\n\n热成型钢板（22MnB5材料）抗拉强度约1500-1650 MPa，理论上能满足要求，但需通过实际测试验证。",
        "suggestion": "针对车身结构设计的修改建议：\n\n1. 补充车顶抗压强度测试报告\n2. 提供A柱材料认证证书\n3. 完善正面碰撞能量吸收设计说明\n4. 添加碰撞后车门开启性能数据\n\n这些补充材料可有效提升合规评分。",
    },
    "动力系统配置": {
        "compliance": "动力系统配置整体合规性良好，主要检查点：\n\n1. 电池能量密度180Wh/kg超过最低要求120Wh/kg ✓\n2. 充电接口符合GB/T 18487.1标准 ✓\n3. 快充30分钟充至80%符合行业标准 ✓\n\n需补充电池热失控测试报告。",
        "interpretation": "GB/T 31484-2015对动力电池的要求解读：\n\n1. 能量密度：不低于120Wh/kg（您的设计180Wh/kg满足要求）\n2. 循环寿命：不少于1000次循环后容量保持率≥80%\n3. 安全测试：需通过针刺、过充、短路等测试\n\n建议补充循环寿命测试数据。",
        "suggestion": "动力系统配置改进建议：\n\n1. 补充电池热失控测试报告\n2. 提供循环寿命测试数据\n3. 添加充电系统过充保护功能说明\n4. 完善电池管理系统（BMS）技术文档",
    },
    "安全配置设计": {
        "compliance": "安全配置设计合规性评估：\n\n1. 安全气囊配置满足GB 27887-2011要求 ✓\n2. ABS/ESP系统符合标准 ✓\n3. AEB功能是C-NCAP加分项 ✓\n\n驾驶员疲劳监测是建议配置，不强制要求。",
        "interpretation": "C-NCAP主动安全评分规则解读：\n\nAEB（自动紧急制动）系统是C-NCAP评分的重要加分项，最高可获得额外加分。测试场景包括：\n- 目标车静止场景\n- 目标车移动场景\n- 目标车制动场景\n\n建议完善AEB系统测试数据以获取更高评分。",
        "suggestion": "安全配置优化建议：\n\n1. 提供AEB系统测试数据\n2. 补充FCW预警功能测试报告\n3. 添加安全气囊展开时间数据\n4. 完善驾驶员疲劳监测系统说明（如有）",
    },
}

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_SYSTEM_STATS: Dict[str, int] = {
    "docs": 5,
    "chunks": 510,
    "vectors": 510,
    "segments": 0,
}

# Keep service responsibilities explicit so downstream behavior stays predictable.
MOCK_SYSTEM_CONFIG: Dict[str, Any] = {
    "llm": {
        "model": "qwen-max",
    },
    "embedding": {
        "model": "text-embedding-v3",
        "dimension": 1536,
    },
    "milvus": {
        "host": "localhost",
        "port": 19530,
    },
    "retrieval": {
        "vector_top_k": 10,
        "final_top_k": 5,
    },
}


def get_mock_documents() -> List[Dict[str, Any]]:
    """Return mock documents."""
    return MOCK_DOCUMENTS


def get_mock_quick_questions() -> List[Dict[str, str]]:
    """Return mock quick questions."""
    return MOCK_QUICK_QUESTIONS


def get_mock_retrieval(query: str, top_k: int = 5) -> List[Dict[str, Any]]:
    """Return mock retrieval."""
    results = []
    for keyword, data in MOCK_RAG_ANSWERS.items():
        if keyword in query:
            for retrieval_id in data.get("retrieval_ids", []):
                for item in MOCK_RETRIEVAL_RESULTS:
                    if item["id"] == retrieval_id:
                        results.append({
                            "id": item["id"],
                            "score": item["score"],
                            "preview": item["preview"],
                            "doc_name": item["doc_name"],
                            "clause": item["clause"],
                        })
                        break
    if not results:
        results = MOCK_RETRIEVAL_RESULTS[:top_k]
    return results[:top_k]


def get_mock_rag_answer(query: str) -> str:
    """Return mock rag answer."""
    for keyword, data in MOCK_RAG_ANSWERS.items():
        if keyword in query:
            return data["text"]
    return "抱歉，暂未找到与您问题直接相关的法规内容。请尝试更具体的问题，或联系交通管理部门获取详细信息。\n\n您可以尝试询问：电动自行车、驾驶证、超速处罚、年检、电池安全、碰撞测试、AEB系统、高速公路规则等话题。"


def get_mock_compliance_result(task_id: str) -> Dict[str, Any]:
    """Return mock compliance result."""
    result = MOCK_COMPLIANCE_RESULT.copy()
    result["task_id"] = task_id
    return result


def get_mock_compliance_chat_response(intent: str, query: str) -> str:
    """Return mock compliance chat response."""
    responses = MOCK_COMPLIANCE_CHAT_RESPONSES.get(intent, {})
    if "合规" in query or "符合" in query:
        return responses.get("compliance", "根据相关法规分析，该段落的合规性需进一步评估。")
    elif "解读" in query or "什么" in query or "如何" in query:
        return responses.get("interpretation", "法规要求详细解读如下...")
    elif "修改" in query or "建议" in query or "完善" in query:
        return responses.get("suggestion", "建议进行以下修改以提升合规性...")
    return f"关于您的问题，{intent}部分涉及多条相关法规。您可以进一步询问合规性评估或修改建议。"


def generate_task_id() -> str:
    """Handle generate task id."""
    return f"task-{uuid.uuid4().hex[:8]}"


def generate_doc_id() -> str:
    """Handle generate doc id."""
    return f"doc-{uuid.uuid4().hex[:8]}"