diff --git a/src/shared/file-storage/image-history.ts b/src/shared/file-storage/image-history.ts index 796ca72..849eb90 100644 --- a/src/shared/file-storage/image-history.ts +++ b/src/shared/file-storage/image-history.ts @@ -2,7 +2,12 @@ const IMAGE_HISTORY_SIZE = 10000; const imageHistory = new Array(IMAGE_HISTORY_SIZE); let index = 0; -type ImageHistory = { url: string; prompt: string, token?: string }; +type ImageHistory = { + url: string; + prompt: string; + inputPrompt: string; + token?: string; +}; export function addToImageHistory(image: ImageHistory) { if (image.token?.length) { diff --git a/src/shared/file-storage/mirror-generated-image.ts b/src/shared/file-storage/mirror-generated-image.ts index 5a0e45a..c8a1884 100644 --- a/src/shared/file-storage/mirror-generated-image.ts +++ b/src/shared/file-storage/mirror-generated-image.ts @@ -68,7 +68,11 @@ export async function mirrorGeneratedImage( } item.url = `${host}/user_content/${path.basename(mirror)}`; await createThumbnail(mirror); - addToImageHistory({ url: item.url, prompt, token: req.user?.token}); + addToImageHistory({ + url: item.url, + prompt, + inputPrompt: req.body.prompt, + token: req.user?.token}); } return result; }