forked from cocktailpeanutlabs/fooocus
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpinokio.js
60 lines (57 loc) · 2.59 KB
/
pinokio.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
const path = require("path")
module.exports = {
version: 1,
title: "Fooocus",
description: "Minimal Stable Diffusion UI",
icon: "icon.jpeg",
menu: async (kernel) => {
// exception handling for windows amd
let windowsAmd = (kernel.gpu === "amd" && kernel.platform === "win32")
let extraFlags = (windowsAmd ? " --directml --disable-in-browser" : " --disable-in-browser")
let installing = kernel.running(__dirname, "install.json")
let installed = await kernel.exists(__dirname, "app", "env")
if (installing) {
return [{ icon: "fa-solid fa-plug", text: "Installing...", href: "install.json" }]
} else if (installed) {
let running = kernel.running(__dirname, "start.json")
if (running) {
let memory = kernel.memory.local[path.resolve(__dirname, "start.json")]
if (memory && memory.url) {
return [
{ icon: "fa-solid fa-rocket", text: "Web UI", href: memory.url },
{ icon: "fa-solid fa-terminal", text: "Terminal", href: "start.json" },
{ icon: "fa-solid fa-rotate", text: "Update", href: "update.json" },
]
} else {
return [
{ icon: "fa-solid fa-terminal", text: "Terminal", href: "start.json" },
{ icon: "fa-solid fa-rotate", text: "Update", href: "update.json" },
]
}
} else {
return [{
icon: "fa-solid fa-power-off",
text: "Start",
menu: [
{ icon: "fa-solid fa-terminal", text: "Default Mode", href: "start.json", params: { flags: `--preset default${extraFlags}` } },
{ icon: "fa-solid fa-terminal", text: "Anime Mode", href: "start.json", params: { flags: `--preset anime${extraFlags}` } },
{ icon: "fa-solid fa-terminal", text: "Realistic Mode", href: "start.json", params: { flags: `--preset realistic${extraFlags}` } },
{ icon: "fa-solid fa-terminal", text: "SAI Mode", href: "start.json", params: { flags: `--preset sai${extraFlags}` } },
{ icon: "fa-solid fa-terminal", text: "LCM Mode", href: "start.json", params: { flags: `--preset lcm${extraFlags}` } },
]
}, {
icon: "fa-solid fa-rotate", text: "Update", href: "update.json"
}, {
icon: "fa-solid fa-plug", text: "Reinstall", href: "install.json"
}, {
icon: "fa-solid fa-broom", text: "Factory Reset", href: "reset.json"
}]
}
} else {
return [
{ icon: "fa-solid fa-plug", text: "Install", href: "install.json" },
{ icon: "fa-solid fa-rotate", text: "Update", href: "update.json" }
]
}
}
}