Spaces:

rorshi
/

autonomous-ai-agent-sim

Sleeping

App Files Files Community

humanda5 commited on 29 days ago

Commit

00b7ce9

1 Parent(s): e01cc89

api key 변경 ui 추가

Browse files

Files changed (3) hide show

npc_social_network/data/vectorstores/alice.faiss +0 -0
npc_social_network/routes/npc_route.py +23 -1
templates/dashboard.html +51 -3

npc_social_network/data/vectorstores/alice.faiss CHANGED Viewed

Binary files a/npc_social_network/data/vectorstores/alice.faiss and b/npc_social_network/data/vectorstores/alice.faiss differ

npc_social_network/routes/npc_route.py CHANGED Viewed

@@ -224,4 +224,26 @@ def toggle_player():
         is_active = simulation_core.npc_manager.player_is_active
         status_text = "활성화" if is_active else "비활성화"
         simulation_core.add_log(f"플레이어 상태가 '{status_text}'로 변경되었습니다.")
-    return jsonify({"success": True, "player_is_active": is_active})

         is_active = simulation_core.npc_manager.player_is_active
         status_text = "활성화" if is_active else "비활성화"
         simulation_core.add_log(f"플레이어 상태가 '{status_text}'로 변경되었습니다.")
+    return jsonify({"success": True, "player_is_active": is_active})
+@npc_bp.route("/api/set_llm_config", methods=['POST'])
+def set_llm_config():
+    """UI로부터 받은 모델과 API 키로 LLM을 실시간으로 재설정하는 API"""
+    from ..models.gemini_setup import initialize_model
+    data = request.json
+    model_name = data.get("model_name")
+    api_key = data.get("api_key")
+    if not all([model_name, api_key]):
+        return jsonify({"success": False, "error": "모델과 API 키를 모두 입력해주세요."}), 400
+    # 새로운 설정으로 모델을 초기화 시도
+    new_model = initialize_model(model_name, api_key)
+    if new_model:
+        # 성공 시, simulation_core의 전역 모델 객체를 새로운 모델로 교체
+        simulation_core.active_llm_model = new_model
+        return jsonify({"success": True})
+    else:
+        return jsonify({"success": False, "error": "API 키가 유효하지 않거나 모델 초기화에 실패했습니다."}), 400

templates/dashboard.html CHANGED Viewed

@@ -63,6 +63,7 @@
 </head>
 <body>
     <div id="api-modal-overlay">
         <div id="api-modal-content">
             <h2>시뮬레이션 시작</h2>
             <p>사용할 LLM 모델과 API 키를 입력해주세요.</p>
@@ -71,7 +72,7 @@
                 <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>
@@ -93,7 +94,18 @@
                     <button id="tick-btn">Next Tick</button>
                     <button id="player-toggle-btn">플레이어 상태: 비활성화</button>
                     <button id="fit-btn">관계도 보기</button>
                 </div>
                 <!-- 개입 기능 선택 -->
                 <div id="intervention-area" style="display: flex; gap: 15px; align-items: flex-start;">
                     <div id="intervention-tabs">
@@ -152,7 +164,12 @@
             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');
@@ -409,6 +426,36 @@
                 });
                 await updateWorld();
             });
             // 모든 이벤트 리스너 복구
             playPauseBtn.addEventListener('click', async () => {
@@ -460,7 +507,8 @@
             });
             // 초기 로드
-            updateWorld();
         });
     </script>
 </body>

 </head>
 <body>
     <div id="api-modal-overlay">
+        <!-- API 키 입력_1 -->
         <div id="api-modal-content">
             <h2>시뮬레이션 시작</h2>
             <p>사용할 LLM 모델과 API 키를 입력해주세요.</p>
                 <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>
                     <button id="tick-btn">Next Tick</button>
                     <button id="player-toggle-btn">플레이어 상태: 비활성화</button>
                     <button id="fit-btn">관계도 보기</button>
+                    <button id="api-settings-toggle-btn" style="margin-left: auto;">API 설정</button>
                 </div>
+                <!-- API 키 입력_2 -->
+                <div id="ingame-api-panel" class="control-panel" style="display: none; flex-direction: row; gap: 10px; background-color: #e9ecef; padding: 10px; border-radius: 8px;">
+                    <select id="ingame-llm-model-select" style="flex: 1;">
+                        <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="ingame-api-key-input" placeholder="API Key를 여기에 입력하세요" style="flex: 2;">
+                    <button id="ingame-api-key-save-btn" style="flex: 1;">API 변경</button>
+                </div>
                 <!-- 개입 기능 선택 -->
                 <div id="intervention-area" style="display: flex; gap: 15px; align-items: flex-start;">
                     <div id="intervention-tabs">
             const apiKeyInput = document.getElementById('api-key-input');
             const startSimBtn = document.getElementById('start-simulation-btn');
             const apiModalMsg = document.getElementById('api-modal-message');
+            // API 입력 2
+            const apiSettingsToggleBtn = document.getElementById('api-settings-toggle-btn');
+            const ingameApiPanel = document.getElementById('ingame-api-panel');
+            const ingameLlmModelSelect = document.getElementById('ingame-llm-model-select');
+            const ingameApiKeyInput = document.getElementById('ingame-api-key-input');
+            const ingameApiKeySaveBtn = document.getElementById('ingame-api-key-save-btn');
             // UI 요소 가져오기
             const networkContainer = document.getElementById('network');
             const logContainer = document.getElementById('log-container');
                 });
                 await updateWorld();
             });
+            // 인게임 API 설정 패널 토글 이벤트
+            apiSettingsToggleBtn.addEventListener('click', () => {
+                const isHidden = ingameApiPanel.style.display === 'none';
+                ingameApiPanel.style.display = isHidden ? 'flex' : 'none';
+            });
+            // 인게임 'API 변경' 버튼 이벤트 리스너
+            ingameApiKeySaveBtn.addEventListener('click', async () => {
+                const model_name = ingameLlmModelSelect.value;
+                const api_key = ingameApiKeyInput.value;
+                if (!api_key) {
+                    return alert('새로운 API 키를 입력해주세요.');
+                }
+                ingameApiKeySaveBtn.textContent = '변경 중...';
+                const response = await fetch('/api/set_llm_config', { // 2단계에서 만들 API
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({ model_name, api_key })
+                });
+                const data = await response.json();
+                if (data.success) {
+                    alert(`API 설정이 변경되었습니다. 이제부터 [${model_name}] 모델을 사용합니다.`);
+                    ingameApiKeyInput.value = ''; // 입력창 비우기
+                } else {
+                    alert(`API 변경 실패: ${data.error}`);
+                }
+                ingameApiKeySaveBtn.textContent = 'API 변경';
+            });
             // 모든 이벤트 리스너 복구
             playPauseBtn.addEventListener('click', async () => {
             });
             // 초기 로드
+            await updateWorld();
+            setInterval(updateWorld, 5000);
         });
     </script>
 </body>