Complete step 10: 감정 상태 UI 연동 및 테스트 확인

app.py

@@ -1,8 +1,9 @@
 # portfolio/app.py
-from flask import Flask, render_template
+from flask import Flask, render_template, url_for, redirect
 from npc_social_network.routes.npc_route import npc_bp
 # from stock.routes.stock_route import stock_bp
 import os
+import subprocess
 
 def create_app():
     app = Flask(__name__) # static/template 경로를 기본값
@@ -15,7 +16,15 @@ def create_app():
     def index():
         return render_template("main.html")
     
+    @app.route("/run_npc_simulation")
+    def run_npc_simulation():
+        # subprocess로 Pygame 실행
+        python_exec = "python"  # 가상환경 사용 필요시 full path로 변경 가능    
+        subprocess.Popen([python_exec, "-m", "npc_social_network.maps.villages.town_hall"]) # 비동기로 실행 (창 띄움)
 
+        # 실행 후 사용자 안내 페이지 표시
+        return render_template("npc_simulation_started.html")
+    
     return app
 
 app = create_app()

npc_social_network/maps/engine/game_engine.py

@@ -1,6 +1,11 @@
 # portfolio/npc_social_network/maps/engine/game_engine.py
 # 게임 루프 및 이벤트 처리
 import pygame
+from pygame import Rect
+import sys
+from npc_social_network.maps.manager.image_loader import ImageLoader
+from npc_social_network.maps.manager.village_map import VillageMap
+from npc_social_network.npc.npc_manager import NPCManager
 from datetime import datetime, timedelta
 
 class GameEngine:
@@ -10,12 +15,27 @@ class GameEngine:
     def __init__(self, screen, font, village, time_states, npcs):
         self.screen = screen
         self.font = font
+        # emotion HUD용 작은 글자 font 추가
+        self.emotion_font = pygame.font.SysFont("malgungothic", 14)
         self.village = village
         self.time_states = time_states
         self.current_time_index = 0
         self.frame_count = 0
         self.frame_per_time_state = 200     # 기존: 시간 상태 전환 프레임
         self.npcs = npcs                    # 전체 NPC 리스트
+        self.selected_npc = None            # 선택된 NPC 상태 기본 필드
+
+        # 감정 바 색상 테이블 (간단 예시)
+        self.emotion_colors = {
+            "joy": (255, 215, 0),
+            "gratitude": (173, 216, 230),
+            "anger": (255, 69, 0),
+            "fear": (138, 43, 226),
+            "sadness": (70, 130 ,180),
+            "trust": (34, 139, 34),
+            "anticipation": (255, 215, 0),
+            "disgust": (128, 128, 128),
+        }
 
         # 실제 시간 흐름 시뮬레이션용
         self.current_time = datetime.now()        # 현재 시뮬레이션 시간
@@ -32,6 +52,68 @@ class GameEngine:
         )
         self.screen.blit(time_text, (10, 10))
 
