복합 감정 계산 및 상태 요약

npc_social_network/npc/npc_base.py

@@ -127,11 +127,11 @@ class NPC:
         return f"{emotion_line} {job_line}"
     
     # NPC가 새로운 기억을 저장하고 감정 상태에 반영
-    def remember(self, content: str, importance: int = 5, emotion: str = None):
+    def remember(self, content: str, importance: int = 5, emotion: str = None, strength: float = 1.0):
         memory = Memory(content=content, importance = importance, emotion=emotion or "neutral")
         self.memory_store.add_memory(memory)
         if emotion:
-            self.update_emotion(emotion)
+            self.update_emotion(emotion, strength)  # strength = importance / 5.0으로도 가능
 
     # 저장된 모든 기억(단기+장기)을 리스트로 반환
     def recall(self):
@@ -172,3 +172,17 @@ class NPC:
                 max_value = val
                 max_emotion = emo
         return max_emotion
+
+    # 복합 감정 계산 (상위 N개 감정 반환)
+    def get_composite_emotion_state(self, top_n: int = 3):
+        sorted_emotions = sorted(self._emotion_buffer.items(), key=lambda x: x[1], reverse=True)
+        composite = [(emo,round(score, 2)) for emo, score in sorted_emotions[:top_n] if score > 0]
+        return composite # 예 [('fear', 2.0), ('anger'), 1.5]
+    
+    # 감정 상태 요약 텍스트 생성
+    def summarize_emotional_state(self):
+        nonzero = [v for v in self._emotion_buffer.values() if v > 0]
+        avg_strength = round(sum(nonzero) / len(nonzero), 2) if nonzero else 0.0
+        composite = self.get_composite_emotion_state()
+        composite_str = ", ".join(f"{emo}({val})" for emo, val in composite)
+        return f"감정 평균 강도: {avg_strength} / 대표 감정: {composite_str if composite else '없음' }"

test.ipynb

@@ -2754,7 +2754,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "id": "ac67ce84",
    "metadata": {},
    "outputs": [
@@ -2762,26 +2762,34 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[DEBUG] update_emotion: Attempting to update 'anger' by 1.0.\n",
-      "[DEBUG] update_emotion: Attempting to update 'fear' by 1.0.\n",
-      "[DEBUG] update_emotion: Attempting to update 'pride' by 1.0.\n",
       "초기 감정 상태:\n",
-      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 1.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 1.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 1.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "[('fear', 2.2), ('anger', 1.5), ('pride', 1.2)]\n",
+      "감정 평균 강도: 1.63 / 대표 감정: fear(2.2), anger(1.5), pride(1.2)\n",
       "\n",
       "[1회 decay 후]\n",
-      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.5, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.4, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.19999999999999996, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 1.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 1.6, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "[('fear', 1.6), ('anger', 1.0), ('pride', 0.2)]\n",
+      "감정 평균 강도: 0.93 / 대표 감정: fear(1.6), anger(1.0), pride(0.2)\n",
       "\n",
       "[2회 decay 후]\n",
-      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.5, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 1.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "[('fear', 1.0), ('anger', 0.5)]\n",
+      "감정 평균 강도: 0.75 / 대표 감정: fear(1.0), anger(0.5)\n",
       "\n",
       "[3회 decay 후]\n",
-      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.4, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "[('fear', 0.4)]\n",
+      "감정 평균 강도: 0.4 / 대표 감정: fear(0.4)\n",
       "\n",
       "[4회 decay 후]\n",
       "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "[]\n",
+      "감정 평균 강도: 0.0 / 대표 감정: 없음\n",
       "\n",
       "[5회 decay 후]\n",
-      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n"
+      "{'core_positive': {'joy': 0.0, 'satisfaction': 0.0, 'gratitude': 0.0, 'calm': 0.0, 'anticipation': 0.0, 'pride': 0.0, 'connectedness': 0.0}, 'core_negative': {'sadness': 0.0, 'anger': 0.0, 'anxiety': 0.0, 'disgust': 0.0, 'fear': 0.0, 'regret': 0.0, 'frustration': 0.0}, 'social_emotion': {'jealousy': 0.0, 'shame': 0.0, 'guilt': 0.0, 'compassion': 0.0, 'awe': 0.0, 'empathy': 0.0}, 'cognitive_state': {'confusion': 0.0, 'nervousness': 0.0, 'apathy': 0.0, 'excitement': 0.0, 'immersion': 0.0, 'skepticism': 0.0}}\n",
+      "[]\n",
+      "감정 평균 강도: 0.0 / 대표 감정: 없음\n"
      ]
     }
    ],
@@ -2795,18 +2803,21 @@
     "\n",
     "# 3. NPC 생성 및 감정 입력\n",
     "npc = NPC(name=\"아린\", job=\"farmer\", path=[(0,0)], image=None)\n",
-    "npc.remember(\"플레이어가 모욕함\", importance=6, emotion=\"anger\")\n",
-    "npc.remember(\"무서운 소리를 들음\", importance=5, emotion=\"fear\")\n",
-    "npc.remember(\"친구가 칭찬함\", importance=5, emotion=\"pride\")\n",
+    "npc.remember(\"플레이어가 모욕함\", importance=6, emotion=\"anger\", strength=1.5)\n",
+    "npc.remember(\"무서운 소리를 들음\", importance=5, emotion=\"fear\", strength=2.2)\n",
+    "npc.remember(\"친구가 칭찬함\", importance=5, emotion=\"pride\", strength=1.2)\n",
     "\n",
     "print(\"초기 감정 상태:\")\n",
-    "print(npc.get_current_emotion_state())\n",
+    "print(npc.get_composite_emotion_state())         # [('fear', 2.0), ('anger', 1.5), ('pride', 1.2)]\n",
+    "print(npc.summarize_emotional_state())           # 감정 평균 강도: 1.57 / 대표 감정: fear(2.0), anger(1.5), pride(1.2)\n",
     "\n",
     "# 4. decay 반복 적용\n",
     "for step in range(1, 6):\n",
     "    npc.decay_emotion()\n",
     "    print(f\"\\n[{step}회 decay 후]\")\n",
-    "    print(npc.get_current_emotion_state())"
+    "    print(npc.get_current_emotion_state())\n",
+    "    print(npc.get_composite_emotion_state())\n",
+    "    print(npc.summarize_emotional_state())"
    ]
   },
   {