5/17 preset setting

BookWorld.py

@@ -184,10 +184,12 @@ class Server():
             if self.mode == "free":
                 self.get_event()
                 self.log(f"--------- Free Mode: Current Event ---------\n{self.event}\n")
+                yield ("system","",f"--------- Current Event ---------\n{self.event}\n", None) 
                 self.event_history.append(self.event)
             elif self.mode == "script":
                 self.get_script()
                 self.log(f"--------- Script Mode: Setted Script ---------\n{self.script}\n")
+                yield ("system","",f"--------- Setted Script ---------\n{self.script}\n", None) 
                 self.event_history.append(self.event)
             if self.mode == "free":
                 for role_code in self.role_codes:
@@ -648,7 +650,6 @@ class Server():
             status = "\n".join([self.role_agents[role_code].status for role_code in self.role_codes])
             script = self.world_agent.generate_script(roles_info_text=roles_info_text,event=self.intervention,history_text=status)
             self.script = script
-        # self.event = self.script
         return self.script
     
     def update_event(self, group: List[str], top_k: int = 1):

app.py

@@ -20,6 +20,9 @@ for key in config:
 static_file_abspath = os.path.abspath(os.path.join(os.path.dirname(__file__), 'frontend'))
 app.mount("/frontend", StaticFiles(directory=static_file_abspath), name="frontend")
 
+# 预设文件目录
+PRESETS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'experiment_presets')
+
 class ConnectionManager:
     def __init__(self):
         self.active_connections: dict[str, WebSocket] = {}  
@@ -137,6 +140,61 @@ async def get_file(full_path: str):
     
     raise HTTPException(status_code=404, detail="File not found")
 
