EU AI Act

# EU AI Act 위험 분류 및 컴플라이언스 체커
from dataclasses import dataclass, field
from typing import List, Dict, Optional
from enum import Enum
from datetime import datetime

class RiskLevel(Enum):
    """EU AI Act 위험 등급"""
    PROHIBITED = "prohibited"        # 금지 - Article 5
    HIGH_RISK = "high_risk"          # 고위험 - Article 6, Annex III
    LIMITED_RISK = "limited_risk"    # 제한적 위험 - Article 50
    MINIMAL_RISK = "minimal_risk"    # 최소 위험

class ActorRole(Enum):
    """EU AI Act 역할 구분"""
    PROVIDER = "provider"            # 제공자 - Article 3(3)
    DEPLOYER = "deployer"            # 배포자 - Article 3(4)
    IMPORTER = "importer"            # 수입자 - Article 3(6)
    DISTRIBUTOR = "distributor"      # 유통자 - Article 3(7)
    AUTHORIZED_REP = "authorized_representative"  # 위임대리인

@dataclass
class EUAIActClassifier:
    """EU AI Act 위험 분류기"""

    # 금지 AI 유형 (Article 5)
    PROHIBITED_PRACTICES = [
        "subliminal_manipulation",      # 잠재의식 조작
        "exploitation_of_vulnerabilities",  # 취약성 악용
        "social_scoring_by_public",     # 공공기관 사회점수
        "predictive_policing_individual",  # 개인 대상 예측 치안
        "emotion_recognition_workplace",   # 직장/학교 감정인식
        "untargeted_facial_scraping",   # 무차별 얼굴 수집
        "biometric_categorization_sensitive",  # 민감정보 생체분류
        "real_time_rbi_public_spaces",  # 실시간 원격 생체인식(예외 제외)
    ]

    # 고위험 분야 (Annex III)
    HIGH_RISK_AREAS = {
        "biometrics": "생체인식 및 분류",
        "critical_infrastructure": "핵심 인프라 관리",
        "education": "교육 및 직업훈련",
        "employment": "고용, 근로자 관리, 자영업 접근",
        "essential_services": "필수 서비스(신용, 보험 등) 접근",
        "law_enforcement": "법 집행",
        "migration": "이민, 망명, 국경 통제",
        "justice": "사법 및 민주적 절차",
    }

    def classify_risk(self, ai_system: Dict) -> Dict:
        """AI 시스템 위험 분류"""
        # 금지 여부 확인
        if self._check_prohibited(ai_system):
            return {
                "risk_level": RiskLevel.PROHIBITED,
                "legal_basis": "Article 5",
                "consequence": "시스템 출시/사용 금지",
                "penalty": "최대 3,500만 유로 또는 전 세계 매출 7%"
            }

        # 고위험 여부 확인
        if self._check_high_risk(ai_system):
            return {
                "risk_level": RiskLevel.HIGH_RISK,
                "legal_basis": "Article 6, Annex III",
                "obligations": self._get_high_risk_obligations(),
                "penalty": "최대 1,500만 유로 또는 전 세계 매출 3%"
            }

        # 제한적 위험 확인 (투명성 의무)
        if self._check_limited_risk(ai_system):
            return {
                "risk_level": RiskLevel.LIMITED_RISK,
                "legal_basis": "Article 50",
                "obligations": ["AI 사용 고지", "딥페이크 라벨링", "감정인식 고지"],
                "penalty": "최대 750만 유로 또는 전 세계 매출 1.5%"
            }

        return {
            "risk_level": RiskLevel.MINIMAL_RISK,
            "obligations": ["자발적 행동강령 권장"],
            "penalty": "없음 (자율 규제)"
        }

    def _check_prohibited(self, system: Dict) -> bool:
        """금지 AI 해당 여부"""
        return system.get("practice_type") in self.PROHIBITED_PRACTICES

    def _check_high_risk(self, system: Dict) -> bool:
        """고위험 AI 해당 여부"""
        # Annex III 분야 확인
        if system.get("application_area") in self.HIGH_RISK_AREAS:
            return True
        # Annex I 제품 안전 규정 해당 여부
        if system.get("safety_component") and system.get("product_regulation"):
            return True
        return False

    def _check_limited_risk(self, system: Dict) -> bool:
        """제한적 위험 해당 여부"""
        limited_types = ["chatbot", "deepfake", "emotion_recognition", "biometric_categorization"]
        return system.get("system_type") in limited_types

    def _get_high_risk_obligations(self) -> List[Dict]:
        """고위험 AI 의무사항"""
        return [
            {"article": 9, "title": "위험관리시스템", "description": "전체 수명주기 위험 식별/완화"},
            {"article": 10, "title": "데이터 거버넌스", "description": "학습/검증/테스트 데이터 품질"},
            {"article": 11, "title": "기술 문서", "description": "부속서 IV 준수 문서화"},
            {"article": 12, "title": "기록 보관", "description": "자동 로깅, 추적성 확보"},
            {"article": 13, "title": "투명성", "description": "사용자 대상 정보 제공"},
            {"article": 14, "title": "인간 감독", "description": "인간 개입/중지 메커니즘"},
            {"article": 15, "title": "정확성/견고성/보안", "description": "성능 수준 유지"},
            {"article": 17, "title": "품질관리시스템", "description": "QMS 구축 및 운영"},
        ]

@dataclass
class ComplianceTimeline:
    """EU AI Act 시행 일정"""

    MILESTONES = {
        "2024-08-01": "EU AI Act 발효",
        "2025-02-02": "금지 AI 조항 적용 (Article 5)",
        "2025-08-02": "GPAI 모델 조항 적용, AI 사무국/위원회 설립",
        "2026-08-02": "고위험 AI 전면 적용 (Chapter III Section 2)",
        "2027-08-02": "Annex I 제품 관련 고위험 AI 적용",
    }

    def get_applicable_date(self, risk_level: RiskLevel) -> str:
        if risk_level == RiskLevel.PROHIBITED:
            return "2025-02-02"
        elif risk_level == RiskLevel.HIGH_RISK:
            return "2026-08-02"
        return "2024-08-01"

# 사용 예시
classifier = EUAIActClassifier()

# AI 채용 시스템 분류
hr_ai = {
    "name": "AI 채용 스크리닝 시스템",
    "application_area": "employment",
    "system_type": "decision_support"
}
result = classifier.classify_risk(hr_ai)
print(f"위험 등급: {result['risk_level'].value}")
print(f"법적 근거: {result.get('legal_basis')}")
print(f"의무사항 수: {len(result.get('obligations', []))}")