From aa1504b12946b8acd789510da2b5c8b3c6477b09 Mon Sep 17 00:00:00 2001 From: bluelovers Date: Fri, 17 May 2024 10:38:43 +0800 Subject: [PATCH] Update formatInfo.ts (#569) --- src/modules/ImageInfo/features/formatInfo.ts | 22 +++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) diff --git a/src/modules/ImageInfo/features/formatInfo.ts b/src/modules/ImageInfo/features/formatInfo.ts index 82de2e2..40b2bb4 100644 --- a/src/modules/ImageInfo/features/formatInfo.ts +++ b/src/modules/ImageInfo/features/formatInfo.ts @@ -1,10 +1,26 @@ import { parseFromRawInfo } from '@bluelovers/auto1111-pnginfo'; -import { Converter } from '@/scripts/formatPrompt'; +import { splitSmartly } from 'split-smartly2'; const formatPrompt = (prompt: string) => { - let newPrompt = prompt.replaceAll('<', '<').replaceAll('>', '>'); - return Converter.convert(newPrompt); + let newPrompt = prompt + .replaceAll('<', '<') + .replaceAll('>', '>') + .replace(/^[\x00\s,,。]+$/gm, '') + .replace(/\n{2,}/g, '\n') + ; + + const entries = splitSmartly(newPrompt.replace(/\n/g, '
'), [',',',','。'], { + brackets: true, + trimSeparators: true, + }) as string[]; + + return entries + .filter(line => line.length) + .join(', ') + .replace(/
/g, '\n') + .replace(/^\s+|\s+$/gm, '') + ; }; export const formatInfo = (info: string) => {