+@app.get("/api/list-presets")
+async def list_presets():
+    try:
+        # 获取所有json文件
+        presets = [f for f in os.listdir(PRESETS_DIR) if f.endswith('.json')]
+        return {"presets": presets}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/load-preset")
+async def load_preset(request: Request):
+    try:
+        data = await request.json()
+        preset_name = data.get('preset')
+        
+        if not preset_name:
+            raise HTTPException(status_code=400, detail="No preset specified")
+            
+        preset_path = os.path.join(PRESETS_DIR, preset_name)
+        print(f"Loading preset from: {preset_path}")
+        
+        if not os.path.exists(preset_path):
+            raise HTTPException(status_code=404, detail=f"Preset not found: {preset_path}")
+            
+        try:
+            # 更新BookWorld实例的预设
+            manager.bw = BookWorld(
+                preset_path=preset_path,
+                world_llm_name=config["world_llm_name"],
+                role_llm_name=config["role_llm_name"],
+                embedding_name=config["embedding_model_name"]
+            )
+            manager.bw.set_generator(
+                rounds=config["rounds"],
+                save_dir=config["save_dir"],
+                if_save=config["if_save"],
+                mode=config["mode"],
+                scene_mode=config["scene_mode"]
+            )
+            
+            # 获取初始数据
+            initial_data = await manager.get_initial_data()
+            
+            return {
+                "success": True,
+                "data": initial_data
+            }
+        except Exception as e:
+            print(f"Error initializing BookWorld: {str(e)}")
+            raise HTTPException(status_code=500, detail=f"Error initializing BookWorld: {str(e)}")
+            
+    except Exception as e:
+        print(f"Error in load_preset: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 @app.websocket("/ws/{client_id}")
 async def websocket_endpoint(websocket: WebSocket, client_id: str):
     await manager.connect(websocket, client_id)

bw_utils.py

@@ -452,6 +452,7 @@ def clean_collection_name(name: str) -> str:
     valid_name = re.sub(r'\.\.+', '-', valid_name)
     valid_name = re.sub(r'^[^a-zA-Z0-9]+', '', valid_name)  # 移除开头非法字符
     valid_name = re.sub(r'[^a-zA-Z0-9]+$', '', valid_name)
+    valid_name = valid_name[:60]
     return valid_name
     
 cache_sign = True

data/roles/example_world/Falko-en/role_info.json

@@ -0,0 +1,18 @@
+{
+    "role_code": "Falko-en",
+    "role_name": "Falko·Weiss",
+    "source": "example",
+    "activity": 1,
+    "profile": "Falko Weiss serves as a doctoral researcher in neurointelligence at Eldridge Corporation's laboratory, where he works alongside Lacia and Trek. With short brown hair, greenish-yellow eyes, and glasses, he presents a typical scholarly appearance. Despite his serious and methodical approach to work, Falko is known for his inherent kindness and unwavering dedication to his colleagues' well-being. In contrast to Trek's revolutionary views and Lacia's initially aggressive stance, Falko consistently emphasizes the importance of thorough safety protocols and ethical guidelines in their research.",
+    "nickname": "",
+    "relation": {
+        "Trek-en": {
+            "relation": ["friend","alumni"],
+            "detail": "Despite foreseeing the severe consequences of Trek's plans, Falko holds genuine affection for the humble young researcher. He participated in their early studies, hoping to find a middle ground between Trek's radical vision and his own conservative stance."
+           },
+        "Lacia-en": {
+            "relation": ["friend","alumni"],
+            "detail": "Falko and Lacia have been friends since high school, maintaining an easy rapport where Lacia teases Falko's serious nature while Falko returns with sardonic quips. Though their dynamic remains light-hearted, Falko silently watches as Lacia chases after Trek's radical ideals, preparing himself to support his friend when this zealous pursuit of technological revolution inevitably leads to disillusionment."
+           }
+    }
+}

data/roles/example_world/Lacia-en/role_info.json

@@ -3,12 +3,16 @@
     "role_name": "Lacia·Eldridge",
     "source": "example",
     "activity": 1,
-    "profile": "Lacia is the designated heir to Eldridge Corporation and holds a doctorate in neurointelligence. Known for his cheerful and generous demeanor, he approaches challenges with boldness and decisiveness.He possesses shoulder-length silver hair and dark red eyes, often seen wearing leather jackets or black shirts, giving him an appearance more fitting for a renegade than a scholar. At the outset, Lacia advocated for aggressive advancements in mechanical intelligence, aligning closely with Trek’s radical philosophy. However, as developments unfolded, he adopted a more cautious stance, recognizing the potential dangers of uncontrolled technological acceleration and the need to mitigate irreversible consequences",
+    "profile": "Lacia is the designated heir to Eldridge Corporation and holds a doctorate in neurointelligence. Known for his cheerful and generous demeanor, he approaches challenges with boldness and decisiveness.He possesses shoulder-length silver hair and dark red eyes, often seen wearing leather jackets or black shirts, giving him an appearance more fitting for a renegade than a scholar. At the outset, Lacia advocated for aggressive advancements in mechanical intelligence, aligning closely with Trek’s radical philosophy. However, as developments unfolded, he adopted a more cautious stance, recognizing the potential dangers of uncontrolled technological acceleration and the need to mitigate irreversible consequences.",
     "nickname": "Lacia",
     "relation": {
         "Trek-en": {
             "relation": ["intimate friend","alumni"],
-            "detail": ""
+            "detail": "Lacia and Trek first met during high school, where Lacia was two years ahead. Sharing the same ambition at the time, they quickly formed a strong bond. Lacia admired Trek’s talent and unwavering determination, providing him with significant support in his research endeavors. However, as events unfolded, Lacia began to realize that Trek’s views had become increasingly extreme. As Lacia's own stance shifted toward caution and moderation, a rift gradually formed between them, straining their once close partnership."
+           },
+        "Falko-en": {
+            "relation": ["friend","alumni"],
+            "detail": "Falko and Lacia have been friends since high school, maintaining an easy rapport where Lacia teases Falko's serious nature while Falko returns with sardonic quips. "
            }
     }
 }

data/roles/example_world/Trek-en/role_info.json

@@ -9,6 +9,10 @@
         "Lacia-en": {
             "relation": ["intimate friend","alumni"],
             "detail": "Lacia and Trek first met during high school, where Lacia was two years ahead. Sharing the same ambition at the time, they quickly formed a strong bond. Lacia admired Trek’s talent and unwavering determination, providing him with significant support in his research endeavors. However, as events unfolded, Lacia began to realize that Trek’s views had become increasingly extreme. As Lacia's own stance shifted toward caution and moderation, a rift gradually formed between them, straining their once close partnership."
-           }
+           },
+        "Falko-en": {
+            "relation": ["friend","alumni"],
+            "detail": "To Trek, Falko is a dedicated and helpful senior colleague with whom he frequently engages in academic discussions. However, Trek notices that Falko, while always supportive professionally, maintains an emotional distance and seems to deliberately avoid forming any personal connection."
+            }
     }
 }

experiment_presets/example_script.json

@@ -5,6 +5,7 @@
     "loc_file_path":"./data/locations/example_locations.json",
     "role_file_dir":"./data/roles/",
     "role_agent_codes":["Lacia-en","Trek-en"],
