감정 고도화 테스트 중

npc_social_network/npc/npc_base.py

@@ -17,7 +17,7 @@ class NPC:
         # npc 기억
         self.memory_store = MemoryStore()
 
-        # 고차원 감정 상태 정의 (GoEmotions 기반)
+        # 고차원 감정 상태 정의 (GoEmotions 기반, 보여주기용)
         self.emotion_state = {
                                 # 기본 정서 (긍정)
                                 "core_positive":{
@@ -58,6 +58,46 @@ class NPC:
                                     "skepticism": 0     # 회의
                                 }
                               }
+        
+        # 감정별 회복 속도 (ms 단위가 아닌 단위당 감소 속도)
+        self.emotion_decay_rate = {
+                                # 기본 정서 (긍정)
+                                    "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,    # 좌절
+
+                                # 사회적 감정
+                                    "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,     # 회의
+                            }
+
+        # 내부: float 감정 수치 관리용 버퍼
+        self._emotion_buffer = {emo: 0.0 for cat in self.emotion_state for emo in self.emotion_state[cat]}
 
         # 확장 가능 속성
         # self.emotion = "neutral"
@@ -76,24 +116,52 @@ class NPC:
         x, y = self.get_position()
         screen.blit(self.image, (x * tile_size, y * tile_size))
 
-    # 감정 상태에 따른 대사 생성 (대표 감정 하나 반영)
+    # 감정 상태와 직업에 따른 대사 생성 (대표 감정 하나 반영)
     def generate_dialogue(self):
-        pass
-
-    # 직업에 따른 기본 대사 생성
-    def generate_dialogue(self):
-        if self.job == "farmer":
-            return "오늘은 밭에 나갈 날이군요!"
-        elif self.job == "blacksmith":
-            return "망치를 들고 일할 준비가 되었어요!"
-        return "좋은 하루입니다."
+        dominant = self.get_dominant_emotion()
+        emotion_line = f"지금 저는 '{dominant}' 상태예요." if dominant else "기분이 안정돼 있어요."
+        job_line = {
+            "farmer": "밭일을 해야겠어요.",
+            "blacksmith": "대장간에 불을 지펴야겠군요."
+        }.get(self.job, "오늘도 바쁘네요.")
+        return f"{emotion_line} {job_line}"
     
-    # NPC가 새로운 기억을 저장
-    def remember(self, content: str, importance: int = 5, emotion: str = "중립"):
-        memory = Memory(content=content, importance = importance, emotion=emotion)
+    # NPC가 새로운 기억을 저장하고 감정 상태에 반영
+    def remember(self, content: str, importance: int = 5, emotion: str = None):
+        memory = Memory(content=content, importance = importance, emotion=emotion or "neutral")
         self.memory_store.add_memory(memory)
+        if emotion:
+            self.update_emotion(emotion)
 
     # 저장된 모든 기억(단기+장기)을 리스트로 반환
     def recall(self):
         return [m.content for m in self.memory_store.get_all_memories()]
-    
+    
+    # 특정 감정을 감정 상태에 반영 (카테고리 기반)
+    def update_emotion(self, emotion: str, strength: int = 1):
+        if emotion in self._emotion_buffer:
+            self._emotion_buffer[emotion] += strength
+    
+    # 시간이 지남에 따라 모든 감정이 서서히 감소
+    def decay_emotion(self):
+        for emotion in self._emotion_buffer:
+            rate = self.emotion_decay_rate.get(emotion, 1.0)
+            self._emotion_buffer[emotion] = max(0.0, self._emotion_buffer[emotion] - rate)
+
+    # 정수 기반 출력용 감정 상태 반환
+    def get_current_emotion_state(self):
+        display_state = {cat: {} for cat in self.emotion_state}
+        for cat in self.emotion_state:
+            for emo in self.emotion_state[cat]:
+                display_state[cat][emo] = float(self._emotion_buffer[emo])
+        return display_state
+
+    # 현재 가장 강한 감정 반환 (전체 감정 중 최댓값 1개)
+    def get_dominant_emotion(self):
+        max_value = 0.0
+        max_emotion = None
+        for emo, val in self._emotion_buffer.items():
+            if val > max_value:
+                max_value = val
+                max_emotion = emo
+        return max_emotion

test.ipynb

@@ -2711,7 +2711,13 @@
      "output_type": "stream",
      "text": [
       "pygame 2.6.1 (SDL 2.28.4, Python 3.11.11)\n",
-      "Hello from the pygame community. https://www.pygame.org/contribute.html\n"
+      "Hello from the pygame community. https://www.pygame.org/contribute.html\n",
+      "\n",
+      "👤 Elin (farmer)\n",
+      " - 위치: (5, 4)\n",
+      " - 대화: 오늘은 밭에 나갈 날이군요!\n",
+      " - 기억: []\n",
+      "아침 - Elin이(가) market에 도착했습니다.\n"
      ]
     },
     {
@@ -2727,7 +2733,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "c:\\Users\\stina\\.conda\\envs\\portfolio\\Lib\\site-packages\\IPython\\core\\interactiveshell.py:3678: UserWarning: To exit: use 'exit', 'quit', or Ctrl-D.\n",
+      "C:\\Users\\human\\AppData\\Roaming\\Python\\Python311\\site-packages\\IPython\\core\\interactiveshell.py:3675: UserWarning: To exit: use 'exit', 'quit', or Ctrl-D.\n",
       "  warn(\"To exit: use 'exit', 'quit', or Ctrl-D.\", stacklevel=1)\n"
      ]
     }
@@ -2745,6 +2751,79 @@
     "\n",
     "# town_hall.py는 __main__이 아니라 직접 실행되므로"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ac67ce84",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "초기 감정 상태:\n",
+      "{'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}}\n",
+      "\n",
+      "[1회 decay 후]\n",
+      "{'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}}\n",
+      "\n",
+      "[2회 decay 후]\n",
+      "{'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}}\n",
+      "\n",
+      "[3회 decay 후]\n",
+      "{'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}}\n",
+      "\n",
+      "[4회 decay 후]\n",
+      "{'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}}\n",
+      "\n",
+      "[5회 decay 후]\n",
+      "{'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}}\n"
+     ]
+    }
+   ],
+   "source": [
+    "# 1. 경로 추가\n",
+    "import sys, os\n",
+    "sys.path.append(os.path.abspath(\".\"))  # 프로젝트 루트 경로\n",
+    "\n",
+    "# 2. 클래스 import\n",
+    "from npc_social_network.npc.npc_base import NPC\n",
+    "\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",
+    "\n",
+    "print(\"초기 감정 상태:\")\n",
+    "print(npc.get_current_emotion_state())\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())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "id": "cc6f2948",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "d:\\private\\Portfolio\\npc_social_network\\npc\\npc_base.py\n"
+     ]
+    }
+   ],
+   "source": [
+    "import npc_social_network.npc.npc_base as base\n",
+    "print(base.__file__)"
+   ]
   }
  ],
  "metadata": {