96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
|
from fastapi import APIRouter, UploadFile, File, HTTPException
|
|||
|
|
from sse_starlette.sse import EventSourceResponse
|
|||
|
|
import uuid
|
|||
|
|
import os
|
|||
|
|
import json
|
|||
|
|
import asyncio
|
|||
|
|
from app.schemas.compliance import (
|
|||
|
|
AnalyzeResponse,
|
|||
|
|
ComplianceChatRequest,
|
|||
|
|
)
|
|||
|
|
from app.services.mock_data import (
|
|||
|
|
generate_task_id,
|
|||
|
|
get_mock_compliance_result,
|
|||
|
|
get_mock_compliance_chat_response,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
router = APIRouter(prefix="/compliance", tags=["合规分析"])
|
|||
|
|
|
|||
|
|
# 临时存储分析任务
|
|||
|
|
tasks_store: dict[str, dict] = {}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/analyze", response_model=AnalyzeResponse)
|
|||
|
|
async def analyze_document(file: UploadFile = File(...)):
|
|||
|
|
"""上传设计方案进行分析"""
|
|||
|
|
# 生成任务ID
|
|||
|
|
task_id = generate_task_id()
|
|||
|
|
|
|||
|
|
# 保存文件
|
|||
|
|
raw_dir = "/airegulation/demo-mao/backend/data/raw"
|
|||
|
|
os.makedirs(raw_dir, exist_ok=True)
|
|||
|
|
file_path = os.path.join(raw_dir, f"compliance_{task_id}_{file.filename}")
|
|||
|
|
|
|||
|
|
content = await file.read()
|
|||
|
|
with open(file_path, "wb") as f:
|
|||
|
|
f.write(content)
|
|||
|
|
|
|||
|
|
# 记录任务
|
|||
|
|
tasks_store[task_id] = {
|
|||
|
|
"task_id": task_id,
|
|||
|
|
"file_path": file_path,
|
|||
|
|
"status": "processing",
|
|||
|
|
"result": None,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 模拟异步处理完成（立即返回结果）
|
|||
|
|
# 实际应用中这应该是后台任务
|
|||
|
|
tasks_store[task_id]["status"] = "completed"
|
|||
|
|
tasks_store[task_id]["result"] = get_mock_compliance_result(task_id)
|
|||
|
|
|
|||
|
|
return AnalyzeResponse(task_id=task_id)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/result/{task_id}")
|
|||
|
|
async def get_result(task_id: str):
|
|||
|
|
"""获取分析结果"""
|
|||
|
|
if task_id not in tasks_store:
|
|||
|
|
# 如果任务ID不存在，返回默认mock结果
|
|||
|
|
return get_mock_compliance_result(task_id)
|
|||
|
|
|
|||
|
|
task = tasks_store[task_id]
|
|||
|
|
|
|||
|
|
if task["status"] == "processing":
|
|||
|
|
return {"status": "processing", "message": "分析进行中"}
|
|||
|
|
|
|||
|
|
return task["result"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/chat/{segment_id}")
|
|||
|
|
async def compliance_chat(segment_id: int, request: ComplianceChatRequest):
|
|||
|
|
"""针对段落进行合规对话"""
|
|||
|
|
# 根据segment_id获取对应的intent
|
|||
|
|
intent_map = {
|
|||
|
|
1: "车身结构设计",
|
|||
|
|
2: "动力系统配置",
|
|||
|
|
3: "安全配置设计",
|
|||
|
|
}
|
|||
|
|
intent = intent_map.get(segment_id, "车身结构设计")
|
|||
|
|
|
|||
|
|
async def generate():
|
|||
|
|
# 获取预设响应
|
|||
|
|
response = get_mock_compliance_chat_response(intent, request.query)
|
|||
|
|
|
|||
|
|
# 流式输出响应
|
|||
|
|
sentences = response.split("\n\n")
|
|||
|
|
for sentence in sentences:
|
|||
|
|
if sentence.strip():
|
|||
|
|
chunks = sentence.split("\n")
|
|||
|
|
for chunk in chunks:
|
|||
|
|
if chunk.strip():
|
|||
|
|
await asyncio.sleep(0.05)
|
|||
|
|
yield {"event": "message", "data": json.dumps({"type": "chunk", "text": chunk + "\n"})}
|
|||
|
|
|
|||
|
|
yield {"event": "message", "data": json.dumps({"type": "done"})}
|
|||
|
|
|
|||
|
|
return EventSourceResponse(generate())
|