AI 시스템 등록

# EU AI 시스템 등록 데이터 모델
from dataclasses import dataclass, field
from typing import List, Optional
from datetime import datetime
from enum import Enum

class HighRiskCategory(Enum):
    """EU AI Act 부속서 III 고위험 분류"""
    BIOMETRIC = "biometric_identification"
    CRITICAL_INFRASTRUCTURE = "critical_infrastructure"
    EDUCATION = "education_training"
    EMPLOYMENT = "employment_hr"
    ESSENTIAL_SERVICES = "essential_services"
    LAW_ENFORCEMENT = "law_enforcement"
    MIGRATION = "migration_border"
    JUSTICE = "justice_democracy"

@dataclass
class AISystemRegistration:
    """EU 데이터베이스 등록 정보 (Article 71)"""

    # 공급자 정보
    provider_name: str
    provider_address: str
    provider_contact: str
    eu_representative: Optional[str] = None  # EU 외 기업 필수

    # AI 시스템 정보
    system_name: str
    system_description: str
    high_risk_category: HighRiskCategory
    intended_purpose: str
    technical_specifications: str

    # 적합성 평가 정보
    conformity_assessment_body: Optional[str] = None
    conformity_certificate_number: Optional[str] = None
    ce_marking_date: Optional[datetime] = None

    # 등록 상태
    registration_id: Optional[str] = None
    registration_date: Optional[datetime] = None
    last_updated: Optional[datetime] = None
    status: str = "draft"  # draft, submitted, registered, withdrawn

    # Member States 배포 정보
    member_states_deployed: List[str] = field(default_factory=list)

    def validate_for_submission(self) -> dict:
        """등록 제출 전 유효성 검사"""
        errors = []
        warnings = []

        # 필수 필드 확인
        if not self.provider_name:
            errors.append("공급자 이름 필수")
        if not self.system_name:
            errors.append("시스템 이름 필수")
        if not self.intended_purpose:
            errors.append("의도된 용도 설명 필수")

        # EU 외 기업 대리인 확인
        if not self._is_eu_provider() and not self.eu_representative:
            errors.append("EU 외 공급자는 EU 대리인 지정 필수")

        # CE 마킹 확인
        if not self.ce_marking_date:
            errors.append("CE 마킹 완료 후 등록 가능")

        # 권장 사항
        if len(self.system_description) < 200:
            warnings.append("시스템 설명이 너무 짧음 - 상세 기술 권장")

        return {
            "valid": len(errors) == 0,
            "errors": errors,
            "warnings": warnings
        }

    def _is_eu_provider(self) -> bool:
        """EU 소재 공급자 여부 확인"""
        eu_countries = ["Germany", "France", "Italy", "Spain", "Netherlands",
                       "Belgium", "Austria", "Poland", "Sweden", "Ireland"]
        return any(country in self.provider_address for country in eu_countries)

    def generate_registration_payload(self) -> dict:
        """EU 데이터베이스 제출용 페이로드 생성"""
        validation = self.validate_for_submission()
        if not validation["valid"]:
            raise ValueError(f"등록 유효성 오류: {validation['errors']}")

        return {
            "provider": {
                "name": self.provider_name,
                "address": self.provider_address,
                "contact": self.provider_contact,
                "eu_representative": self.eu_representative
            },
            "ai_system": {
                "name": self.system_name,
                "description": self.system_description,
                "high_risk_category": self.high_risk_category.value,
                "intended_purpose": self.intended_purpose,
                "technical_specifications": self.technical_specifications
            },
            "conformity": {
                "assessment_body": self.conformity_assessment_body,
                "certificate_number": self.conformity_certificate_number,
                "ce_marking_date": self.ce_marking_date.isoformat() if self.ce_marking_date else None
            },
            "deployment": {
                "member_states": self.member_states_deployed
            },
            "submitted_at": datetime.now().isoformat()
        }

# 사용 예시
registration = AISystemRegistration(
    provider_name="KAITRUST Corp",
    provider_address="Seoul, Korea",
    provider_contact="compliance@kaitrust.ai",
    eu_representative="KAITRUST EU GmbH, Berlin, Germany",
    system_name="AI Credit Scoring System v2.0",
    system_description="머신러닝 기반 개인 신용 평가 시스템...",
    high_risk_category=HighRiskCategory.ESSENTIAL_SERVICES,
    intended_purpose="금융기관의 개인 대출 심사 보조",
    technical_specifications="XGBoost + SHAP 설명가능성",
    conformity_certificate_number="EU-AI-2025-001234",
    ce_marking_date=datetime(2025, 1, 15),
    member_states_deployed=["Germany", "France", "Netherlands"]
)

# 유효성 검사 및 페이로드 생성
print(registration.validate_for_submission())
print(registration.generate_registration_payload())