update

backend/app/schemas/compliance.py

@@ -0,0 +1,69 @@
+from pydantic import BaseModel
+from typing import Optional
+from enum import Enum
+
+
+class RiskLevel(str, Enum):
+    high = "high"
+    medium = "medium"
+    low = "low"
+
+
+class ComplianceStatus(str, Enum):
+    pass_status = "pass"
+    warning = "warning"
+    fail = "fail"
+
+
+class Regulation(BaseModel):
+    id: int
+    name: str
+    clause: Optional[str] = None
+    score: float
+    match_keyword: str
+    category: RiskLevel
+    full_content: str
+
+
+class ComplianceSegment(BaseModel):
+    id: int
+    index: int
+    intent: str
+    start_pos: int
+    end_pos: int
+    content: str
+    risk_level: RiskLevel
+    regulations: list[Regulation]
+
+
+class RiskDashboard(BaseModel):
+    score: float
+    high_risk_count: int
+    medium_risk_count: int
+    low_risk_count: int
+    need_fix_segments: int
+    status: ComplianceStatus
+    status_label: str
+
+
+class PriorityAction(BaseModel):
+    regulation: str
+    issue: str
+    suggestion: str
+    severity: RiskLevel
+
+
+class ComplianceResult(BaseModel):
+    task_id: str
+    dashboard: RiskDashboard
+    segments: list[ComplianceSegment]
+    priority_actions: list[PriorityAction]
+
+
+class ComplianceChatRequest(BaseModel):
+    query: str
+
+
+class AnalyzeResponse(BaseModel):
+    task_id: str
+    status: str = "processing"