Complete step 6

npc_social_network/npc/emotion_config.py

@@ -3,79 +3,88 @@
 
 # 고차원 감정 상태 정의 (GoEmotions 기반, 보여주기용)
 EMOTION_STATE_TEMPLATE = {
-                        # 기본 정서 (긍정)
-                        "core_positive":{
-                            "joy": 0,           # 기쁨
-                            "satisfaction": 0,  # 만족
-                            "gratitude": 0,     # 감사
-                            "calm": 0,          # 평온
-                            "anticipation": 0,  # 기대
-                            "pride": 0,         # 자부심
-                            "connectedness": 0  # 유대감
-                        },
-                        # 기본 정서 (부정)
-                        "core_negative": {
-                                "sadness": 0,       # 슬픔
-                                "anger": 0,         # 분노
-                                "anxiety": 0,       # 불안
-                                "disgust": 0,       # 혐오
-                                "fear": 0,          # 공포
-                                "regret": 0,        # 후회
-                                "frustration": 0    # 좌절
-                        },
-                        # 사회적 감정
-                        "social_emotion": {
-                            "jealousy": 0,      # 질투
-                            "shame": 0,         # 수치심
-                            "guilt": 0,         # 죄책감
-                            "compassion": 0,    # 동정심
-                            "awe": 0,           # 경외심
-                            "empathy": 0        # 공감
-                        },
-                        # 인지적 상태
-                        "cognitive_state": {
-                            "confusion": 0,     # 혼란
-                            "nervousness": 0,   # 긴장
-                            "apathy": 0,        # 무관심
-                            "excitement": 0,    # 흥분
-                            "immersion": 0,     # 몰입
-                            "skepticism": 0     # 회의
-                        }
-                        }
+    # 기본 정서 (긍정)
+    "core_positive":{
+        "joy": 0,           # 기쁨
+        "satisfaction": 0,  # 만족
+        "gratitude": 0,     # 감사
+        "calm": 0,          # 평온
+        "anticipation": 0,  # 기대
+        "pride": 0,         # 자부심
+        "connectedness": 0  # 유대감
+    },
+    # 기본 정서 (부정)
+    "core_negative": {
+            "sadness": 0,       # 슬픔
+            "anger": 0,         # 분노
+            "anxiety": 0,       # 불안
+            "disgust": 0,       # 혐오
+            "fear": 0,          # 공포
+            "regret": 0,        # 후회
+            "frustration": 0    # 좌절
+    },
+    # 사회적 감정
+    "social_emotion": {
+        "jealousy": 0,      # 질투
+        "shame": 0,         # 수치심
+        "guilt": 0,         # 죄책감
+        "compassion": 0,    # 동정심
+        "awe": 0,           # 경외심
+        "empathy": 0        # 공감
+    },
+    # 인지적 상태
+    "cognitive_state": {
+        "confusion": 0,     # 혼란
+        "nervousness": 0,   # 긴장
+        "apathy": 0,        # 무관심
+        "excitement": 0,    # 흥분
+        "immersion": 0,     # 몰입
+        "skepticism": 0     # 회의
+    }
+}
 
 # 감정별 회복 속도 (ms 단위가 아닌 단위당 감소 속도)
 EMOTION_DECAY_RATE = {
-                    # 기본 정서 (긍정)
-                        "joy": 1,           # 기쁨
-                        "satisfaction": 1,  # 만족
-                        "gratitude": 1,     # 감사
-                        "calm": 1,          # 평온
-                        "anticipation": 1,  # 기대
-                        "pride": 1,         # 자부심
-                        "connectedness": 1,  # 유대감
+# 기본 정서 (긍정)
+    "joy": 1,           # 기쁨
+    "satisfaction": 1,  # 만족
+    "gratitude": 1,     # 감사
+    "calm": 1,          # 평온
+    "anticipation": 1,  # 기대
+    "pride": 1,         # 자부심
+    "connectedness": 1,  # 유대감
 
-                    # 기본 정서 (부정) - 느림
-                        "sadness": 0.2,       # 슬픔
-                        "anger": 0.5,         # 분노
-                        "anxiety": 0.4,       # 불안
-                        "disgust": 0.8,       # 혐오
-                        "fear": 0.6,          # 공포
-                        "regret": 0.3,        # 후회
-                        "frustration": 0.5,    # 좌절
+# 기본 정서 (부정) - 느림
+    "sadness": 0.2,       # 슬픔
+    "anger": 0.5,         # 분노
+    "anxiety": 0.4,       # 불안
+    "disgust": 0.8,       # 혐오
+    "fear": 0.6,          # 공포
+    "regret": 0.3,        # 후회
+    "frustration": 0.5,    # 좌절
 
-                    # 사회적 감정
-                        "jealousy": 0.4,      # 질투
-                        "shame": 0.4,         # 수치심
-                        "guilt": 0.4,         # 죄책감
-                        "compassion": 1,    # 동정심
-                        "awe": 1,           # 경외심
-                        "empathy": 1,        # 공감
+# 사회적 감정
+    "jealousy": 0.4,      # 질투
+    "shame": 0.4,         # 수치심
+    "guilt": 0.4,         # 죄책감
+    "compassion": 1,    # 동정심
+    "awe": 1,           # 경외심
+    "empathy": 1,        # 공감
 
-                    # 인지적 상태
-                        "confusion": 0.8,     # 혼란
-                        "nervousness": 0.7,   # 긴장
-                        "apathy": 0.4,        # 무관심
-                        "excitement": 1.2,    # 흥분
-                        "immersion": 1.0,     # 몰입
-                        "skepticism": 0.5,     # 회의
-                    }
+# 인지적 상태
+    "confusion": 0.8,     # 혼란
+    "nervousness": 0.7,   # 긴장
+    "apathy": 0.4,        # 무관심
+    "excitement": 1.2,    # 흥분
+    "immersion": 1.0,     # 몰입
+    "skepticism": 0.5,     # 회의
+}
+
+# 개성 템플릿(0~1 범위)
+PERSONALITY_TEMPLATE = {
+    "sensitive" : 0.8,          # 감정 반응 강도 증가
+    "stoic": 0.2,               # 감정 반응 둔화
+    "positive_bias": 0.6,            # 긍정 감정에 민감
+    "negative_bias": 0.4,       # 부정 감정에 민감
+    "sociality" : 0.5,          # 사회적 감정 민감
+}

