Stable Diffusion Fooocus批量画图脚本 - ToB企服应用市场:ToB评测及商务社交产业平台

// ==UserScript==
// @name Fooocus批量绘图
// @namespace https://blog.aidec.tw/ || https://blog.csdn.net/liuqihang11?type=blog
// @version 2024-09-17
// @description 输入以句号结尾的多行提示词实现Fooocus批量绘图，提示词格式：一行为一组，末尾用英文句号隔开。
// @author AidecLi Create || Syphomn Modify
// @match http://192.0.0.1:7865/*
// @icon https://www.google.com/s2/favicons?sz=64&domain=0.1
// @grant none
// ==/UserScript==
(function() {
'use strict';
let isProcessing = false;
let stopRequested = false;
let promptInputTextarea, processedTextarea, startButton, stopButton, clearButton, progressBar, statusDiv;
// 添加UI元素
function addUI() {
const targetElement = document.querySelector('#component-3');
if (!targetElement) {
console.error('目标元素 #component-3 未找到');
return;
}
const container = document.createElement('div');
container.style.marginTop = '20px';
container.style.display = 'flex';
container.style.flexDirection = 'column';
container.style.gap = '10px';
// 待处理文本框
promptInputTextarea = document.createElement('textarea');
promptInputTextarea.id = 'promptList';
promptInputTextarea.rows = 5;
promptInputTextarea.style.width = '100%';
promptInputTextarea.style.color = 'red'; // 设置文字为红色
promptInputTextarea.placeholder = 'Unprocessed Prompts';
// 已处理文本框
processedTextarea = document.createElement('textarea');
processedTextarea.id = 'processedList';
processedTextarea.rows = 5;
processedTextarea.style.width = '100%';
processedTextarea.style.color = 'green'; // 设置文字为绿色
processedTextarea.placeholder = 'Processed Prompts';
processedTextarea.readOnly = true;
// 开始批次处理按钮
startButton = document.createElement('button');
startButton.textContent = 'Start Batch-Drawing';
startButton.onclick = processPrompts;
startButton.style.padding = '5px 10px';
startButton.style.cursor = 'pointer';
startButton.className = 'lg secondary type_row svelte-cmf5ev';
// 停止批次处理按钮
stopButton = document.createElement('button');
stopButton.textContent = 'Stop Batch-Drawing';
stopButton.onclick = stopProcessing;
stopButton.style.padding = '5px 10px';
stopButton.style.cursor = 'pointer';
stopButton.className = 'lg secondary type_row svelte-cmf5ev';
// 清除已处理文本框按钮
clearButton = document.createElement('button');
clearButton.textContent = 'Clear Processed-Text';
clearButton.onclick = clearProcessed;
clearButton.style.padding = '5px 10px';
clearButton.style.cursor = 'pointer';
clearButton.className = 'lg secondary type_row svelte-cmf5ev';
clearButton.disabled = true; // 初始化为不可用状态
// 进度条
progressBar = document.createElement('progress');
progressBar.id = 'progressBar';
progressBar.value = 0;
progressBar.max = 100;
progressBar.style.width = '100%';
// 状态显示
statusDiv = document.createElement('div');
statusDiv.id = 'statusDiv';
statusDiv.textContent = 'Enter Multiple Lines of Prompts Separated by a Period';
statusDiv.style.fontWeight = 'bold';
// 将UI元素添加到容器中
container.appendChild(statusDiv);
container.appendChild(promptInputTextarea);
container.appendChild(processedTextarea);
container.appendChild(startButton);
container.appendChild(stopButton);
container.appendChild(clearButton);
container.appendChild(progressBar);
targetElement.appendChild(container);
}
// 处理批次提示词
async function processPrompts() {
if (isProcessing) return;
isProcessing = true;
stopRequested = false;
disableUI();
// 重新读取未处理文本框中的内容
const promptList = promptInputTextarea.value.split('\n').filter(p => p.trim() !== '');
const totalPrompts = promptList.length;
const positivePromptTextarea = document.querySelector('#positive_prompt textarea');
const generateButton = document.querySelector('#generate_button');
let i = 0;
for (const prompt of promptList) {
if (stopRequested) break;
statusDiv.textContent = `正在处理第 ${i + 1} 组，共 ${totalPrompts} 组提示词: ${prompt}`;
positivePromptTextarea.value = prompt;
positivePromptTextarea.dispatchEvent(new Event('input', { bubbles: true }));
console.log('开始绘制' + prompt);
await new Promise(resolve => setTimeout(resolve, 500));
generateButton.click();
await new Promise(resolve => setTimeout(resolve, 500));
await waitForGeneration(generateButton);
if (stopRequested) break;
console.log('结束绘制' + prompt);
// 将已处理的移到已处理区
processedTextarea.value += prompt + '\n';
// 从待处理区移除已处理的提示词
promptInputTextarea.value = promptList.slice(i + 1).join('\n');
// 更新进度条
progressBar.value = ((i + 1) / totalPrompts) * 100;
// 等待2秒再继续下一个提示词
statusDiv.textContent = `第 ${i + 1} 组已处理完毕，即将进入下一组。`;
await new Promise(resolve => setTimeout(resolve, 1000));
i++;
}
if (stopRequested) {
statusDiv.textContent = '批量绘图已停止';
} else {
statusDiv.textContent = '所有提示词均处理完毕';
clearButton.disabled = false; // 处理完毕后启用清除按钮
}
isProcessing = false;
enableUI();
}
// 停止处理
function stopProcessing() {
stopRequested = true;
isProcessing = false;
enableUI();
// 也停止本轮正在运作的生图
const stopButton = document.querySelector('#stop_button');
stopButton.click();
// 启用清除按钮
clearButton.disabled = false;
}
// 清除已处理文本框的内容
function clearProcessed() {
processedTextarea.value = '';
clearButton.disabled = true; // 清除后禁用清除按钮
}
// 禁用UI元素
function disableUI() {
promptInputTextarea.disabled = true;
startButton.disabled = true;
startButton.textContent = 'Processing...';
}
// 启用UI元素
function enableUI() {
promptInputTextarea.disabled = false;
startButton.disabled = false;
startButton.textContent = 'Start Batch-Drawing';
}
// 等待生成过程
function waitForGeneration(button) {
return new Promise(resolve => {
const observer = new MutationObserver(() => {
if (!button.disabled && !button.hidden) {
// 等待按钮
setTimeout(() => {
observer.disconnect();
resolve();
}, 500);
}
});
observer.observe(button, { attributes: true, attributeFilter: ['disabled', 'hidden'] });
if (stopRequested) resolve();
// 如果按钮可用且没有hidden，则继续
if (!button.disabled && !button.hidden) {
observer.disconnect();
resolve();
}
});
}
// 将自定义UI添加到页面上
function initScript() {
const maxAttempts = 10;
let attempts = 0;
function tryAddUI() {
if (document.querySelector('#component-3')) {
addUI();
} else if (attempts < maxAttempts) {
attempts++;
setTimeout(tryAddUI, 1000); // 等待1秒重新尝试
} else {
console.error('在 ' + maxAttempts + ' 次尝试后未能找到 #component-3');
}
}
tryAddUI();
}
// 页面完成加载UI到页面
if (document.readyState === 'loading') {
window.addEventListener('DOMContentLoaded', initScript);
} else {
initScript();
}
})();

复制代码