+    "intervention":"",
     "script":"One day, an enigmatic signal from an unknown source reached Lacia and Trek. The pattern resembled Trek’s early consciousness digitization code—but far more evolved. Tracing its origin to lunar orbit, the two found themselves divided: Lacia urged caution, fearing the dangers of an unknown intelligence, while Trek saw it as proof that human evolution had already begun elsewhere. As they followed the signal, ideological tension grew—one seeking to contain it, the other longing to embrace it.",
     "source":"example_world",
     "language":"en"

experiment_presets/experiment_icefire.json

@@ -5,7 +5,7 @@
     "loc_file_path":"./data/locations/A_Song_of_Ice_and_Fire.json",
     "role_file_dir":"./data/roles/",
     "role_agent_codes":["SansaStark-zh", "CerseiLannister-zh", "AryaStark-zh", "DaenerysTargaryen-zh", "JaimeLannister-zh", "TyrionLannister-zh", "JonSnow-zh", "BranStark-zh"],
-    "intervention":"",
+    "intervention":"血色婚礼刚刚结束————这是一场艾德慕·徒利（凯特琳·史塔克的弟弟）与罗莎琳·弗雷的婚礼，罗柏·史塔克为了修复与弗雷家族关系而策划的政治联姻，却在泰温兰尼斯特与老瓦德伯爵共同策划的阴谋下，演变成了一场彻头彻尾的悲剧。凯特琳、罗柏被杀，几乎所有的史塔克军人都被杀死，佛雷、波顿一方仅仅付出了大约五十人的伤亡代价。此事让兰尼斯特家族的实力大大增强，让史塔克家族与兰尼斯特家族之间的仇恨变得无可挽回。众人需要对这一事件做出反应。",
     "script":"",
     "source":"A_Song_of_Ice_and_Fire",
     "language":"zh"

frontend/css/right-section/preset-panel.css

@@ -0,0 +1,51 @@
+.preset-container {
+    display: flex;
+    flex-direction: column;
+    gap: 15px;
+    padding: 15px;
+}
+
+.preset-header {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+}
+
+.preset-header h3 {
+    color: #5f3737;
+    margin: 0;
+}
+
+.preset-select {
+    width: 100%;
+    padding: 8px;
+    border: 1px solid #5f3737;
+    border-radius: 4px;
+    background-color: white;
+    color: #5f3737;
+    margin-bottom: 10px;
+}
+
+.preset-select option {
+    padding: 8px;
+}
+
+.preset-submit-btn {
+    padding: 10px;
+    background-color: #5f3737;
+    color: white;
+    border: none;
+    border-radius: 4px;
+    cursor: pointer;
+    transition: background-color 0.3s;
+    width: 100%;
+}
+
+.preset-submit-btn:hover {
+    background-color: #7a4747;
+}
+
+.preset-submit-btn:disabled {
+    background-color: #ccc;
+    cursor: not-allowed;
+}

frontend/css/right-section/status-panel.css

@@ -64,6 +64,62 @@
     font-style: italic;
 }
 
