배포 및 최종 수정 완료 버전

app.py

@@ -22,6 +22,8 @@ class LogFilter(logging.Filter):
         # /api/world_state 경로에 대한 로그는 기록하지 않음
         if "/api/world_state" in record.getMessage():
             return False
+        if "/api/npc_details" in record.getMessage():
+            return False
         return True
 
 def create_app():
@@ -48,23 +50,11 @@ def create_app():
 if __name__ == '__main__':
     # 1. flask 앱 실행
     app = create_app()
-    # 2. NPC 시뮬레이션 초기화 및 백그라운드 스레드 시작
-    try:
-        from npc_social_network import simulation_core
-        simulation_core.initialize_simulation()
-        print("✅ 시뮬레이션이 성공적으로 초기화되었습니다.")
-    except Exception as e:
-        import traceback
-        print("="*60)
-        print("❌ CRITICAL ERROR: 시뮬레이션 초기화 중 에러가 발생했습니다!")
-        print(f"   에러 메시지: {e}")
-        print("="*60)
-        traceback.print_exc()
-        print("="*60)
-    
-    # 3. Flask 웹 서버 실행
+
+     # Flask 앱이 실행되기 전에 로그 필터를 적용합니다.
     log = logging.getLogger('werkzeug')
     log.addFilter(LogFilter())
+
     # use_reloader=False는 디버그 모드에서 앱이 두 번 실행되는 것을 방지하여,
     # 시뮬레이션 스레드가 두 번 시작되지 않도록 합니다.
     app.run(debug=True, use_reloader=False, port=5000)

npc_social_network/manager/conversation_manager.py

@@ -97,6 +97,9 @@ class ConversationManager:
 
                 simulation_core.add_log(f"[{initiator.korean_name}, {target.korean_name}] 대화 경험이 기억과 관계, 성격에 반영되었습니다.")
 
+                # 9. 모든 백그라운드 작업 종료 후, 현재까지의 시뮬레이션 전체 상태를 즉시 저장.
+                simulation_core.save_simulation(simulation_core.npc_manager)
+
         except Exception as e:
             print(f"[에러] 백그라운드 기억 저장 중 문제 발생: {e}")
             import traceback
@@ -137,10 +140,6 @@ class ConversationManager:
         )
         background_thread.start()
 
-        # 3. 대화가 종료된 직후, 현재까지의 시뮬레이션 전체 상태를 즉시 저장.
-        with simulation_core.simulation_lock:
-            simulation_core.save_simulation(simulation_core.npc_manager)
-
 
     def next_turn(self):
         """대화의 다음 턴을 진행, 생성된 대사 처리"""

npc_social_network/models/gemini_setup.py

@@ -4,10 +4,25 @@ from dotenv import load_dotenv
 import os
 
 
-def load_gemini():
-    load_dotenv() # .env 파일에서 환경 변수 로드
+# def load_gemini():
+#     load_dotenv() # .env 파일에서 환경 변수 로드
 
-    genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
-    model = genai.GenerativeModel("gemini-2.0-flash")  # Gemini Flash 사용
+#     genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+#     model = genai.GenerativeModel("gemini-2.0-flash")  # Gemini Flash 사용
 
-    return model
+#     return model
+
+def initialize_model(model_name: str, api_key: str):
+    """ 주어진 모델과 API 키로 Gemini 모델을 동적으로 초기화"""
+    try:
+        # API 키를 설정
+        genai.configure(api_key=api_key)
+        # 지정된 모델을 생성
+        model = genai.GenerativeModel(model_name)
+        # 간단한 테스트 호출로 API 키 유효성 검증
+        model.generate_content("hello", generation_config={"max_output_tokens": 1})
+        print(f"✅ Gemini 모델 '{model_name}'이(가) 성공적으로 초기화되었습니다.")
+        return model
+    except Exception as e:
+        print(f"❌ Gemini 모델 초기화 실패: {e}")
+        return None

npc_social_network/models/llm_helper.py

