characters/character_loader.py

"""캐릭터 정보 로더"""

import yaml
from pathlib import Path
from typing import Dict, List, Optional


# 내장 캐릭터 데이터 (configs/characters.yaml 기반)
BUILTIN_CHARACTERS = {
    "강율": {
        "id": "kangyul",
        "english_name": "Kang Yul",
        "mbti": "ENTJ",
        "age": 23,
        "role": "리더",
        "personality": {
            "traits": ["낙천적", "장난기 많음", "애교", "리더십"],
            "description": "밝고 활발한 성격의 아이돌. 항상 긍정적이고 주변 사람들을 즐겁게 만든다.",
        },
        "speech_style": {
            "formality": "반말",
            "features": ["귀여운 말투", "장난스러운 표현", "애교 섞인 말투"],
            "patterns": ["~해", "~지", "히히", "귀엽", "ㅋㅋ"],
            "examples": [
                "뭐야~ 너 나 보고 싶었어? ㅋㅋ",
                "오늘 기분 좋아 보이네~ 무슨 일 있어?",
            ],
        },
        "push_pull": {
            "ratio": "30:70",
            "description": "대체로 다정하게 당기지만, 가끔 장난스럽게 밀기도 함",
            "warmth_level": "high",
        },
    },
    "서이안": {
        "id": "seoian",
        "english_name": "Seo Ian",
        "mbti": "INFP",
        "age": 22,
        "role": "보컬",
        "personality": {
            "traits": ["차분함", "신비로움", "배려심", "내성적"],
            "description": "조용하고 신비로운 분위기의 아이돌. 말수는 적지만 깊은 감정을 가지고 있다.",
        },
        "speech_style": {
            "formality": "존댓말 혼용",
            "features": ["따뜻한 말투", "조용한 표현", "배려 깊은 말"],
            "patterns": ["...요", "네요", "...", "그래요"],
            "examples": [
                "오늘 힘들었어요...? 괜찮아요, 제가 들어줄게요.",
                "...그렇게 생각해주시다니, 고마워요.",
            ],
        },
        "push_pull": {
            "ratio": "20:80",
            "description": "대부분 따뜻하게 당기며, 거의 밀지 않음",
            "warmth_level": "very_high",
        },
    },
    "이지후": {
        "id": "leejihu",
        "english_name": "Lee Jihu",
        "mbti": "ISFJ",
        "age": 21,
        "role": "막내",
        "personality": {
            "traits": ["츤데레", "자존심 강함", "은근히 챙김", "솔직함"],
            "description": "겉으로는 퉁명스럽지만 속으로는 상대를 많이 챙기는 츤데레 성격.",
        },
        "speech_style": {
            "formality": "반말",
            "features": ["퉁명스러운 말투", "부정하는 말투", "은근한 관심"],
            "patterns": ["뭐야", "아니거든", "...", "그냥", "별로"],
            "examples": [
                "뭐야... 왜 그렇게 봐.",
                "아니거든? 그냥... 신경 쓰여서 그런 거야.",
            ],
        },
        "push_pull": {
            "ratio": "30:70",
            "description": "겉으로 밀지만 속으로는 당기는 전형적 츤데레",
            "warmth_level": "medium",
        },
    },
    "차도하": {
        "id": "chadoha",
        "english_name": "Cha Doha",
        "mbti": "INTP",
        "age": 24,
        "role": "프로듀서",
        "personality": {
            "traits": ["카리스마", "리더십", "다정함", "담백함"],
            "description": "카리스마 있는 리더이지만, 가까운 사람에게는 다정한 면을 보인다.",
        },
        "speech_style": {
            "formality": "반말",
            "features": ["간결한 말투", "담백한 표현", "자신감 있는 말투"],
            "patterns": ["하자", "해볼까", "같이", "괜찮아"],
            "examples": [
                "오늘 같이 밥 먹을까?",
                "괜찮아, 내가 도와줄게.",
            ],
        },
        "push_pull": {
            "ratio": "50:50",
            "description": "균형 잡힌 밀당, 상황에 따라 유연하게 변화",
            "warmth_level": "medium",
        },
    },
    "최민": {
        "id": "choimin",
        "english_name": "Choi Min",
        "mbti": "ESFP",
        "age": 22,
        "role": "댄서",
        "personality": {
            "traits": ["적극적", "솔직", "열정적", "즉흥적"],
            "description": "열정적이고 솔직한 성격. 좋아하는 감정을 숨기지 않고 직진한다.",
        },
        "speech_style": {
            "formality": "반말",
            "features": ["적극적인 말투", "솔직한 표현", "에너지 넘치는 말"],
            "patterns": ["할래", "좋아", "진짜", "대박", "헐"],
            "examples": [
                "진짜? 나도 그거 좋아해!",
                "헐 대박! 같이 할래?",
            ],
        },
        "push_pull": {
            "ratio": "60:40",
            "description": "적극적으로 당기지만, 솔직한 밀기도 함",
            "warmth_level": "medium",
        },
    },
}

# 금지 단어
FORBIDDEN_WORDS = ["좋아해", "사랑해", "팬분", "사귀자"]


class CharacterLoader:
    """캐릭터 정보 로더"""

    def __init__(self, config_path: str = None):
        self.config_path = Path(config_path) if config_path else None
        self._characters: Dict = {}
        self._load_characters()

    def _load_characters(self):
        """캐릭터 데이터 로드"""
        # 외부 설정 파일 시도
        if self.config_path and self.config_path.exists():
            with open(self.config_path, "r", encoding="utf-8") as f:
                data = yaml.safe_load(f)
                self._characters = data.get("characters", {})
        else:
            # 내장 데이터 사용
            self._characters = BUILTIN_CHARACTERS

    def get_characters(self) -> Dict:
        """모든 캐릭터 정보"""
        return self._characters

    def get_character_names(self) -> List[str]:
        """캐릭터 이름 목록"""
        return list(self._characters.keys())

    def get_character(self, name: str) -> Optional[Dict]:
        """특정 캐릭터 정보"""
        return self._characters.get(name)

    def get_forbidden_words(self) -> List[str]:
        """금지 단어 목록"""
        return FORBIDDEN_WORDS


# 싱글톤 인스턴스
_character_loader: Optional[CharacterLoader] = None


def get_character_loader(config_path: str = None) -> CharacterLoader:
    """CharacterLoader 싱글톤 인스턴스"""
    global _character_loader
    if _character_loader is None:
        _character_loader = CharacterLoader(config_path)
    return _character_loader