+        # 감정 상태 HUD 렌더링
+        if self.selected_npc:
+            self.draw_emotion_hud()
+
+    def draw_emotion_hud(self):
+        npc = self.selected_npc
+        top_n=5
+        top_emotions = npc.get_top_emotions(top_n=top_n)
+
+        # HUD 위치/크기 설정
+        hud_height = 150   # 원하는 높이
+        margin_right = 20  # 오른쪽 여백
+        margin_bottom = 20 # 아래쪽 여백 (화면 아래 띄우고 싶으면 사용)
+
+        hud_width = self.screen.get_width() * 0.45  # 화면 절반 너비
+        hud_rect = Rect(
+            self.screen.get_width() - hud_width - margin_right,
+            self.screen.get_height() - hud_height - margin_bottom,
+            hud_width,
+            hud_height
+        )
+        pygame.draw.rect(self.screen, (30, 30, 30), hud_rect)
+
+        padding_x = 20  # 오른쪽에서 얼마나 띄울지
+        padding_y = 10  # 위에서 얼마나 띄울지
+        spacing_y = 20 # 감정 간 간격
+        label_bar_gap = 100 # label과 bar 사이 간격
+        title_gap = 30  # 제목과 감정 표시 사이 간격
+        bar_width_max = hud_width - 2 * padding_x - label_bar_gap
+        bar_height = 18
+
+        # 작은 글자용 font 사용
+        font = self.emotion_font
+
+        # 제목
+        self.screen.blit(
+            font.render(f"{npc.name} 감정 상태", True, (255, 255, 255)),
+            (hud_rect.x + padding_x, hud_rect.y + padding_y)
+        )
+
+        # 감정 표시
+        for i, (emotion_name, value) in enumerate(top_emotions):
+            y = hud_rect.y + padding_y + title_gap + i * spacing_y
+
+            text_surface = font.render(f"{emotion_name} ({value:.1f})", True, (255, 255, 255))
+            self.screen.blit(text_surface, (hud_rect.x + padding_x, y))
+
+            # bar background
+            bar_bg_rect = Rect(hud_rect.x + padding_x + label_bar_gap, y + 3, bar_width_max, bar_height)
+            pygame.draw.rect(self.screen, (50, 50, 50), bar_bg_rect)
+
+            # bar fill
+            color = self.emotion_colors.get(emotion_name, (100, 200, 255))
+            # color 값이 유효한 튜플인지 확인
+            if not (isinstance(color, tuple) and len(color) == 3 and all(isinstance(c, int) for c in color)):
+                print(f"[경고] 잘못된 color 값 감지: {emotion_name} → {color}, 디폴트 색상 사용")
+                color = (100, 200, 255)
+
+            bar_fill_width = int((value / 10) * bar_width_max)
+            bar_fill_rect = Rect(hud_rect.x + padding_x + label_bar_gap, y + 3, bar_fill_width, bar_height)
+            pygame.draw.rect(self.screen, color, bar_fill_rect)
+
     def update(self):
         self.frame_count += 1
 
@@ -61,4 +143,16 @@ class GameEngine:
             print(f"\n👤 {npc.name} ({npc.job})")
             print(f" - 위치: ({tile_x}, {tile_y})")
             print(f" - 대화: {npc.generate_dialogue()}")
-            print(f" - 기억: {npc.recall()}")
+            print(f" - 기억: {npc.recall()}")
+            self.set_selected_npc(npc)
+
+    def set_selected_npc(self, npc):
+        self.selected_npc = npc
+        print(f"[HUD] {npc.name} 선택됨 → 감정 상태 표시 시작")
+
+    def clear_selected_npc(self):
+        """
+        ESC 키로 Selected_npc 해제 메서드
+        """
+        print("[HUD] 감정 상태 표시 해제됨")
+        self.selected_npc = None

npc_social_network/maps/villages/town_hall.py

@@ -1,51 +1,57 @@
 # portfolio/npc_social_network/maps/villages/town_hall.py
-# 메인 실행 파일
 