@@ -1,20 +1,19 @@
 # npc_social_network/models/llm_helper.py
 
 from npc_social_network.npc.emotion_config import EMOTION_LIST
-from npc_social_network.models.gemini_setup import load_gemini
 from npc_social_network.npc.npc_memory import Memory
+from .. import simulation_core
 from typing import List
 import json
 import re
+import time
+from google.api_core import exceptions as google_exceptions
 from typing import TYPE_CHECKING, List
 
 if TYPE_CHECKING:
     from ..npc.npc_base import NPC
     from ..manager.conversation_manager import Conversation
 
-# Gemini 모델 초기화
-gemini_model = load_gemini()
-
 def query_llm_for_emotion(user_input):
     """
     LLM을 통해 플레이어 입력에서 감정 추출
@@ -28,7 +27,7 @@ def query_llm_for_emotion(user_input):
     반드시 감정 이름을 한 개만 출력하세요. (예: joy)
     """
     try:
-        response = gemini_model.generate_content(prompt)
+        response = simulation_core.active_llm_model.generate_content(prompt)
         return response.text.strip()
     except Exception as e:
         return f"[LLM Error] {str(e)}"
@@ -36,11 +35,57 @@ def query_llm_for_emotion(user_input):
 def query_llm_with_prompt(prompt: str) -> str:
     """
     prompt 문자열을 받아 LLM 호출
+     - api 사용량 오류를 감지하고 자동 대처
     """
+    if not simulation_core.active_llm_model:
+        return "[LLM Error] API 모델이 정상적으로 설정되지 않았습니다."
+    
     try:
-        response = gemini_model.generate_content(prompt)
+        # 1. 첫 번째 API 호출 시도
+        response = simulation_core.active_llm_model.generate_content(prompt)
         return response.text.strip()
+    except google_exceptions.ResourceExhausted as e:
+        error_message = str(e)
+
+        # 2. 오류 메시지를 분석하여 원인 파악
+        if "PerMinute" in error_message:
+            # -- 분당 사용량 초과 --
+            simulation_core.add_log("[API 경고] 분당 사용량을 초과했습니다. 60초 후 자동으로 재시도 합니다.")
+            time.sleep(60)
+
+            try:
+                # 2-1. 두 번째 API 호출 시도
+                simulation_core.add_log("[API 정보] API 호출을 재시도합니다.")
+                response = simulation_core.active_llm_model.generate_content(prompt)
+                return response.text.strip()
+            except Exception as retry_e:
+                # 재시도마저 실패할 경우, 최종 에러 반환
+                simulation_core.add_log("========= [API Error] =========")
+
+                # 시물레이션을 안전하게 '일시정지' 상태로 변경
+                with simulation_core.simulation_lock:
+                    simulation_core.simulation_paused = True
+
+                return f"[API 에러] 재시도에 실패했습니다: {retry_e}"
+            
+        elif "PerDay" in error_message:
+            # -- 일일 사용량 초과 시 --
+            simulation_core.add_log("[API Error] 일일 사용량을 모두 소진했습니다.")
+            simulation_core.add_log("다른 API 키를 사용하거나, 내일 다시 시도해주세요.")
+
+            # 시뮬레이션을 안전하게 '일시정지' 상태로 변경
+            with simulation_core.simulation_lock:
+                simulation_core.simulation_paused = True
+            return "[LLM Error] Daily quota exceeded"
+        
+        else:
+            # 그 외 다른 ResourceExhausted 오류
+            with simulation_core.simulation_lock:
+                simulation_core.simulation_paused = True
+            return f"[LLM Error] {error_message}"
+        
     except Exception as e:
+        # 그 외 모든 종류의 오류
         return f"[LLM Error] {str(e)}"
     
 def summarize_text(text: str) -> str:
@@ -163,6 +208,8 @@ def _query_llm_for_json_robustly(prompt: str) -> dict | list:
         else:
             text_to_parse = response_text   # 최후의 경우 원본 텍스트 사용
 