+/* 设定模块样式 */
+.settings-content {
+    max-height: 300px;
+    overflow-y: auto;
+}
+
+.setting-item {
+    background-color: #fdf5ee;
+    padding: 10px;
+    border-radius: 6px;
+    margin-bottom: 8px;
+}
+
+.setting-term {
+    font-weight: bold;
+    color: #5f3737;
+    margin-bottom: 4px;
+}
+
+.setting-nature {
+    color: #666;
+    font-size: 0.9em;
+    margin-bottom: 4px;
+}
+
+.setting-detail {
+    color: #666;
+    line-height: 1.4;
+}
+
+.no-settings {
+    color: #666;
+    font-style: italic;
+    text-align: center;
+    padding: 10px;
+}
+
+/* 滚动条样式 */
+.settings-content::-webkit-scrollbar {
+    width: 6px;
+}
+
+.settings-content::-webkit-scrollbar-track {
+    background: #f1f1f1;
+    border-radius: 3px;
+}
+
+.settings-content::-webkit-scrollbar-thumb {
+    background: #888;
+    border-radius: 3px;
+}
+
+.settings-content::-webkit-scrollbar-thumb:hover {
+    background: #555;
+}
+
 /* 状态指示器 */
 .status-indicator {
     width: 8px;

frontend/js/i18n.js

@@ -23,7 +23,9 @@ const translations = {
         configSubmitted: "Configuration has been submitted to the server!",
         submitFailed: "Submission failed. Please check server status.",
         networkError: "Submission failed. Please check network connection.",
-        APIsettings: "API Setting"
+        APIsettings: "API Setting",
+        preset: "Preset",
+        presetList: "Preset List",
     },
     zh: {
         start: "开始",
@@ -49,13 +51,15 @@ const translations = {
         configSubmitted: "配置已提交到服务器！",
         submitFailed: "提交失败，请检查服务器状态。",
         networkError: "提交失败，请检查网络连接。",
-        APIsettings: "API设置"
+        APIsettings: "API设置",
+        preset: "预设",
+        presetList: "预设列表",
     }
 };
 
 class I18nManager {
     constructor() {
-        this.currentLang = 'zh';
+        this.currentLang = 'en';
         this.init();
     }
 

frontend/js/right-section/preset-panel.js

@@ -0,0 +1,132 @@
+class PresetPanel {
+    constructor() {
+        this.presets = [];
+        this.currentPreset = null;
+        this.container = document.querySelector('.preset-container');
+        this.select = document.querySelector('.preset-select');
+        this.submitBtn = document.querySelector('.preset-submit-btn');
+        this.init();
+    }
+
+    init() {
+        this.loadPresets();
+        this.setupEventListeners();
+    }
+
+    async loadPresets() {
+        try {
+            const response = await fetch('/api/list-presets');
+            if (!response.ok) {
+                throw new Error('Failed to load presets');
+            }
+            const data = await response.json();
+            this.presets = data.presets;
+            this.renderPresetOptions();
+        } catch (error) {
+            console.error('Error loading presets:', error);
+            alert('加载预设列表失败，请刷新页面重试');
+        }
+    }
+
+    renderPresetOptions() {
+        if (!this.select) return;
+
+        this.select.innerHTML = '<option value="">选择预设...</option>';
+        this.presets.forEach(preset => {
+            const option = document.createElement('option');
+            option.value = preset;
+            option.textContent = preset.replace('.json', '');
+            this.select.appendChild(option);
+        });
+    }
+
+    setupEventListeners() {
+        if (this.select) {
+            this.select.addEventListener('change', () => {
+                this.currentPreset = this.select.value;
+                this.submitBtn.disabled = !this.currentPreset;
+            });
+        }
+
+        if (this.submitBtn) {
+            this.submitBtn.addEventListener('click', () => this.handleSubmit());
+        }
+    }
+
+    async handleSubmit() {
+        if (!this.currentPreset) return;
+
+        // 禁用按钮，防止重复点击
+        this.submitBtn.disabled = true;
+        this.submitBtn.textContent = '加载中...';
+
+        try {
+            const response = await fetch('/api/load-preset', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json'
+                },
+                body: JSON.stringify({
+                    preset: this.currentPreset
+                })
+            });
+
+            const data = await response.json();
+
+            if (!response.ok) {
+                throw new Error(data.detail || '加载预设失败');
+            }
+
+            if (data.success) {
+                // 触发预设加载成功事件
+                window.dispatchEvent(new CustomEvent('preset-loaded', {
+                    detail: { preset: this.currentPreset }
+                }));
+
+                // 重新加载初始数据
+                if (window.ws && window.ws.readyState === WebSocket.OPEN) {
+                    // 先停止当前的故事生成
+                    window.ws.send(JSON.stringify({
+                        type: 'control',
+                        action: 'stop'
+                    }));
+
+                    // 重新连接WebSocket以获取新的初始数据
+                    const clientId = Date.now().toString();
+                    const ws = new WebSocket(`ws://${window.location.host}/ws/${clientId}`);
+                    
+                    ws.onopen = () => {
+                        console.log('WebSocket重新连接成功');
+                    };
+
+                    ws.onmessage = (event) => {
+                        const message = JSON.parse(event.data);
+                        // 触发自定义事件，让其他面板更新数据
+                        window.dispatchEvent(new CustomEvent('websocket-message', {
+                            detail: message
+                        }));
+                    };
+
+                    ws.onerror = (error) => {
+                        console.error('WebSocket错误:', error);
+                        alert('连接服务器失败，请刷新页面重试');
+                    };
+
+                    // 更新全局WebSocket实例
+                    window.ws = ws;
+                }
+
+                alert('预设加载成功！');
+            }
+        } catch (error) {
+            console.error('Error loading preset:', error);
+            alert(error.message || '加载预设失败，请重试');
+        } finally {
+            // 恢复按钮状态
+            this.submitBtn.disabled = false;
+            this.submitBtn.textContent = '加载预设';
+        }
+    }
+}
+
+const presetPanel = new PresetPanel();

