', token: "lora-end", next: "start" },
49 | { regex: /:\d+(\.\d+)?/, token: "attention" },
50 | { regex: /\w+/, token: "lora-inner" },
51 | ],
52 | hypernetwork: [
53 | { regex: '>', token: "hypernetwork-end", next: "start" },
54 | { regex: /:\d+(\.\d+)?/, token: "attention" },
55 | { regex: /\w+/, token: "hypernetwork-inner" },
56 | ],
57 | hypernet: [
58 | { regex: '>', token: "hypernet-end", next: "start" },
59 | { regex: /:\d+(\.\d+)?/, token: "attention" },
60 | { regex: /\w+/, token: "hypernet-inner" },
61 | ],
62 | lyco: [
63 | { regex: '>', token: "lyco-end", next: "start" },
64 | { regex: /:\d+(\.\d+)?/, token: "attention" },
65 | { regex: /\w+/, token: "lyco-inner" },
66 | ],
67 | inner: [
68 | {
69 | token: () => {
70 | bracketLevel++;
71 | return `open-bracket.open-bracket-${(bracketLevel) % 4}`;
72 | },
73 | regex: openBracket,
74 | next: "inner"
75 | },
76 | { regex: /:\d+(\.\d+)?/, token: "attention" },
77 | { regex: /[,|:]/, token: "token" },
78 | { regex: /__.+__/, token: "wildcard" },
79 | { regex: /\w+/, token: "inner-bracket" },
80 | {
81 | token: () => {
82 | bracketLevel--;
83 | return `close-bracket.close-bracket-${(bracketLevel+1) % 4}`;
84 | },
85 | regex: closeBracket,
86 | next: "start"
87 | },
88 | ]
89 | };
90 | }
91 |
92 | // Inherit from the base TextHighlightRules
93 | oop.inherits(CustomHighlightRules, TextHighlightRules);
94 |
95 | // Export the highlight rules
96 | exports.CustomHighlightRules = CustomHighlightRules;
97 | });
98 |
--------------------------------------------------------------------------------
/cozy-nest-client/cozy-prompt/useExternalTextareaObserver.js:
--------------------------------------------------------------------------------
1 | import { useState, useEffect } from 'react';
2 | import {CozyLogger} from "../main/CozyLogger.js";
3 |
4 | const useExternalTextareaObserver = (textareaSelector) => {
5 | const [value, setValue] = useState('');
6 |
7 | useEffect(() => {
8 | const observerCallback = (mutationsList) => {
9 | for (const mutation of mutationsList) {
10 | if (mutation.type === 'attributes') {
11 | const externalTextarea = document.querySelector(textareaSelector);
12 | setValue(externalTextarea.value);
13 | }
14 | }
15 | };
16 |
17 | const observerOptions = {
18 | attributes: true,
19 | characterData: true,
20 | childList: true,
21 | subtree: true,
22 | };
23 |
24 | const observer = new MutationObserver(observerCallback);
25 | const externalTextarea = document.querySelector(textareaSelector);
26 |
27 | if (externalTextarea) {
28 | observer.observe(externalTextarea, observerOptions);
29 | setValue(externalTextarea.value);
30 | }
31 |
32 | return () => {
33 | observer.disconnect();
34 | };
35 | }, [textareaSelector]);
36 |
37 | return value;
38 | };
39 |
40 | export default useExternalTextareaObserver;
41 |
--------------------------------------------------------------------------------
/cozy-nest-client/cozy-types.d.ts:
--------------------------------------------------------------------------------
1 | // type for image
2 | export type Image = {
3 | path: string,
4 | hash: string,
5 | metadata: {
6 | date: number
7 | exif: {
8 | parameters: string,
9 | 'cozy-nest-tags'?: string
10 | 'cozy-nest-hidden'?: string
11 | }
12 | }
13 | }
14 |
--------------------------------------------------------------------------------
/cozy-nest-client/cozy.build.js:
--------------------------------------------------------------------------------
1 | //post build script run by Nodejs
2 |
3 | //copy cozy-nest-style.css and cozy-nest-style-sdnext.css to cozy-nest-client/assets
4 | import {copyFile, readdir, rename } from 'fs/promises';
5 | import {join} from 'path';
6 | import { fileURLToPath } from 'url';
7 | import * as path from "path";
8 | import chalk from 'chalk';
9 |
10 | const log = console.log
11 | const relPath = (to) => path.relative(join(process.cwd(), '../'), to)
12 |
13 | log("")
14 | log(
15 | chalk.blue.bold("## "),
16 | chalk.blue("Cozy Nest post build script"),
17 | chalk.blue.bold(" ##"))
18 |
19 | const __filename = fileURLToPath(import.meta.url);
20 | const __dirname = path.dirname(__filename);
21 |
22 | const cozyNestStyleCss = join(__dirname, 'main/cozy-nest-style.css');
23 | const cozyNestStyleSdnextCss = join(__dirname, 'main/cozy-nest-style-sdnext.css');
24 |
25 | const cozyNestStyleCssDest = join(__dirname, '../client/assets/cozy-nest-style.css');
26 | const cozyNestStyleSdnextCssDest = join(__dirname, '../client/assets/cozy-nest-style-sdnext.css');
27 |
28 | const config = {
29 | folderCopy: [
30 | {
31 | from: join(__dirname, 'static'),
32 | to: join(__dirname, '../client/assets')
33 | }
34 | ],
35 | copy: [
36 | {
37 | from: join(__dirname, 'node_modules/ace-builds/src-noconflict/worker-json.js'),
38 | to: join(__dirname, '../client/assets/worker-json.js')
39 | },
40 | ],
41 | // move: [
42 | // {
43 | // from: join(__dirname, '../client/assets/cozy-nest.loader.min.js'),
44 | // to: join(__dirname, '../javascript/cozy-nest.loader.min.js')
45 | // }
46 | // ]
47 | }
48 |
49 |
50 | async function build() {
51 |
52 | config.folderCopy = config.folderCopy || [];
53 | config.copy = config.copy || [];
54 | config.move = config.move || [];
55 |
56 | // copy whole folders
57 | for (const folderCopy of config.folderCopy) {
58 |
59 | log(
60 | chalk.blue(`Copying folders`),
61 | chalk.blue.bold(` ${relPath(folderCopy.from)} → ${relPath(folderCopy.to)}`)
62 | )
63 |
64 | // list each file in the folder
65 | const files = await readdir(folderCopy.from);
66 | for (const file of files) {
67 | // copy each file to the destination
68 | await copyFile(
69 | path.join(folderCopy.from, file),
70 | path.join(folderCopy.to, file)
71 | );
72 | log(
73 | chalk.green(' COPY'),
74 | chalk.green.bold(` ${relPath(path.join(folderCopy.from, file))} → ${relPath(path.join(folderCopy.to, file))}`))
75 | }
76 | }
77 |
78 | // copy files
79 | log(
80 | chalk.blue(`Copying files`))
81 | for (const copy of config.copy) {
82 | await copyFile(copy.from, copy.to);
83 | log(
84 | chalk.green(' COPY'),
85 | chalk.green.bold(` ${relPath(copy.from)} → ${relPath(copy.to)}`))
86 | }
87 |
88 | // move file
89 | log(
90 | chalk.blue(`Moving files`))
91 | for (const move of config.move) {
92 | await rename(move.from, move.to);
93 | log(
94 | chalk.green(' MOVE'),
95 | chalk.green.bold(` ${relPath(move.from)} → ${relPath(move.to)}`))
96 | }
97 | }
98 | (async () => {
99 | await build();
100 | })();
101 |
102 |
103 |
104 |
--------------------------------------------------------------------------------
/cozy-nest-client/cozy_extra_network/CozyExtraNetworks.jsx:
--------------------------------------------------------------------------------
1 | import React, {useContext} from "react";
2 | import {useEffect} from "react";
3 | import {Loading} from "../image-browser/App.jsx";
4 | import {Checkbox, Tab, TabList, TabPanel, TabPanels, Tabs} from "@chakra-ui/react";
5 | import './CozyExtraNetworks.scss'
6 | import {ExtraNetworksCard} from "./ExtraNetworksCard.jsx";
7 | import {Column, Row, RowFullWidth} from "../main/Utils.jsx";
8 | import {SvgForReact} from "../main/svg_for_react.jsx";
9 | import {FolderTreeFilter} from "./FolderTreeFilter.jsx";
10 | import {FiRefreshCcw as RefreshIcon} from "react-icons/fi";
11 |
12 | const nevyshaScrollbar = {
13 | '&::-webkit-scrollbar': {
14 | width: '5px',
15 | },
16 | '&::-webkit-scrollbar-track': {
17 | backgroundColor: 'transparent'
18 | },
19 | '&::-webkit-scrollbar-thumb': {
20 | backgroundColor: 'var(--ae-primary-color)',
21 | borderRadius: '20px',
22 | },
23 | }
24 |
25 | const indexRef = []
26 |
27 | export function CozyExtraNetworks() {
28 |
29 | const [extraNetworks, setExtraNetworks] = React.useState([])
30 | const [folders, setFolders] = React.useState([])
31 | const [ready, setReady] = React.useState(false)
32 | const [fullyLoaded, setFullyLoaded] = React.useState(false)
33 |
34 | const [searchString, setSearchString] = React.useState('')
35 | const [displayFolderFilter, setDisplayFolderFilter] = React.useState({
36 | 'models': false,
37 | })
38 | const [nsfwFilter, setNsfwFilter] = React.useState(false)
39 |
40 | const [selectedTab, setSelectedTab] = React.useState(null)
41 |
42 | useEffect(() => {
43 | (async () => {
44 | await load()
45 | })();
46 | }, [])
47 |
48 | const load = async () => {
49 |
50 | const endpoint = COZY_NEST_CONFIG.deferred_cozy_extra_networks_loading ?
51 | '/cozy-nest/extra_networks' : '/cozy-nest/extra_networks/full'
52 |
53 | const response = await fetch(endpoint)
54 | if (response.status !== 200) {
55 | CozyLogger.error('failed to fetch extra networks', response)
56 | return;
57 | }
58 | const _enJson = await response.json()
59 |
60 | const _displayFolderFilter = {}
61 | Object.keys(_enJson).forEach((network, index) => {
62 | network = String(network);
63 | _displayFolderFilter[network] = false
64 | })
65 |
66 | const responseFolders = await fetch('/cozy-nest/extra_networks/folders')
67 | if (responseFolders.status !== 200) {
68 | CozyLogger.error('failed to fetch extra networks folders', responseFolders)
69 | return;
70 | }
71 | const _folders = await responseFolders.json()
72 |
73 | setFolders(_folders)
74 | setExtraNetworks(_enJson)
75 | setDisplayFolderFilter(_displayFolderFilter)
76 | setReady(true)
77 | if (!COZY_NEST_CONFIG.deferred_cozy_extra_networks_loading) {
78 | setFullyLoaded(true)
79 | }
80 | }
81 |
82 | const reload = async () => {
83 | setFolders([])
84 | setExtraNetworks([])
85 | setReady(false)
86 | setFullyLoaded(false)
87 | setSelectedTab(null)
88 |
89 | await load()
90 | }
91 |
92 |
93 |
94 | useEffect(() => {
95 |
96 | if (!nsfwFilter) return;
97 | if (fullyLoaded) return;
98 |
99 | setReady(false);
100 |
101 | (async () => {
102 | const response = await fetch('/cozy-nest/extra_networks/full')
103 | if (response.status === 200) {
104 | const json = await response.json()
105 | setExtraNetworks(json)
106 | setReady(true)
107 | setFullyLoaded(true)
108 | setSelectedTab(indexRef[0])
109 | }
110 | else {
111 | CozyLogger.error('failed to fetch full extra networks info', response)
112 | }
113 | })()
114 | }, [nsfwFilter])
115 |
116 | function buildExtraNetworks() {
117 |
118 | const EnTabs = [];
119 | const EnTabPanels = [];
120 |
121 | const style = {
122 | border: 'none',
123 | height: '100%',
124 | borderBottom: '1px solid var(--ae-input-border-color)',
125 | borderTop: '1px solid var(--tab-nav-background-color-selected)',
126 | };
127 |
128 | Object.keys(extraNetworks).forEach((network, index) => {
129 | let tabName = String(network);
130 | indexRef.push(network)
131 | if (network === 'embeddings') {
132 | tabName = 'Textual Inversion'
133 | }
134 | EnTabs.push(
135 | {tabName}
136 | )
137 | EnTabPanels.push(
138 |
139 |
140 | {extraNetworks[network].map((item, index) => {
141 | return (
142 |
148 | )
149 | })}
150 |
151 |
152 | )
153 | })
154 |
155 | return {EnTabs, EnTabPanels}
156 | }
157 |
158 | function onTabSelect(index) {
159 | setSelectedTab(indexRef[index])
160 | setSelectedFolder(null)
161 | }
162 |
163 | const [selectedFolder, setSelectedFolder] = React.useState(null)
164 | function folderSelectHandler({element}) {
165 | CozyLogger.debug('folderSelectHandler', {element})
166 |
167 | if (element.name === 'all' || !element.metadata) {
168 | setSelectedFolder(null)
169 | return
170 | }
171 | setSelectedFolder(element.metadata.path)
172 |
173 | }
174 |
175 | function changeFolderFilterForTab(selectedTabName, value) {
176 | displayFolderFilter[selectedTabName] = value
177 | setDisplayFolderFilter({...displayFolderFilter})
178 | }
179 |
180 | const Ui = buildExtraNetworks()
181 |
182 | const hasSubFolders = folders[selectedTab] && !folders[selectedTab].empty
183 |
184 | return (
185 |
186 | {!ready && }
187 | {ready &&
188 |
189 |