+    # 불리언 값의 대소문자를 표준에 맞게 수정 (True -> true)
+    text_to_parse = text_to_parse.replace(": True", ": true").replace(": False", ": false")
     # 딕셔너리나 리스트의 마지막 요소 뒤에 붙은 꼬리표 쉼표 제거
     text_to_parse = re.sub(r',\s*(\}|\])', r'\1', text_to_parse)
     
@@ -183,6 +230,9 @@ Do not include any other text, explanations, or markdown.
 # Corrected JSON:
 """
         corrected_response = query_llm_with_prompt(correction_prompt)
+        # 자가 교정된 텍스트도 한번 더 정리
+        corrected_response = corrected_response.replace(": True", ": true").replace(": False", ": false")
+        corrected_response = re.sub(r',\s*(\}|\])', r'\1', corrected_response)
         try:
             return json.loads(corrected_response)   # 교정된 응답 파싱 시도
         except json.JSONDecodeError:
@@ -368,11 +418,11 @@ def evaluate_goal_achievement(initiator_name: str, goal: str, conversation_trans
 결과는 반드시 아래 JSON 형식이어야 합니다.
 
 # Evaluation (JSON format only)
-{
+{{
     "goal_achieved": True,
     "reason": "목표 달성 또는 실패에 대한 간결한 이유 (예: '상대방을 위로하고 지지하며 긍정적인 반응을 이끌어냈다.)",
     "initiator_emotion": "이 결과를 얻은 후 대화 시작자가 느낄 감정 (아래 'Emotion_List에서 찾아서 한 단어로 표현)
-}
+}}
 
 # Emotion_List
 {EMOTION_LIST}

npc_social_network/routes/npc_route.py

@@ -1,6 +1,7 @@
 # portfolio/npc_social_network/routes/npc_route.py
 from flask import Blueprint, render_template, request, jsonify, url_for
 from .. import simulation_core
+from ..models.gemini_setup import initialize_model
 import os
 
 
@@ -8,15 +9,43 @@ import os
 npc_bp = Blueprint(
     "npc_social",
     __name__,
-    static_folder="../static"
+    static_folder="../static",
+    static_url_path="/npc_social_network"
 )
 
 @npc_bp.route("/")
 def dashboard():
     return render_template("dashboard.html")
 
+# 시뮬레이션을 시작하는 API
+@npc_bp.route("/api/initialize_simulation", methods=['POST'])
+def api_initialize_simulation():
+    """API 키를 받아 시뮬레이션을 초기화하고 시작하는 API"""
+    from ..models.gemini_setup import initialize_model
+    data = request.json
+    model_name = data.get("model_name", "gemini-2.0-flash")
+    api_key = data.get("api_key")
+
+    if not api_key:
+        return jsonify({"success": False, "error": "API 키를 입력해주세요."}), 400
+    
+    # 1. 모델 초기화 시도
+    new_model = initialize_model(model_name, api_key)
+    
+    if new_model:
+        # 2. 성공시, 전역 모델을 설정하고 시뮬레이션을 시작
+        simulation_core.active_llm_model = new_model
+        if not simulation_core.simulation_initialized:
+            simulation_core.initialize_simulation()
+            simulation_core.start_simulation_loop()
+        return jsonify({"success": True, "message": f"'{model_name}' 모델로 시뮬레이션을 시작합니다."})
+    else:
+        return jsonify({"success": False, "error": "API 키가 유효하지 않거나 모델 초기화에 실패했습니다."}), 400
+
 @npc_bp.route("/api/world_state", methods=['GET'])
 def get_world_state():
+    if not simulation_core.simulation_initialized:
+        return jsonify({"status": "needs_initialization", "log": simulation_core.event_log})
     with simulation_core.simulation_lock:
         if not simulation_core.npc_manager:
             return jsonify({"error": "Simulation not started"}), 500
@@ -26,7 +55,7 @@ def get_world_state():
 
     nodes = []
     for npc in all_npcs:
-        # 각 NPC의 영어 ID를 기반으로 이미지 파일명을 생성합니다. (예: elin -> npc_elin.png)
+        # 각 NPC의 영어 ID를 기반으로 이미지 파일명을 생성합니다. (예: elin -> elin.png)
         image_filename = f"images/npc/{npc.name}.png"
         # if not os.path.exists(image_filename):
         #     image_filename = f"images/npc/npc.png"
@@ -78,8 +107,6 @@ def get_world_state():
 
     return jsonify({ "nodes": nodes, "edges": edges, "log": current_log, "paused": is_paused, "waiting_for_player": waiting_for_player, "player_conversation": player_conversation_info })
 
-# 다른 API 엔드포인트들은 이 테스트 동안에는 호출되지 않으므로 그대로 두어도 괜찮습니다.
-
 @npc_bp.route("/api/toggle_simulation", methods=['POST'])
 def toggle_simulation():
     with simulation_core.simulation_lock:

npc_social_network/simulation_core.py

@@ -15,6 +15,8 @@ from typing import List, Optional
 # 1. 시뮬레이션 상태 (전역 변수)
 # - 이 변수들이 시뮬레이션 세계의 모든 정보를 담습니다.
 #------------------------------------------
+simulation_initialized = False # 초기화 여부 플래그
+active_llm_model = None # 현재 활성화된 LLM 모델
 npc_manager: NPCManager = None
 simulation_paused = True
 event_log:List[str] = []
@@ -120,16 +122,23 @@ def simulation_loop():
         time.sleep(5)   # 5초에 한 번씩 틱 발생 (이것보다 줄이면 Gemini API 사용량 초과 발생 가능)
 
 def initialize_simulation():
-    """서버 시작 시 시뮬레이션을 초기화합니다."""
-    global npc_manager, conversation_manager
+    """시뮬레이션의 준비"""
+    global npc_manager, conversation_manager, simulation_initialized, active_llm_model
+
+    # 이미 초기화되었다면 다시 실행 x
+    if simulation_initialized:
+        return
+
     npc_manager = load_simulation()
     conversation_manager = ConversationManager()
     if npc_manager is None:
         npc_manager = setup_initial_scenario()
         save_simulation(npc_manager)
 
+    simulation_initialized = True
     add_log("시뮬레이션이 성공적으로 초기화되었습니다.")
 
+def start_simulation_loop():
     # 백그라운드 스레드에서 시뮬레이션 루프 시작
     simulation_thread = threading.Thread(target=simulation_loop, daemon=True)
     simulation_thread.start()

templates/dashboard.html

@@ -20,6 +20,14 @@
         #player-input-text { flex: 1; padding: 8px 12px; border-radius: 6px; border: 1px solid #ddd; }
         #player-send-btn { padding: 8px 15px; border-radius: 6px; border: none; cursor: pointer; color: white; background-color: var(--primary-color); }
 
+        /* API 설정 */
+        #api-modal-overlay { position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.7); display: flex; justify-content: center; align-items: center; z-index: 1000; }
+        #api-modal-content { background: white; padding: 30px; border-radius: 12px; box-shadow: 0 5px 15px rgba(0,0,0,0.3); width: 400px; text-align: center; }
+        #api-modal-content h2 { margin-top: 0; }
+        #api-modal-content select, #api-modal-content input, #api-modal-content button { width: 100%; padding: 12px; margin-top: 10px; box-sizing: border-box; border-radius: 6px; border: 1px solid #ccc; }
+        #api-modal-content button { background: var(--primary-color); color: white; cursor: pointer; font-weight: bold; }
+        .hidden { display: none !important; }
+
         /* 전체 컨트롤 패널 (가장 바깥쪽) */
         #controls { flex-shrink: 0; padding: 12px; display: flex; flex-direction: column; gap: 15px; background-color: #3e3f41; border-radius: 0 0 12px 12px;}
         
@@ -54,6 +62,20 @@
     </style>
 </head>
 <body>
+    <div id="api-modal-overlay">
+        <div id="api-modal-content">
+            <h2>시뮬레이션 시작</h2>
+            <p>사용할 LLM 모델과 API 키를 입력해주세요.</p>
+            <select id="llm-model-select">
+                <option value="gemini-2.0-flash">[추천] Gemini 2.0 Flash</option>
+                <option value="gemini-2.5-flash">Gemini 2.5 Flash</option>
+                <option value="gemini-2.5-pro">Gemini 2.5 Pro</option>
+            </select>
+            <input type="password" id="api-key-input" placeholder="API Key를 여기에 붙여넣으세요">
+            <button id="start-simulation-btn">시작하기</button>
+            <p id="api-modal-message" style="color: red; margin-top: 10px;"></p>
+        </div>
+    </div>
     <div id="main-container">
         <div id="network-panel">
             <div id="network"></div>
@@ -69,7 +91,7 @@
                 <div id="main-controls" style="display: flex; gap: 10px; justify-content: flex-start;">
                     <button id="play-pause-btn">Play / Pause</button>
                     <button id="tick-btn">Next Tick</button>
-                    <button id="player-toggle-btn">플레이어 비활성화</button>
+                    <button id="player-toggle-btn">플레이어 상태: 비활성화</button>
                     <button id="fit-btn">관계도 보기</button>
                 </div>
                 <!-- 개입 기능 선택 -->
@@ -124,6 +146,13 @@
 
     <script>
         document.addEventListener('DOMContentLoaded', async function () {
+            // API 입력
+            const apiModalOverlay = document.getElementById('api-modal-overlay');
+            const llmModelSelect = document.getElementById('llm-model-select');
+            const apiKeyInput = document.getElementById('api-key-input');
+            const startSimBtn = document.getElementById('start-simulation-btn');
+            const apiModalMsg = document.getElementById('api-modal-message');
+
             // UI 요소 가져오기
             const networkContainer = document.getElementById('network');
             const logContainer = document.getElementById('log-container');
@@ -172,12 +201,46 @@
                 interaction: { hover: true, zoomView: true, dragView: true, minZoom: 0.2, maxZoom: 4.0 }
             };
 
+            // 시작 버튼 이벤트
+            startSimBtn.addEventListener('click', async () => {
+                const model_name = llmModelSelect.value;
+                const api_key = apiKeyInput.value;
+
+                if (!api_key) {
+                    apiModalMsg.textContent = 'API 키를 입력해주세요.';
+                    return;
+                }
+                
+                startSimBtn.textContent = '인증 중...';
+                startSimBtn.disabled = true;
+                apiModalMsg.textContent = '';
+
+                const response = await fetch('/api/initialize_simulation', {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({ model_name, api_key })
+                });
+                const data = await response.json();
+
+                if (data.success) {
+                    apiModalOverlay.classList.add('hidden'); // 모달 숨기기
+                    // 시뮬레이션이 성공적으로 시작되었으므로, 이제부터 월드 상태를 주기적으로 업데이트합니다.
+                    await updateWorld();
+                    if (!autoUpdateInterval) { // 중복 실행 방지
+                        autoUpdateInterval = setInterval(updateWorld, 5000);
+                    }
+                } else {
+                    apiModalMsg.textContent = `오류: ${data.error}`;
+                    startSimBtn.textContent = '시작하기';
+                    startSimBtn.disabled = false;
+                }
+            });
+
             fitBtn.addEventListener('click', () => {
                 if (network) {
                     network.fit();
                 }
             });
-
             // 탭 버튼 클릭 이벤트
             tabEvent.addEventListener('click', () =>{
                 tabEvent.classList.add('active');
@@ -213,7 +276,7 @@
                 if (!npc1_name || !npc2_name || !situation || npc1_name === npc2_name) {
                     return alert ('서로 다른 두 NPC와 상황을 모두 입력해주세요.');
                 }
-                await fetch('/npc_social_network/api/orchestrate_conversation', {
+                await fetch('/api/orchestrate_conversation', {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify({ npc1_name, npc2_name, situation })
@@ -223,8 +286,14 @@
 
             async function updateWorld() {
                 try {
-                    const response = await fetch('/npc_social_network/api/world_state');
+                    const response = await fetch('/api/world_state');
                     const data = await response.json();
+                    // 서버가 아직 준비되지 않았음을 확인하고 대기
+                    if (data.status === 'needs_initialization') {
+                        console.log("시뮬레이션 초기화 대기 중...");
+                        updateLog(data.log); // 초기 로그는 표시
+                        return;
+                    }
                     if (data.error) throw new Error(data.error);
 
                     // 데이터셋을 직접 업데이트하면 vis-network가 알아서 변경사항을 감지하고 다시 그립니다.
@@ -280,7 +349,7 @@
                     const npcName = selectedNodeObject.label; // 영어 ID가 아닌 한글 라벨로 요청
 
                     try {
-                        const response = await fetch(`/npc_social_network/api/npc_details/${npcName}`);
+                        const response = await fetch(`/api/npc_details/${npcName}`);
                         const details = await response.json();
                         let detailsHtml = `<h3>${details.name} (${details.job}, ${details.age}세)</h3>`;
                         detailsHtml += `<p><strong>성격:</strong> ${details.personality_summary}</p>`;
@@ -333,7 +402,7 @@
                 if (!npc1_name || !npc2_name || npc1_name === npc2_name) {
                     return alert('서로 다른 두 명의 NPC를 선택해주세요.');
                 }
-                await fetch('/npc_social_network/api/force_relationship', {
+                await fetch('/api/force_relationship', {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify({ npc1_name, npc2_name, relationship_type })
@@ -343,16 +412,16 @@
             
             // 모든 이벤트 리스너 복구
             playPauseBtn.addEventListener('click', async () => {
-                const response = await fetch('/npc_social_network/api/toggle_simulation', { method: 'POST' });
+                const response = await fetch('/api/toggle_simulation', { method: 'POST' });
                 const data = await response.json();
                 updatePlayPauseButton(data.paused);
             });
-            tickBtn.addEventListener('click', async () => { await fetch('/npc_social_network/api/manual_tick', { method: 'POST' }); await updateWorld(); });
+            tickBtn.addEventListener('click', async () => { await fetch('/api/manual_tick', { method: 'POST' }); await updateWorld(); });
             injectEventBtn.addEventListener('click', async () => {
                 const npc_name = eventNpcSelect.value; // 이제 한글 이름이 전송됩니다.
                 const event_text = eventTextInput.value;
                 if (!npc_name || !event_text) return alert('NPC와 이벤트 내용을 모두 입력해주세요.');
-                await fetch('/npc_social_network/api/inject_event', {
+                await fetch('/api/inject_event', {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify({ npc_name, event_text })
@@ -361,10 +430,10 @@
                 await updateWorld(); 
             });
             playerToggleBtn.addEventListener('click', async () => {
-                const response = await fetch('/npc_social_network/api/toggle_player', {method: 'POST'});
+                const response = await fetch('/api/toggle_player', {method: 'POST'});
                 const data = await response.json();
                 // 버튼 텍스트 업데이트
-                playerToggleBtn.textContent = data.player_is_active ? '플레이어 비활성화' : '플레이어 활성화';
+                playerToggleBtn.textContent = data.player_is_active ? '플레이어 상태: 활성화' : '플레이어 상태: 비활성화';
             });
             // '전송' 버튼 이벤트 리스너
             playerSendBtn.addEventListener('click', async () => {
@@ -372,7 +441,7 @@
                 if (!utterance) return;
 
                 playerSendBtn.disabled = true;
-                await fetch('/npc_social_network/api/player_response', {
+                await fetch('/api/player_response', {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify({ utterance })