-
-import pygame
-import sys
-from npc_social_network.maps.manager.image_loader import ImageLoader
-from npc_social_network.maps.manager.village_map import VillageMap
-from npc_social_network.maps.engine.game_engine import GameEngine
-
-# 초기 설정
-pygame.init()
-
-tile_size = 32
-width, height = 20, 20
-screen = pygame.display.set_mode((width * tile_size, height * tile_size))
-pygame.display.set_caption("town_hall")
-
-# 캐릭터가 움직이는 일정 시간
-clock = pygame.time.Clock()
-font = pygame.font.SysFont("malgungothic", 20) # 한글 출력용
-
-# 로딩
-image_loader = ImageLoader(tile_size)
-village = VillageMap(image_loader, tile_size, width, height)
-village.generate()
-
-
-# npc 리스트 추출
-npcs = village.npc_manager.npcs
-
-# 게임 엔진 초기 세팅
-engine = GameEngine(screen, font, village, ["아침", "점심", "저녁", "밤"], npcs)
-
-# pygame 루프
-runnig = True
-while runnig:
-    screen.fill((255, 255, 255))
-    engine.draw()
-    pygame.display.update()
-    clock.tick(10) # 초당 10프레임
-    engine.update()
-
-    for event in pygame.event.get():
-        if event.type == pygame.QUIT:
-            runnig = False
-        elif event.type == pygame.MOUSEBUTTONDOWN:
-            engine.handle_click(pygame.mouse.get_pos())
-
-pygame.quit()
-sys.exit()
+def run_town_hall_simulation():
+    # 메인 실행 파일
+    import pygame
+    import sys
+    from npc_social_network.maps.manager.image_loader import ImageLoader
+    from npc_social_network.maps.manager.village_map import VillageMap
+    from npc_social_network.maps.engine.game_engine import GameEngine
+
+    # 초기 설정
+    pygame.init()
+
+    tile_size = 32
+    width, height = 20, 20
+    screen = pygame.display.set_mode((width * tile_size, height * tile_size))
+    pygame.display.set_caption("town_hall")
+
+    # 캐릭터가 움직이는 일정 시간
+    clock = pygame.time.Clock()
+    font = pygame.font.SysFont("malgungothic", 20) # 한글 출력용
+
+    # 로딩
+    image_loader = ImageLoader(tile_size)
+    village = VillageMap(image_loader, tile_size, width, height)
+    village.generate()
+
+
+    # npc 리스트 추출
+    npcs = village.npc_manager.npcs
+
+    # 게임 엔진 초기 세팅
+    engine = GameEngine(screen, font, village, ["아침", "점심", "저녁", "밤"], npcs)
+
+    # pygame 루프
+    runnig = True
+    while runnig:
+        screen.fill((255, 255, 255))
+        engine.draw()
+        pygame.display.update()
+        clock.tick(10) # 초당 10프레임
+        engine.update()
+
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                runnig = False
+            elif event.type == pygame.MOUSEBUTTONDOWN:
+                engine.handle_click(pygame.mouse.get_pos())
+            elif event.type == pygame.KEYDOWN:
+                if event.key == pygame.K_ESCAPE:
+                    engine.clear_selected_npc()     # ESC로 선택 해제
+
+    pygame.quit()
+    sys.exit()
+
+if __name__ == "__main__":
+    run_town_hall_simulation()

npc_social_network/npc/npc_base.py

@@ -4,6 +4,7 @@ from .npc_emotion import EmotionManager
 from .npc_behavior import BehaviorManager
 from .emotion_config import EMOTION_CATEGORY_MAP, EMOTION_DECAY_RATE, PERSONALITY_TEMPLATE
 from .npc_relationship import RelationshipManager