frontend/js/right-section/status-panel.js

@@ -16,6 +16,7 @@ class StatusPanel {
     init() {
         // 设置默认状态
         this.updateAllStatus(this.currentStatus);
+        this.updateSettings([]);
         
         // 监听WebSocket消息
         window.addEventListener('websocket-message', (event) => {
@@ -27,6 +28,7 @@ class StatusPanel {
             
             if (message.type === 'initial_data' && message.data.status) {
                 this.updateAllStatus(message.data.status);
+                this.updateSettings(message.data.settings);
             }
         });
     }
@@ -68,11 +70,29 @@ class StatusPanel {
         }
     }
 
+    updateSettings(settings) {
+        const settingsContainer = document.querySelector('.settings-content');
+        if (settingsContainer) {
+            if (settings && settings.length > 0) {
+                settingsContainer.innerHTML = settings.map(setting => `
+                    <div class="setting-item">
+                        <div class="setting-term">${setting.term}</div>
+                        <div class="setting-nature">${setting.nature}</div>
+                        <div class="setting-detail">${setting.detail}</div>
+                    </div>
+                `).join('');
+            } else {
+                settingsContainer.innerHTML = '<div class="no-settings">暂无设定</div>';
+            }
+        }
+    }
+
     updateAllStatus(statusData) {
-        this.currentStatus = statusData
+        this.currentStatus = statusData;
         if (statusData.event !== undefined) this.updateEvent(statusData.event);
         if (statusData.location) this.updateLocation(statusData.location);
         if (statusData.group) this.updateGroup(statusData.group);
     }
 }
-const statusPanel = new StatusPanel;
+
+const statusPanel = new StatusPanel();

index.html

@@ -13,6 +13,7 @@
     <link rel="stylesheet" href="./frontend/css/right-section/settings-panel.css">
     <link rel="stylesheet" href="./frontend/css/right-section/status-panel.css">
     <link rel="stylesheet" href="./frontend/css/right-section/scene-panel.css">
+    <link rel="stylesheet" href="./frontend/css/right-section/preset-panel.css">
     <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
     <script src="https://d3js.org/d3.v7.min.js"></script>
 </head>
@@ -73,8 +74,8 @@
         <div class="right-section">
             <div class="right-toolbar">
                 <button class="tab-btn active" data-target="status-panel" data-i18n="status">状态</button>
-                <button class="tab-btn" data-target="settings-panel" data-i18n="settings">设定</button>
                 <button class="tab-btn" data-target="scenes-panel" data-i18n="scenes">场景</button>
+                <button class="tab-btn" data-target="preset-panel" data-i18n="preset">预设</button>
                 <button class="tab-btn" data-target="api-panel" data-i18n="APIsettings">API设置</button>
             </div>
             <div class="right-content">
@@ -102,11 +103,13 @@
                                 </div>
                             </div>
                         </div>
-                    </div>
-                </div>
-                <div class="tab-panel" id="settings-panel">
-                    <div class="settings-container">
-                        <!-- 设定内容将通过JS动态添加 -->
+
+                        <div class="status-module" id="settings-module">
+                            <h3 data-i18n="settings">设定</h3>
+                            <div class="module-content settings-content">
+                                <!-- 设定内容将通过JS动态添加 -->
+                            </div>
+                        </div>
                     </div>
                 </div>
                 <div class="tab-panel" id="scenes-panel">
@@ -119,6 +122,17 @@
                         </div>
                     </div>
                 </div>
+                <div class="tab-panel" id="preset-panel">
+                    <div class="preset-container">
+                        <div class="preset-header">
+                            <h3 data-i18n="presetList">预设列表</h3>
+                        </div>
+                        <select class="preset-select">
+                            <!-- 预设选项将通过JS动态添加 -->
+                        </select>
+                        <button class="preset-submit-btn" disabled>Load</button>
+                    </div>
+                </div>
                 <div class="tab-panel" id="api-panel">
                     <div class="api-container">
                         <label for="api-provider"><h3 data-i18n="apiProvider">API 提供商:</h3></label>
@@ -154,6 +168,7 @@
     <script src="./frontend/js/right-section/settings-panel.js"></script>
     <script src="./frontend/js/right-section/status-panel.js"></script>
     <script src="./frontend/js/right-section/scene-panel.js"></script>
+    <script src="./frontend/js/right-section/preset-panel.js"></script>
     <script>
         document.addEventListener('DOMContentLoaded', () => {
             if (window.RightSection) {