From f8bd32783b0217c0c97d57c636d6ede4b05ab964 Mon Sep 17 00:00:00 2001
From: Wx-2025 <351320169@qq.com>
Date: Tue, 23 Dec 2025 13:15:53 +0800
Subject: [PATCH] Add support for selected table indices in reorganizer

---
 core/table-system/reorganizer.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/core/table-system/reorganizer.js b/core/table-system/reorganizer.js
index 339a7fb..99e7c3c 100644
--- a/core/table-system/reorganizer.js
+++ b/core/table-system/reorganizer.js
@@ -2,12 +2,12 @@ import { getContext, extension_settings } from "/scripts/extensions.js";
 import { saveChat } from "/script.js";
 import { renderTables } from '../../ui/table-bindings.js';
 import { extensionName } from "../../utils/settings.js";
-import { convertTablesToCsvString, saveStateToMessage, getMemoryState, updateTableFromText, getBatchFillerRuleTemplate, getBatchFillerFlowTemplate } from './manager.js';
+import { convertTablesToCsvString, convertSelectedTablesToCsvString, saveStateToMessage, getMemoryState, updateTableFromText, getBatchFillerRuleTemplate, getBatchFillerFlowTemplate } from './manager.js';
 import { getPresetPrompts, getMixedOrder } from '../../PresetSettings/index.js';
 import { callAI, generateRandomSeed } from '../api.js';
 import { callNccsAI } from '../api/NccsApi.js';
 
-export async function reorganizeTableContent() {
+export async function reorganizeTableContent(selectedTableIndices) {
     const settings = extension_settings[extensionName];
 
     if (window.AMILY2_SYSTEM_PARALYZED === true) {
@@ -24,7 +24,13 @@ export async function reorganizeTableContent() {
     try {
         toastr.info('正在重新整理表格内容...', 'Amily2-重新整理');
         
-        const currentTableDataString = convertTablesToCsvString();
+        let currentTableDataString;
+        if (selectedTableIndices && Array.isArray(selectedTableIndices) && selectedTableIndices.length > 0) {
+            currentTableDataString = convertSelectedTablesToCsvString(selectedTableIndices);
+        } else {
+            currentTableDataString = convertTablesToCsvString();
+        }
+
         if (!currentTableDataString.trim()) {
             toastr.warning('当前没有表格内容需要整理。', 'Amily2-重新整理');
             return;