+import random
 
 # 관계에 긍/부정 영향으로 해석할 감정 매핑 정의 (논문 설계 기반 적용)
 POSITIVE_RELATION_EMOTIONS = [
@@ -45,6 +46,33 @@ class NPC:
         self.behavior = BehaviorManager()
         # 내부: float 감정 수치 관리용 버퍼
         self._emotion_buffer = self.emotion.get_buffer()
+        # baseline 감정 값 부여
+        baseline_buffer = {}
+        for emo in self._emotion_buffer:
+            base_min = 0.05
+            base_max = 0.3
+
+            category = EMOTION_CATEGORY_MAP.get(emo, None)
+
+            if category == "core":
+                bias = self.personality.get("affect_bias", 1.0)
+            elif category == "social":
+                bias = self.personality.get("social_bias", 1.0)
+            elif category == "cognitive":
+                bias = self.personality.get("cognitive_bias", 1.0)
+            elif category == "complex":
+                bias = self.personality.get("complex_bias", 1.0)
+            else:
+                bias = self.personality.get("sensitive", 1.0)
+
+            baseline_value = base_min + (base_max - base_min) * bias
+            baseline_value *= self.personality.get("sensitive", 1.0)    # 전체 민감도 적용
+
+            baseline_buffer[emo] = round(baseline_value, 3)
+            self._emotion_buffer[emo] = baseline_buffer[emo]    # emotion_buffer 초기 적용
+
+        # baseline buffer를 EmotionManager에 등록
+        self.emotion.set_baseline(baseline_buffer)
 
         # 확장 가능 속성
         # self.emotion = "neutral"
@@ -76,10 +104,19 @@ class NPC:
         behavior_output, behavior_trace = self.behavior.perform_sequence(
             self.name, self.job, emotion_buffer=self._emotion_buffer, return_trace = True
         )
-        # 강한 감정 기반 행동일수록 Memory에 더 강하게 기록됨
-        dominant_score = max([score for _, score in self.behavior.decide_layered_sequence(
-            self.behavior.get_layer_dominant_emotions(self._emotion_buffer))]) if self._emotion_buffer else 5
+        
+        # 단계별 dominant emotion sequence 계산
+        dominant_emotions = self.behavior.get_layer_dominant_emotions(self._emotion_buffer)
+        dominant_sequence = self.behavior.decide_layered_sequence(dominant_emotions)
+
+        # dominant_sequence가 empty인 경우 방어적 처리
+        if dominant_sequence:
+            dominant_score = max([score for _, score in dominant_sequence])
+        else:
+            # dominant_sequence가 비었으면 기본값 사용 (예: 감정 없음 상태)
+            dominant_score = 5
 
+        # 중요도 제한
         importance = min(int(dominant_score), 10)   # importance 1~10제한
 
         # Behavior Trace를 Memory에 기록
@@ -90,6 +127,13 @@ class NPC:
             behavior_trace = behavior_trace
         )
         self.memory_store.add_memory(memory_entry)
+
+        # 행동에 해당하는 감정에 update_emotion 호출 (정확하게 반영)
+        for action, score in behavior_trace:
+            emotion_name = self.behavior.action_to_emotion.get(action)
+            if emotion_name:
+                self.update_emotion(emotion_name, strength=score)
+
         return behavior_output
     
     def remember(self, content: str, importance: int = 5, emotion: str = None, strength: float = 1.0):
@@ -201,3 +245,8 @@ class NPC:
         # 관계 갱신
         self.relationships.update_relationship(other_npc_name, base_influence)
 
+    def get_top_emotions(self, top_n=6):
+        """
+        현재 NPC 감정 상태 상위 N개 반환
+        """
+        return self.emotion.get_top_emotions(top_n=top_n)

npc_social_network/npc/npc_behavior.py

@@ -53,6 +53,9 @@ class BehaviorManager:
             "groundedness": "steady_posture",       # 안정감
         }
 
