AI 거버넌스

# AI 거버넌스 체크리스트 자동화 도구
from dataclasses import dataclass, field
from enum import Enum
from typing import List, Dict, Optional
from datetime import datetime

class RiskLevel(Enum):
    UNACCEPTABLE = "unacceptable"
    HIGH = "high"
    LIMITED = "limited"
    MINIMAL = "minimal"

class ComplianceStatus(Enum):
    COMPLIANT = "compliant"
    PARTIAL = "partial"
    NON_COMPLIANT = "non_compliant"
    NOT_APPLICABLE = "not_applicable"

@dataclass
class GovernanceCheckItem:
    """거버넌스 체크 항목"""
    id: str
    category: str
    requirement: str
    eu_ai_act_article: str
    status: ComplianceStatus = ComplianceStatus.NON_COMPLIANT
    evidence: Optional[str] = None
    last_reviewed: Optional[datetime] = None

@dataclass
class AIGovernanceFramework:
    """AI 거버넌스 프레임워크"""
    organization: str
    ai_system_name: str
    risk_level: RiskLevel
    checklist: List[GovernanceCheckItem] = field(default_factory=list)

    def __post_init__(self):
        self._initialize_checklist()

    def _initialize_checklist(self):
        """EU AI Act 기반 체크리스트 초기화"""
        base_items = [
            GovernanceCheckItem(
                id="GOV-001",
                category="위험관리",
                requirement="위험 관리 시스템 수립 및 문서화",
                eu_ai_act_article="Article 9"
            ),
            GovernanceCheckItem(
                id="GOV-002",
                category="데이터 거버넌스",
                requirement="학습/검증/테스트 데이터 품질 관리 프로세스",
                eu_ai_act_article="Article 10"
            ),
            GovernanceCheckItem(
                id="GOV-003",
                category="기술문서",
                requirement="기술 문서 작성 및 최신 상태 유지",
                eu_ai_act_article="Article 11"
            ),
            GovernanceCheckItem(
                id="GOV-004",
                category="기록보관",
                requirement="자동 로깅 및 10년 보관 체계",
                eu_ai_act_article="Article 12"
            ),
            GovernanceCheckItem(
                id="GOV-005",
                category="투명성",
                requirement="사용자 대상 명확한 정보 제공",
                eu_ai_act_article="Article 13"
            ),
            GovernanceCheckItem(
                id="GOV-006",
                category="인적감독",
                requirement="인간 감독 체계 및 개입 메커니즘",
                eu_ai_act_article="Article 14"
            ),
            GovernanceCheckItem(
                id="GOV-007",
                category="품질관리",
                requirement="품질 관리 시스템(QMS) 운영",
                eu_ai_act_article="Article 17"
            ),
        ]
        self.checklist.extend(base_items)

    def update_status(self, item_id: str, status: ComplianceStatus,
                      evidence: str) -> bool:
        """체크 항목 상태 업데이트"""
        for item in self.checklist:
            if item.id == item_id:
                item.status = status
                item.evidence = evidence
                item.last_reviewed = datetime.now()
                return True
        return False

    def get_compliance_score(self) -> Dict[str, float]:
        """준수율 계산"""
        total = len(self.checklist)
        compliant = sum(1 for i in self.checklist
                       if i.status == ComplianceStatus.COMPLIANT)
        partial = sum(1 for i in self.checklist
                     if i.status == ComplianceStatus.PARTIAL)

        return {
            "total_items": total,
            "compliant_count": compliant,
            "partial_count": partial,
            "compliance_rate": (compliant + partial * 0.5) / total * 100,
            "status": "READY" if compliant == total else "IN_PROGRESS"
        }

    def generate_gap_analysis(self) -> List[Dict]:
        """Gap 분석 보고서 생성"""
        gaps = []
        for item in self.checklist:
            if item.status != ComplianceStatus.COMPLIANT:
                gaps.append({
                    "id": item.id,
                    "category": item.category,
                    "requirement": item.requirement,
                    "current_status": item.status.value,
                    "action_required": self._get_remediation(item)
                })
        return gaps

# 사용 예시
framework = AIGovernanceFramework(
    organization="KAITRUST Corp",
    ai_system_name="AI 신용평가 시스템",
    risk_level=RiskLevel.HIGH
)

# 준수 상태 업데이트
framework.update_status(
    "GOV-001",
    ComplianceStatus.COMPLIANT,
    "위험관리 문서 RMS-2024-001 참조"
)

print(f"준수율: {framework.get_compliance_score()}")