Update index.js

index.js

@@ -1,9 +1,12 @@
 import { createDrawer } from "./ui/drawer.js";
-import "./MiZheSi/index.js";
+import "./MiZheSi/index.js"; 
-import "./PreOptimizationViewer/index.js";
+import "./PreOptimizationViewer/index.js"; 
 import { registerSlashCommands } from "./core/commands.js";
 import { onMessageReceived, handleTableUpdate } from "./core/events.js";
-import { injectTableData } from "./core/table-system/injector.js";
+import { processPlotOptimization } from "./core/summarizer.js";
+import { getContext } from "/scripts/extensions.js";
+import { characters, this_chid } from '/script.js';
+import { injectTableData } from "./core/table-system/injector.js"; 
 import { loadTables, clearHighlights } from './core/table-system/manager.js';
 import { renderTables } from './ui/table-bindings.js';
 import { log } from './core/table-system/logger.js';
@@ -193,7 +196,7 @@ async function handleMessageBoard() {
 function loadPluginStyles() {
     const loadStyleFile = (fileName) => {
         const styleId = `amily2-style-${fileName.split('.')[0]}`; 
-        if (document.getElementById(styleId)) return;
+        if (document.getElementById(styleId)) return; 
 
         const extensionPath = `scripts/extensions/third-party/${extensionName}/assets/${fileName}?v=${Date.now()}`;
 
@@ -206,11 +209,12 @@ function loadPluginStyles() {
         console.log(`[Amily2号-皇家制衣局] 已为帝国披上华服: ${fileName}`);
     };
 
-    // 颁布四道制衣圣谕
+    // 颁布五道制衣圣谕
     loadStyleFile("style.css"); // 【第一道圣谕】为帝国主体宫殿披上通用华服
     loadStyleFile("historiography.css"); // 【第二道圣谕】为敕史局披上其专属华服
     loadStyleFile("hanlinyuan.css"); // 【第三道圣谕】为翰林院披上其专属华服
     loadStyleFile("table.css"); // 【第四道圣谕】为内存储司披上其专属华服
+    loadStyleFile("optimization.css"); // 【第五道圣谕】为剧情优化披上其专属华服
 }
 
 
@@ -229,9 +233,7 @@ jQuery(async () => {
   if (!extension_settings[extensionName]) {
     extension_settings[extensionName] = {};
   }
-
   const combinedDefaultSettings = { ...defaultSettings, ...tableSystemDefaultSettings };
-
   Object.assign(extension_settings[extensionName], {
     ...combinedDefaultSettings,
     ...extension_settings[extensionName],
@@ -261,35 +263,112 @@ jQuery(async () => {
 
 
         console.log("[Amily2号-开国大典] 步骤四：部署帝国哨兵网络...");
-        if (!window.amily2EventsRegistered) {
 
+        let isProcessingPlotOptimization = false;
+        async function onPlotGenerationAfterCommands(type, params, dryRun) {
+            console.log("[Amily2-剧情优化] Generation after commands triggered", { type, params, dryRun, isProcessing: isProcessingPlotOptimization });
+            
+            if (type === 'regenerate' || isProcessingPlotOptimization || dryRun) {
+                console.log("[Amily2-剧情优化] Skipping due to conditions:", { type, isProcessing: isProcessingPlotOptimization, dryRun });
+                return;
+            }
+        
+            const globalSettings = extension_settings[extensionName];
+            
+            console.log("[Amily2-剧情优化] Checking plot optimization settings", { 
+                enabled: globalSettings?.plotOpt_enabled, 
+                hasApiUrl: !!globalSettings?.apiUrl,
+                plotSettings: globalSettings
+            });
+
+            console.log("[Amily2-剧情优化] Detailed settings check:", {
+                globalEnabled: globalSettings?.enabled,
+                plotSettingsExists: !!globalSettings,
+                plotSettingsEnabled: globalSettings?.plotOpt_enabled,
+                plotSettingsApiUrl: globalSettings?.apiUrl,
+                fullPlotSettings: globalSettings
+            });
+            
+            const isPlotOptEnabled = globalSettings?.plotOpt_enabled !== false; // 如果未设置，默认启用
+            if (!isPlotOptEnabled || !globalSettings?.apiUrl) {
+                console.log("[Amily2-剧情优化] Plot optimization disabled or missing API URL", {
+                    optimizationEnabled: globalSettings?.plotOpt_enabled,
+                    apiUrl: globalSettings?.apiUrl
+                });
+                return;
+            }
+        
+            isProcessingPlotOptimization = true;
+            let $toast = toastr.info('正在进行剧情优化...', '剧情优化', { timeOut: 0, extendedTimeOut: 0 });
+        
+            try {
+                console.log("[Amily2-剧情优化] Event parameters:", { type, params, dryRun });
+                const userMessage = $('#send_textarea').val();
+                
+                if (!userMessage) {
+                    console.log("[Amily2-剧情优化] No user message found in textarea");
+                    if ($toast) toastr.clear($toast);
+                    return;
+                }
+        
+                console.log("[Amily2-剧情优化] Processing plot optimization for message:", userMessage);
+        
+                const context = getContext();
+                const contextTurnCount = globalSettings.plotOpt_contextLimit || 10;
+                let slicedContext = [];
+                if (contextTurnCount > 0) {
+                    slicedContext = context.chat.slice(-contextTurnCount);
+                }
+        
+                const result = await processPlotOptimization({ mes: userMessage }, slicedContext);
+        
+                if (result && result.contentToAppend) {
+                    const currentUserInput = $('#send_textarea').val();
+                    const finalMessage = currentUserInput + '\n' + result.contentToAppend;
+                    
+                    $('#send_textarea').val(finalMessage);
+                    $('#send_textarea').trigger('input');
+                    
+                    if ($toast) toastr.clear($toast);
+                    toastr.success('剧情优化已完成并注入。', '操作成功');
+                    console.log("[Amily2-剧情优化] Plot optimization completed and appended successfully.");
+                } else {
+                    if ($toast) toastr.clear($toast);
+                    console.log("[Amily2-剧情优化] Plot optimization returned no result to append.");
+                }
+        
+            } catch (error) {
+                console.error(`[Amily2-剧情优化] 处理发送前事件时出错:`, error);
+                if ($toast) toastr.clear($toast);
+                toastr.error('剧情优化处理失败。', '错误');
+            } finally {
+                isProcessingPlotOptimization = false;
+            }
+        }
+        if (!window.amily2EventsRegistered) {
+            eventSource.on(event_types.GENERATION_AFTER_COMMANDS, onPlotGenerationAfterCommands);
+            
             eventSource.on(event_types.MESSAGE_RECEIVED, onMessageReceived);
             eventSource.on(event_types.IMPERSONATE_READY, onMessageReceived);
 
-
             eventSource.on(event_types.MESSAGE_RECEIVED, (chat_id) => handleTableUpdate(chat_id));
             eventSource.on(event_types.MESSAGE_SWIPED, (chat_id) => {
-
                 clearHighlights();
                 handleTableUpdate(chat_id);
             });
             eventSource.on(event_types.MESSAGE_EDITED, (mes_id) => handleTableUpdate(mes_id));
 
             eventSource.on(event_types.CHAT_CHANGED, () => {
-
                 setTimeout(() => {
                     log("【监察系统】检测到“朝代更迭”(CHAT_CHANGED)，开始重修史书并刷新宫殿...", 'info');
-
                     clearHighlights();
                     loadTables();
                     renderTables();
                 }, 100);
             });
 
-
             eventSource.on(event_types.MESSAGE_DELETED, (message, index) => {
                 log(`【监察系统】检测到消息 ${index} 被删除，开始精确回滚UI状态。`, 'warn');
-
                 clearHighlights();
                 loadTables(index);
                 renderTables();
@@ -303,11 +382,8 @@ jQuery(async () => {
         const originalFunction = window[officialFunctionName];
 
         if (typeof originalFunction === 'function') {
-
             window[officialFunctionName] = async function(...args) {
-
                 injectTableData(...args);
-
                 await originalFunction.apply(this, args); 
             };
             console.log(`[Amily2-内存储司] 已成功代理 ${officialFunctionName}，与翰林院协同工作。`);
@@ -337,7 +413,7 @@ jQuery(async () => {
             } catch (error) {
                 log(`【凤凰阁】内联主题系统初始化失败: ${error}`, 'error');
             }
-        }, 500); 
+        }, 500); // 延迟500毫秒，给予充分的渲染时间
 
       } catch (error) {
         console.error("!!!【开国大典失败】在执行系列法令时发生严重错误:", error);