+        # 행동에 따른 감정 (역매핑)
+        self.action_to_emotion = {v: k for k, v in self.mapping.items()}
+
         # Layer 우선 순위 / weight 설정 (튜닝 가능)
         self.layer_priority = ["core", "social", "cognitive", "complex"]
         self.layer_weights = {
@@ -105,7 +108,7 @@ class BehaviorManager:
         if not sequence:
             behavior_output = f"{name}은(는) 특별한 행동을 하지 않습니다."
             if return_trace:
-                return behavior_output, "특별한 행동 없음"
+                return behavior_output, []
             else:
                 return behavior_output
         
@@ -158,11 +161,11 @@ class BehaviorManager:
 
         # 행동 문장 생성
         lines = []
-        trace_lines = []
+        trace_pairs = []
         for action, score in sequence:
             msg = behavior_lines.get(action, f"{action} 행동을 합니다.")
             lines.append(f"({round(score,1)}) {msg}")
-            trace_lines.append(f"{action}({round(score, 1)})")
+            trace_pairs.append((action, round(score, 1)))
 
         # 직업 라인 추가
         job_line = {
@@ -171,10 +174,9 @@ class BehaviorManager:
         }.get(job, "오늘도 바쁜 하루네요.")
 
         behavior_output = f"{name} 행동 시퀀스: \n" + "\n".join(lines) + f"\n{job_line}"
-        behavior_trace = ", ".join(trace_lines)
 
         # 최종 출력 문장
         if return_trace:
-            return behavior_output, behavior_trace
+            return behavior_output, trace_pairs
         else:
             return behavior_output

npc_social_network/npc/npc_emotion.py

@@ -3,9 +3,11 @@
 from .emotion_config import EMOTION_LIST, EMOTION_DECAY_RATE, EMOTION_CATEGORY_MAP
 
 class EmotionManager:
-    def __init__(self, decay_rate, personality = None):
+    def __init__(self, decay_rate, personality=None, initial_baseline=None):
         # 내부: float 감정 수치 관리용 버퍼
         self._emotion_buffer = {emo: 0.0 for emo in EMOTION_LIST}
+        # baseline 버퍼 추가
+        self._baseline_buffer = initial_baseline or {emo: 0.0 for emo in EMOTION_LIST}
         # 감정별 회복 속도 (ms 단위가 아닌 단위당 감소 속도)
         self.decay_rate = decay_rate
         self.personality = personality or {}
@@ -80,4 +82,26 @@ class EmotionManager:
         """
         emotion_buffer 복사본 반환
         """
-        return self._emotion_buffer.copy()
+        return self._emotion_buffer.copy()
+
+    def get_current_emotions_state(self):
+        """
+        현재 감정 상태 반환 (dict)
+        """
+        return self._emotion_buffer
+    
+    def get_top_emotions(self, top_n=5):
+        """
+        현재 감정 버퍼에서 상위 N개 감정 반환
+        :return: List of (emotion_name, value) tuples
+        """
+        # 정렬 후 상위 N개 반환
+        sorted_emotions = sorted(self._emotion_buffer.items(), key=lambda x: x[1], reverse=True)
+        top_emotions = sorted_emotions[:top_n]
+        return top_emotions
+    
+    def set_baseline(self, baseline_buffer):
+        """
+        baseline buffer를 외부에서 설정 (npc_base.py 초기화 시 사용)
+        """
+        self._baseline_buffer = baseline_buffer.copy()

templates/main.html

@@ -1,3 +1,4 @@
+<!--portfolio/templates/main.html-->
 <!DOCTYPE html>
 <html lang="en">
 <head>
@@ -8,5 +9,9 @@
 <body>
     <h1>Welcome to the Portfolio</h1>
     <a href="{{ url_for('npc_social.home') }}">Go to Project 1</a>
+    <br><br>
+    <form action="{{ url_for('run_npc_simulation') }}" method="get">
+        <button type = "submit"> Run NPC Simulation (pygame) </button>
+    </form>
 </body>
 </html>

templates/npc_simulation_started.html

@@ -0,0 +1,14 @@
+<!--portfolio/templates/npc_simulation_started.html-->
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>NPC Simulation Running</title>
+</head>
+<body>
+    <h1>NPC Simulation is now running!</h1>
+    <p>You can interact with the Pygame window that has opened.</p>
+    <a href="{{ url_for('index') }}">Back to Main Page</a>
+</body>
+</html>

test.ipynb

@@ -47,7 +47,8 @@
    "source": [
     "# 나중에 할 내용\n",
     "# 1. ai 모델 학습을 위해 어떤 내용이 필요한지\n",
-    "# 2. 학습한 모델을 저장, 활용할 수 있는지"
+    "# 2. 학습한 모델을 저장, 활용할 수 있는지\n",
+    "# 3. NPC 초기 기억 주입"
    ]
   },
   {