npc_social_network/npc/npc_base.py

@@ -2,7 +2,7 @@
 from .npc_memory import Memory, MemoryStore
 from .npc_emotion import EmotionManager
 from .npc_behavior import BehaviorManager
-from .emotion_config import EMOTION_STATE_TEMPLATE, EMOTION_DECAY_RATE
+from .emotion_config import EMOTION_STATE_TEMPLATE, EMOTION_DECAY_RATE, PERSONALITY_TEMPLATE
 
 # NPC 클래스 정의
 class NPC:
@@ -10,23 +10,25 @@ class NPC:
     NPC 클래스
     NPC에 대한 기본 정보를 객체화
     """
-    def __init__(self, name, job, path, image):
+    def __init__(self, name, job, path, image, personality=None):
         self.name = name
         self.job = job
         self.path = path
         self.image = image
         self.current_index = 0
+        self.personality = personality or {}
 
         # npc 기억
         self.memory_store = MemoryStore()
-        # 감정 상태 및 감정 관리자 초기화
+        # 감정 상태 등 감정 관리자 초기화
         self.emotion_state = EMOTION_STATE_TEMPLATE
         self.emotion_decay_rate = EMOTION_DECAY_RATE
-        self.emotion = EmotionManager(self.emotion_state, self.emotion_decay_rate)
+        self.personality = personality or PERSONALITY_TEMPLATE
+
+        self.emotion = EmotionManager(self.emotion_state, self.emotion_decay_rate, self.personality)
 
         # 행동 관리자
         self.behavior = BehaviorManager()
-
         # 내부: float 감정 수치 관리용 버퍼
         self._emotion_buffer = self.emotion.get_buffer()
 

npc_social_network/npc/npc_emotion.py

@@ -1,29 +1,49 @@
 # 감정 상태, 감정 업데이트, decay 처리
 # portfolio/npc_social_network/npc/npc_emotion.py
+from .emotion_config import EMOTION_DECAY_RATE
 
 class EmotionManager:
-    def __init__(self, emotion_state, decay_rate):
+    def __init__(self, emotion_state, decay_rate, personality = None):
         # 고차원 감정 상태 (GoEmotions 기반)
         self.emotion_state = emotion_state
         # 감정별 회복 속도 (ms 단위가 아닌 단위당 감소 속도)
         self.decay_rate = decay_rate
         # 내부: float 감정 수치 관리용 버퍼
-        self._emotion_buffer = {emo: 0.0 for cat in emotion_state for emo in emotion_state[cat]}
-        for cat in emotion_state:
-            for emo in emotion_state[cat]:
-                self._emotion_buffer[emo] = float(emotion_state[cat][emo])
+        self._emotion_buffer = {emo: 0.0 for cat in emotion_state.values() for emo in cat}
+        self.personality = personality or {}
 
     # 특정 감정을 감정 상태에 반영 (카테고리 기반)
     def update_emotion(self, emotion: str, strength: float = 1.0):
-        if emotion in self._emotion_buffer:
-            self._emotion_buffer[emotion] += strength
-            self._emotion_buffer[emotion] = max(0.0, self._emotion_buffer[emotion])     # 최소값을 0으로 고정
+        if emotion not in self._emotion_buffer:
+            print(f"[경고] '{emotion}'은(는) 정의된 감정이 아닙니다.")
+            return
+        
+        # 개성 계수 반영
+        multiplier = 1.0
+        if emotion in ["joy", "gratitude", "pride"]:
+            multiplier *= self.personality.get("positive_bias", 1)
+        elif emotion in ["anger", "fear", "regret"]:
+            multiplier *= self.personality.get("negative_bias", 1)
+
+        multiplier *= self.personality.get("sensitive", 1)
+
+        self._emotion_buffer[emotion] += strength * multiplier
+        for category in self.emotion_state:
+            if emotion in self.emotion_state[category]:
+                self.emotion_state[category][emotion] = int(self._emotion_buffer[emotion])
+                break
+
 
     # 시간이 지남에 따라 모든 감정이 서서히 감소
     def decay_emotion(self):
-        for emo in self._emotion_buffer:
-            rate = self.decay_rate.get(emo, 1.0)
-            self._emotion_buffer[emo] = max(0.0, self._emotion_buffer[emo] - rate)      # 최소값을 0으로 고정
+        for emotion, value in self._emotion_buffer.items():
+            rate = self.decay_rate.get(emotion, 0.5)
+            modifier = 1.0 - self.personality.get("stoic", 0)
+            self._emotion_buffer[emotion] = max(0.0, value - rate * modifier)
+
+        for category in self.emotion_state:
+            for emotion in self.emotion_state[category]:
+                self.emotion_state[category][emotion] = int(self._emotion_buffer[emotion])
 
     # 정수 기반 출력용 감정 상태 반환
     def get_state(self):
@@ -31,7 +51,9 @@ class EmotionManager:
     
     # 현재 가장 강한 감정 반환 (전체 감정 중 최댓값 1개)
     def get_dominant_emotion(self):
-        return max(self._emotion_buffer.items(), key=lambda x: x[1], default=(None, 0.0))[0]
+        if not self._emotion_buffer:
+            return "neutral"
+        return max(self._emotion_buffer.items(), key=lambda x: x[1])[0]
     
     def get_buffer(self):
         return self._emotion_buffer.copy()

test.ipynb

@@ -2822,7 +2822,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 1,
    "id": "cc6f2948",
    "metadata": {},
    "outputs": [
@@ -2830,13 +2830,24 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "d:\\private\\Portfolio\\npc_social_network\\npc\\npc_base.py\n"
+      "감정 평균 강도: 0.96 / 대표 감정: fear(0.96)\n"
      ]
     }
    ],
    "source": [
-    "import npc_social_network.npc.npc_base as base\n",
-    "print(base.__file__)"
+    "from npc_social_network.npc.npc_base import NPC\n",
+    "\n",
+    "# 민감하고 부정에 취약한 캐릭터\n",
+    "custom_personality = {\n",
+    "    \"sensitive\": 1.0,\n",
+    "    \"negative_bias\": 1.5,\n",
+    "    \"stoic\": 0.1\n",
+    "}\n",
+    "\n",
+    "npc = NPC(\"아린\", \"farmer\", [(0, 0)], None, personality=custom_personality)\n",
+    "npc.remember(\"무서운 소리를 들음\", emotion=\"fear\", importance=5)\n",
+    "npc.decay_emotion()\n",
+    "print(npc.summarize_emotional_state())"
    ]
   }
  ],