[ { "path": ".eslintignore", "content": ".DS_Store\nnode_modules\n/javascript\n.env\n.env.*\n!.env.example\n\n# Ignore files for PNPM, NPM and YARN\npnpm-lock.yaml\npackage-lock.json\nyarn.lock\n\n# Config files\n.eslintrc.cjs\nvite.config.ts\nsvelte.config.js\n*.config.cjs\n" }, { "path": ".eslintrc.cjs", "content": "module.exports = {\n parser: \"@typescript-eslint/parser\",\n parserOptions: {\n tsconfigRootDir: __dirname,\n project: [\"./tsconfig.json\"],\n extraFileExtensions: [\".svelte\"],\n },\n extends: [\n \"eslint:recommended\",\n \"plugin:@typescript-eslint/recommended\",\n \"plugin:@typescript-eslint/recommended-requiring-type-checking\",\n \"plugin:svelte/recommended\",\n \"plugin:svelte/prettier\",\n \"prettier\",\n ],\n plugins: [\"@typescript-eslint\"],\n overrides: [\n {\n files: [\"**/*.svelte\"],\n parser: \"svelte-eslint-parser\",\n parserOptions: {\n parser: \"@typescript-eslint/parser\",\n },\n },\n {\n files: [\"**/*.ts\", \"**/*.svelte\"],\n rules: {\n \"no-undef\": \"off\",\n },\n },\n ]\n};\n" }, { "path": ".gitignore", "content": "# Logs\nlogs\n*.log\nnpm-debug.log*\nyarn-debug.log*\nyarn-error.log*\npnpm-debug.log*\nlerna-debug.log*\n\nnode_modules\ndev-scripts/tmp\ndist\ndist-ssr\n*.local\n\n# Editor directories and files\n.vscode/*\n!.vscode/extensions.json\n.idea\n.DS_Store\n*.suo\n*.ntvs*\n*.njsproj\n*.sln\n*.sw?\n\n# Directories created at runtime\n__pycache__/\nuser-data/\n" }, { "path": ".husky/pre-commit", "content": "#!/usr/bin/env sh\n. \"$(dirname -- \"$0\")/_/husky.sh\"\n\npnpm exec lint-staged\n" }, { "path": ".lintstagedrc.json", "content": "{\n \"client-src/**/*.{js,ts,svelte}\": [\n \"pnpm run check\",\n \"prettier --write\",\n \"eslint --fix\"\n ]\n}\n" }, { "path": ".prettierignore", "content": ".DS_Store\nnode_modules\n/javascript\n.env\n.env.*\n!.env.example\n\n# Ignore files for PNPM, NPM and YARN\npnpm-lock.yaml\npackage-lock.json\nyarn.lock\n" }, { "path": ".prettierrc", "content": "{\n \"useTabs\": false,\n \"singleQuote\": false,\n \"trailingComma\": \"es5\",\n \"printWidth\": 100,\n \"plugins\": [\n \"prettier-plugin-svelte\",\n \"prettier-plugin-tailwindcss\"\n ],\n \"pluginSearchDirs\": [\".\"],\n \"overrides\": [\n {\n \"files\": \"*.svelte\",\n \"options\": {\n \"parser\": \"svelte\"\n }\n }\n ]\n}\n" }, { "path": "CHANGELOG.md", "content": "## v0.4.1\n### Add\n- Added a cancel button to the registration dialog of \"My Prompt\".\n## Fix\n- Fixed an issue where the option values were not reflected in the new version of the Web UI.\n### Remove\n- Removed version change feature.\n- Removed update notification feature.\n\n## v0.4.0\n### Add\n- Added \"My Prompt\" feature, which was originally developed as \"Aliases\". Refer to the [My Prompt Tab](https://github.com/eideehi/sd-webui-better-prompt#my-prompt-tab) section in the README for details.\n### Change\n- Changed the format of the translation file.\n### Fix\n- When setting the default weight for LoRA, it now references the Web UI configuration. **([#13])**\n- Fixed the issue where the prompt input area didn't maximize its width when wrapped.\n\n## v0.3.0\n### Update\n- Updated Danbooru tags.\n### Change\n- (For developers) Introduced the Svelte framework.\n- Changed the word \"Multiplier\" to \"Weight\".\n- Enabled mouse wheel control for the Weight slider.\n- Improved prompt parsing performance. **([#12])**\n### Fix\n- Improved synchronization with the original prompt area. **([#8])**\n### Remove\n- Temporarily removed the Undo Redo feature.\n\n## v0.2.0\n### Add\n- Added a button to display metadata in thumbnail preview for Textual Inversion and LoRA.\n- Added undo/redo functionality to the prompt.\n### Fix\n- Fixed an issue where list elements would flicker when updating the prompt.\n- Fixed bugs related to Git.\n\n## v0.1.0\nThis is the initial release of Better Prompt.\n\n\n[#8]: https://github.com/eideehi/sd-webui-better-prompt/issues/8\n[#12]: https://github.com/eideehi/sd-webui-better-prompt/issues/12\n[#13]: https://github.com/eideehi/sd-webui-better-prompt/issues/13\n" }, { "path": "LICENSE", "content": "The MIT License (MIT)\n\nCopyright (c) 2023 EideeHi\n\nPermission is hereby granted, free of charge, to any person obtaining a copy\nof this software and associated documentation files (the \"Software\"), to deal\nin the Software without restriction, including without limitation the rights\nto use, copy, modify, merge, publish, distribute, sublicense, and/or sell\ncopies of the Software, and to permit persons to whom the Software is\nfurnished to do so, subject to the following conditions:\n\nThe above copyright notice and this permission notice shall be included in\nall copies or substantial portions of the Software.\n\nTHE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\nIMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\nFITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\nAUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\nLIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\nOUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\nTHE SOFTWARE." }, { "path": "README.adoc", "content": "= Better Prompt\n\nEnglish | link:docs/README-ja.adoc[日本語]\n\nBetter Prompt is an extension of the https://github.com/AUTOMATIC1111/stable-diffusion-webui[Stable Diffusion web UI] that adds a UI to assist with prompt input and editing.\n\nimage::docs/images/overview.png[Image - Overview]\n\n== Overview\nBetter Prompt was created to reduce the various inconveniences of traditional prompt input and editing. It allows you to set Textual Inversion and LoRA without displaying Extra networks (🎴), rearrange the order of prompts by drag and drop, and adjust the emphasis level of prompts through a GUI.\n\n== Precautions\nBetter Prompt is a JavaScript-based extension. Please be aware that if JavaScript files are not loaded correctly, such as when using the `--data-dir` option, Better Styles not function properly.\n\n== Installation\n=== Installing from the Browser (Recommended)\nYou can install it from the \"Install from URL\" option in the Extensions tab. Simply enter https://github.com/eideehi/sd-webui-better-prompt.git in the \"URL for extension's git repository\" field and press the \"Install\" button.\n\nimage::docs/images/install.png[Image - Install]\n\n=== Installing using Git\nYou can also install it by executing the following command in the directory where Stable Diffusion web UI is installed.\n[source,shell]\n----\ngit clone https://github.com/eideehi/sd-webui-better-prompt.git extensions/sd-webui-better-prompt\n----\n\n=== Better Prompt is not displayed or the display is broken\nBy installing it using the above methods, the latest version will be installed. However, it may not work well with the version of Web UI you are using. If Better Prompt is not displayed properly, it may be possible to resolve the issue by using a different version.\n\nTIP: You can change the version of Better Prompt from the Settings tab. For more information, refer to <>.\n\n== Usage\nBetter Prompt adds the part enclosed by the red line in the image below. We will explain how to use this added content from now on.\n\nimage::docs/images/components.png[Image - Components]\n\n=== Prompt Edit Tab\nThis is the default content displayed where you can edit the prompt.\n\n==== Prompt Addition Form [[input-form]]\nAdding a prompt is done using the three elements shown in the following image.\n\nimage::docs/images/input-component.png[Image - Input Component]\n\n===== [1] Prompt Input Field [[input-field]]\nEnter a prompt in this field and press the Enter key to add the content to the positive prompt. Alternatively, press the Shift key while pressing Enter to add the content to the negative prompt. If there are *Textual Inversion*, *LoRA*, *Danbooru tags*, or *My Prompt* similar to the input content, they will be displayed in a list in <>.\n\nTIP: Fuse.js is used for similarity determination, allowing various determination methods such as \"exact match,\" \"prefix/suffix,\" and \"AND/OR/NOT\" to be used. For more information, refer to the https://fusejs.io/examples.html#extended-search[fusejs.io documentation].\n\nTIP: Press the Tab key to move the focus to the elements in <>, and press the Escape key to return the focus. The selected element can be added to the positive (negative) prompt by pressing Enter (or Shift + Enter).\n\n===== [2] Suggest Filters\nOnly elements with a check in these filters will be displayed in <>.\n\n===== [3] Suggest Area [[suggest-area]]\nDisplays a list of up to 20 elements similar to the content entered in <>. An example of the items to be added is shown in the following image.\n\nimage::docs/images/suggest-items.png[Image - Suggest Items]\n\nGreen is Textual Inversion, blue is LoRA, and the last is Danbooru tags. Clicking (or Shift-clicking) these elements allows them to be added to the positive (negative) prompt.\n\nNOTE: LoRA cannot be added to the negative prompt, so please be careful.\n\nTIP: You can check the thumbnail by right-clicking on the elements of Textual Inversion and LoRA.\n\n==== Prompt Editing\nPrompt editing is done using the two elements shown in the following image.\n\nimage::docs/images/prompt-component.png[Image - Prompt Component]\n\n===== [1] Positive Prompt [[positive-prompt]]\nPrompts added using <> are displayed in this area. This element is synchronized with the positive prompt input area of the web UI.\n\nTIP: Each prompt can be reordered by drag and drop, and can be deleted by clicking while holding the Shift key.\n\nTIP: Right-clicking on LoRA and normal prompts displays a popup for adjusting the emphasis level.\n\n===== [2] Negative Prompt\nExcept for not being able to add LoRA, it is the same as <>.\n\n=== My Prompt Tab\nYou can assign aliases and tags to any prompt for easy management. The registered My Prompts will be displayed in the <> when editing prompts.\n\nimage::docs/images/my-prompt.png[Image - My Prompt Overview]\n\n==== Registering My Prompt\nWhen you click the ``Add new My Prompt`` button, a popup will appear for registering a new My Prompt.\n\nimage::docs/images/add-new-my-prompt.png[Image - Add new My Prompt]\n\n===== Label [[my-prompt-label]]\nSet the display name for the My Prompt. This will be used when displayed in suggestions, among other places. *This field is required and cannot be omitted*. Additionally, you cannot set a label that already exists.\n\n===== Tags [[my-prompt-tags]]\nBy setting tags, you can filter My Prompts based on their tags. When the input field is active, existing tags will be suggested and displayed as a list. You can add a new tag by pressing the Enter key after entering the desired string. This field is optional, and you can omit the input.\n\n===== Prompt [[my-prompt-prompt]]\nSet any prompt for the My Prompt. When you click on a My Prompt displayed in the <>, the prompt you set here will be expanded. This field is required and cannot be omitted.\n\n==== Deleting My Prompt\nBy clicking on a My Prompt in the displayed list, you can select it by clicking the ``Select and delete My Prompt`` button. Then, you can delete all the selected My Prompts by pressing the ``Delete selected My Prompts`` button. Please note that once you delete a My Prompt, it cannot be restored, so use this feature with caution.\n\nimage::docs/images/delete-my-prompts.png[Image - Delete My Prompts]\n\n==== My Prompt Search\nThe search for My Prompts references the elements <>, <>, and <>. If you enter ``style`` in the search form, any My Prompts that contain the string ``style`` in any of the aforementioned elements will be displayed in the search results.\n\n== Configuration\nBetter Prompt creates its own configuration section in the Settings tab. Here, we will explain each item.\n\nimage::docs/images/settings.png[Image - Settings]\n\n=== Version of Better Prompt [[version-change]]\nYou can change the version of Better Prompt. Selecting a blank space will change it to the latest version at that point. If you change the settings, it is necessary to restart the Web UI. (not just reload)\n\nTIP: The current version of Better Prompt is displayed in the console of the Web UI. Refer to the table below for the Web UI versions corresponding to each version.\n\n|===\n| Version | Web UI Version (Minimum) | Web UI Version (Maximum)\n| 0.1.0 - 0.3.0 | 9e1afa9e (2023-03-25) | 1.3.2\n| 0.4.0 | 9e1afa9e (2023-03-25) | 1.4.0\n| 0.4.1 | 1.1.1 | ~\n|===\n\n=== Display update notifications\nIf checked, it will display notifications when updates are available.\n\n=== Notify of updates only once per version\nIf checked, it will only notify once for each version when updates are available.\n\n=== Interval at which to display update notifications\nSpecify the interval for displaying update notifications. The unit is \"days\", and the default value is 1 day.\n\n=== Language of Better Prompt\nSpecify the language used by Better Prompt. The default value is blank (English). Currently, ja_JP language is available. If you change the settings, it is necessary to reload the Web UI.\n\n== License\nBetter Prompt is developed and published under the MIT license. For details on the license, please refer to the link below.\n\nlink:./LICENSE[MIT License]\n" }, { "path": "client-src/@types/global.d.ts", "content": "type ExtensionAvailableTab = \"txt2img\" | \"img2img\";\ntype WebUiTab = ExtensionAvailableTab | \"other\";\n\ntype Nullable = T | null | undefined;\ntype Callback = () => unknown;\ntype Callback1 = (arg1: T) => unknown;\n\ntype OneOrMany = T | T[];\n" }, { "path": "client-src/@types/modules.ts", "content": "declare module \"#/widgets/Toast.svelte\" {\n export { SvelteComponentDev as default } from \"svelte/internal\";\n\n export type ToastType = \"info\" | \"success\" | \"warning\" | \"error\";\n\n export type ToastMessage = {\n type?: ToastType;\n text: string;\n duration?: number;\n };\n\n export function showToast(message: ToastMessage): void;\n\n export function closeToast(message: ToastMessage): void;\n}\n" }, { "path": "client-src/@types/vite-env.d.ts", "content": "/// \n/// \n" }, { "path": "client-src/@types/webui.d.ts", "content": "// injected-script\ndeclare let localization: Record;\n\n// script.js\ndeclare function gradioApp(): Document | ShadowRoot;\ndeclare function get_uiCurrentTabContent(): Nullable;\ndeclare function onUiLoaded(callback: Callback): void;\ndeclare function onUiTabChange(callback: Callback): void;\ndeclare function onUiUpdate(callback: Callback): void;\n\n// javascript/extraNetworks.js\ndeclare function extraNetworksRequestMetadata(\n event: Event,\n extraPage: string,\n cardName: string\n): void;\n\n// javascript/localization.js\ndeclare function getTranslation(text: string): Nullable;\n\n// javascript/ui.js\ndeclare type OptionValue = string | number | boolean;\ndeclare let opts: Record;\ndeclare function updateInput(target: HTMLElement): void;\n" }, { "path": "client-src/components/better-prompt/BetterPrompt.svelte", "content": "\n\n
\n
\n \n {promptEdit.translate()}\n \n \n
\n
\n \n \n
\n
\n\n\n" }, { "path": "client-src/components/better-prompt/_logic/adjustPrompt.ts", "content": "import type { Prompt } from \"@/libs/prompt\";\nimport type { ExtraNetworksData } from \"@/libs/extra-networks\";\n\nexport function adjustPrompt(prompt: Prompt, textualInversions: ExtraNetworksData[]): Prompt {\n if (prompt.type !== \"plain\") return prompt;\n const data = textualInversions.find((data) => data.name === prompt.value);\n if (data == null) return prompt;\n return {\n type: \"extra-networks\",\n name: data.type,\n args: [data.name],\n };\n}\n" }, { "path": "client-src/components/better-prompt/_logic/betterPrompt.ts", "content": "import type { ExtraNetworksData, ExtraNetworksType } from \"@/libs/extra-networks\";\nimport { type Readable, readable } from \"svelte/store\";\nimport { getElement, getElementAll } from \"@/libs/util/dom\";\nimport { generateHashCode } from \"@/libs/util/string\";\n\nexport function createLoraReadable(tabName: ExtensionAvailableTab): Readable {\n return createReadable(`#${tabName}_lora_cards`, \"lora\");\n}\n\nexport function createTextualInversionReadable(\n tabName: ExtensionAvailableTab\n): Readable {\n return createReadable(`#${tabName}_textual_inversion_cards`, \"textual-inversion\");\n}\n\nfunction createReadable(selector: string, type: ExtraNetworksType): Readable {\n return readable([], (set) => {\n let timeout: number;\n\n let currentHash = -1;\n const update = () => {\n const element = getElement(selector);\n if (element == null) {\n timeout = window.setTimeout(update, 1000);\n return;\n }\n\n const hash = generateHashCode(element.innerHTML);\n if (hash === currentHash) {\n timeout = window.setTimeout(update, 1000);\n return;\n }\n currentHash = hash;\n\n set(\n getElementAll(`${selector} > .card`).map((card) => {\n const name = getElement(card, \".actions > .name\")?.textContent || \"\";\n const search_term =\n getElement(card, \".actions > .additional > .search_term\")?.textContent || \"\";\n const thumbnail = card.style.backgroundImage.trim();\n return thumbnail.length === 0\n ? { type, name, search_term }\n : { type, name, search_term, thumbnail };\n })\n );\n timeout = window.setTimeout(update, 1000);\n };\n\n timeout = window.setTimeout(update, 0);\n return () => {\n window.clearTimeout(timeout);\n };\n });\n}\n" }, { "path": "client-src/components/better-prompt/_logic/context.ts", "content": "import type { Writable } from \"svelte/store\";\nimport type { Prompt } from \"@/libs/prompt\";\n\nexport type BetterPromptContext = {\n tabName: ExtensionAvailableTab;\n prompts: {\n positive: Writable;\n negative: Writable;\n };\n};\n\nexport const betterPromptContextKey = Symbol();\n" }, { "path": "client-src/components/better-prompt/_logic/danbooruTags.ts", "content": "import { type Readable, writable } from \"svelte/store\";\nimport type { DanbooruTag } from \"@/libs/danbooru\";\n\nconst _danbooruTags = writable([]);\nexport const danbooruTags: Readable = { subscribe: _danbooruTags.subscribe };\n\nlet init = false;\n\nexport function initDanbooruTags(tags: DanbooruTag[]): void {\n if (init) return;\n init = true;\n _danbooruTags.set(tags);\n}\n" }, { "path": "client-src/components/better-prompt/_logic/extraNetworks.ts", "content": "import type { ExtraNetworksData } from \"@/libs/extra-networks\";\nimport { type Readable, writable } from \"svelte/store\";\n\nconst _lora = writable([]);\nexport const lora: Readable = { subscribe: _lora.subscribe };\n\nconst _textualInversion = writable([]);\nexport const textualInversion: Readable = {\n subscribe: _textualInversion.subscribe,\n};\n\nlet loraInit = false;\nlet textualInversionInit = false;\n\nexport function initLora(tags: ExtraNetworksData[]): void {\n if (loraInit) return;\n loraInit = true;\n _lora.set(tags);\n}\n\nexport function initTextualInversion(tags: ExtraNetworksData[]): void {\n if (textualInversionInit) return;\n textualInversionInit = true;\n _textualInversion.set(tags);\n}\n" }, { "path": "client-src/components/better-prompt/_logic/messages.ts", "content": "import { t } from \"@/libs/util/webui\";\n\nexport interface Message {\n translate(args?: Array): string;\n}\n\nclass MessageImpl implements Message {\n private readonly key: string;\n private readonly defaultValue: string;\n\n constructor(key: string, defaultValue: string) {\n this.key = key;\n this.defaultValue = defaultValue;\n }\n\n translate(args?: Array): string {\n return t(this.key, { defaultValue: this.defaultValue, args });\n }\n}\n\nconst addNewMyPrompt = new MessageImpl(\"add-new-my-prompt\", \"Add new My Prompt\");\nconst addThisMyPrompt = new MessageImpl(\"add-this-my-prompt\", \"Add this My Prompt\");\nconst cancelAddingMyPrompt = new MessageImpl(\"cancel-adding-my-prompt\", \"Cancel adding My Prompt\");\nconst cancelMyPromptDeletion = new MessageImpl(\n \"cancel-my-prompt-deletion\",\n \"Cancel My Prompt deletion\"\n);\nconst deleteSelectedMyPrompt = new MessageImpl(\n \"delete-selected-my-prompt\",\n \"Delete selected My Prompts\"\n);\nconst editorNegativePrompt = new MessageImpl(\"editor-negative-prompt\", \"Negative prompt\");\nconst editorPrompt = new MessageImpl(\"editor-prompt\", \"Prompt\");\nconst emptyMyPrompt = new MessageImpl(\n \"empty-my-prompts\",\n 'You don\\'t have any My Prompt yet. Press the \"Add My Prompt\" button to add your My Prompt.'\n);\nconst inputPrompt = new MessageImpl(\"input-prompt\", \"Input prompt...\");\nconst loraNegativePromptError = new MessageImpl(\n \"lora-negative-prompt-error\",\n \"LoRA cannot be add to negative prompt\"\n);\nconst myPrompt = new MessageImpl(\"my-prompt\", \"My Prompt\");\nconst myPromptLabel = new MessageImpl(\"my-prompt-label\", \"Label\");\nconst myPromptPrompt = new MessageImpl(\"my-prompt-prompt\", \"Prompt\");\nconst myPromptTags = new MessageImpl(\"my-prompt-tags\", \"Tags\");\nconst myPromptTagsEmpty = new MessageImpl(\"my-prompt-tags-empty\", \"No tags are set\");\nconst pagenationEllipsis = new MessageImpl(\"pagenation-ellipsis\", \"…\");\nconst pagenationNext = new MessageImpl(\"pagenation-next\", \"Next\");\nconst pagenationPrevious = new MessageImpl(\"pagenation-previous\", \"Previous\");\nconst promptEdit = new MessageImpl(\"prompt-edit\", \"Prompt Edit\");\nconst searchMyPrompts = new MessageImpl(\"search-my-prompts\", \"Search my prompts...\");\nconst selectAndDeleteMyPrompt = new MessageImpl(\n \"select-and-delete-my-prompt\",\n \"Select and delete My Prompt\"\n);\nconst suggestFilterAll = new MessageImpl(\"suggest-filter-all\", \"All\");\nconst suggestFilterDanbooruCharacter = new MessageImpl(\n \"suggest-filter-danbooru-character\",\n \"Danbooru Tag (Character)\"\n);\nconst suggestFilterDanbooruCopyright = new MessageImpl(\n \"suggest-filter-danbooru-copyright\",\n \"Danbooru Tag (Copyright)\"\n);\nconst suggestFilterDanbooruGeneral = new MessageImpl(\n \"suggest-filter-danbooru-general\",\n \"Danbooru Tag (General)\"\n);\nconst suggestFilterLora = new MessageImpl(\"suggest-filter-lora\", \"LoRA\");\nconst suggestFilterMyPrompt = new MessageImpl(\"suggest-filter-my-prompt\", \"My Prompt\");\nconst suggestFilterTextualInversion = new MessageImpl(\n \"suggest-filter-textual-inversion\",\n \"Textual Inversion\"\n);\nconst weight = new MessageImpl(\"weight\", \"Weight\");\n\nexport {\n addNewMyPrompt,\n addThisMyPrompt,\n cancelAddingMyPrompt as cancelAndClose,\n cancelMyPromptDeletion,\n deleteSelectedMyPrompt,\n editorNegativePrompt,\n editorPrompt,\n emptyMyPrompt,\n inputPrompt,\n loraNegativePromptError,\n myPrompt,\n myPromptLabel,\n myPromptPrompt,\n myPromptTags,\n myPromptTagsEmpty,\n pagenationEllipsis,\n pagenationNext,\n pagenationPrevious,\n promptEdit,\n searchMyPrompts,\n selectAndDeleteMyPrompt,\n suggestFilterAll,\n suggestFilterDanbooruCharacter,\n suggestFilterDanbooruCopyright,\n suggestFilterDanbooruGeneral,\n suggestFilterLora,\n suggestFilterMyPrompt,\n suggestFilterTextualInversion,\n weight,\n};\n" }, { "path": "client-src/components/better-prompt/_logic/myPrompts.ts", "content": "import { type Readable, writable } from \"svelte/store\";\nimport type { MyPrompt } from \"@/libs/my-prompt\";\nimport { updateMyPrompts } from \"@/libs/api\";\n\nlet values: MyPrompt[] = [];\nconst _myPrompts = writable(values);\nexport const myPrompts: Readable = { subscribe: _myPrompts.subscribe };\n\nconst _allMyPromptTags = writable([]);\nexport const allMyPromptTags: Readable = { subscribe: _allMyPromptTags.subscribe };\n\n_myPrompts.subscribe((values) => {\n _allMyPromptTags.set([...new Set(values.flatMap((value) => value.tags))].sort());\n});\n\nlet init = false;\n\nexport function initMyPrompts(myPrompts: MyPrompt[]): void {\n if (init) return;\n init = true;\n\n values = [...myPrompts];\n _myPrompts.set(values);\n}\n\nexport function addMyPrompt(myPrompt: MyPrompt): void {\n if (!init) return;\n\n const newValues = [...values, myPrompt];\n void updateMyPrompts(newValues).then((success) => {\n if (!success) return;\n values = newValues;\n _myPrompts.set(values);\n });\n}\n\nexport function removeMyPrompts(myPrompts: MyPrompt[]): void {\n if (!init) return;\n\n const newValues = values.filter((value) => !myPrompts.includes(value));\n void updateMyPrompts(newValues).then((success) => {\n if (!success) return;\n values = newValues;\n _myPrompts.set(values);\n });\n}\n" }, { "path": "client-src/components/better-prompt/_logic/prompt.ts", "content": "import { type ExtraNetworksPrompt, type Prompt, toString } from \"@/libs/prompt\";\nimport {\n alternatePromptToString,\n emphasizedNegativePromptToString,\n emphasizedPositivePromptToString,\n plainPromptToString,\n promptCombinationToString,\n scheduledPromptToString,\n} from \"@/libs/prompt/util/toString\";\n\nexport function getTextContent(prompt: Prompt): string {\n return promptToText(prompt).replaceAll(\"\\\\\", \"\");\n}\n\nfunction promptToText(prompt: Prompt): string {\n switch (prompt.type) {\n case \"alternate\":\n return alternatePromptToString(prompt);\n case \"combination\":\n return promptCombinationToString(prompt);\n case \"emphasized-positive\":\n return emphasizedPositivePromptToString(prompt);\n case \"emphasized-negative\":\n return emphasizedNegativePromptToString(prompt);\n case \"emphasized-weighted\":\n return `${toString(prompt.values)}: ${prompt.weight}`;\n case \"extra-networks\":\n return extraNetworksPromptToText(prompt);\n case \"plain\":\n return plainPromptToString(prompt);\n case \"scheduled\":\n return scheduledPromptToString(prompt);\n }\n return \"\";\n}\n\nfunction extraNetworksPromptToText(prompt: ExtraNetworksPrompt): string {\n if (prompt.name === \"textual-inversion\") {\n return prompt.args[0];\n } else if (prompt.name === \"lora\") {\n return `${prompt.args[0]}: ${prompt.args[1]}`;\n }\n return toString(prompt);\n}\n\nexport function createDataset(prompt: Prompt): object {\n const dataset = {};\n dataset[\"data-prompt-type\"] = prompt.type;\n dataset[\"data-prompt-value\"] = toString(prompt);\n switch (prompt.type) {\n case \"extra-networks\":\n dataset[\"data-extra-networks\"] = prompt.name;\n break;\n }\n return dataset;\n}\n" }, { "path": "client-src/components/better-prompt/my-prompt/AddNewMyPrompt.svelte", "content": "\n\n\n\n\n
\n \n \n