Skip to content

Commit e97ec16

Browse files
authored
v1.01b
v1.01b
2 parents 8e33cee + 8a84eb1 commit e97ec16

File tree

20 files changed

+978
-1684
lines changed

20 files changed

+978
-1684
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
env/

WebUI/electron/electron-env.d.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ type LocalSettings = {
3535
settingPath: string;
3636
debug: number;
3737
envType: string;
38+
port:number;
3839
} & KVObject;
3940

4041
type ModelPaths = {
@@ -57,6 +58,7 @@ type ModelLists = {
5758
} & { [key: string]: Array<string> }
5859

5960
type SetupData = {
61+
apiHost:string,
6062
modelPaths: ModelPaths,
6163
modelLists: ModelLists,
6264
envType: string,

WebUI/electron/main.ts

Lines changed: 22 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,8 @@ const settings: LocalSettings = {
6767
settingPath: "",
6868
isAdminExec: false,
6969
debug: 0,
70-
envType: "ultra"
70+
envType: "ultra",
71+
port: 59999
7172
};
7273

7374

@@ -86,6 +87,7 @@ function loadSettings() {
8687
}
8788
});
8889
}
90+
settings.apiHost = `http://127.0.0.1:${settings.port}`;
8991
}
9092

9193
async function createWindow() {
@@ -263,14 +265,14 @@ function initEventHandle() {
263265
ipcMain.handle(
264266
"restorePathsSettings",
265267
(event: IpcMainInvokeEvent) => {
266-
const paths = app.isPackaged ? {
268+
const paths = app.isPackaged ? {
267269
"llm": "./resources/service/models/llm/checkpoints",
268270
"embedding": "./resources/service/models/llm/embedding",
269271
"stableDiffusion": "./resources/service/models/stable_diffusion/checkpoints",
270272
"inpaint": "./resources/service/models/stable_diffusion/inpaint",
271273
"lora": "./resources/service/models/stable_diffusion/lora",
272274
"vae": "./resources/service/models/stable_diffusion/vae"
273-
}:{
275+
} : {
274276
"llm": "../service/models/llm/checkpoints",
275277
"embedding": "../service/models/llm/embedding",
276278
"stableDiffusion": "../service/models/stable_diffusion/checkpoints",
@@ -362,11 +364,12 @@ function initEventHandle() {
362364
const win = BrowserWindow.fromWebContents(event.sender);
363365
if (!win) { return; }
364366
return {
367+
apiHost:settings.apiHost,
365368
modelLists: pathsManager.sacanAll(),
366369
modelPaths: pathsManager.modelPaths,
367370
envType: settings.envType,
368371
isAdminExec: settings.isAdminExec,
369-
version :app.getVersion()
372+
version: app.getVersion()
370373
};
371374

372375
});
@@ -451,25 +454,27 @@ function isProcessRunning(pid: number) {
451454
function wakeupApiService() {
452455
const wordkDir = path.resolve(app.isPackaged ? path.join(process.resourcesPath, "service") : path.join(__dirname, "../../../service"));
453456
const baseDir = app.isPackaged ? process.resourcesPath : path.join(__dirname, "../../../");
454-
const pytonExe = path.resolve(path.join(baseDir, "env/python.exe"));
457+
const pythonExe = path.resolve(path.join(baseDir, "env/python.exe"));
455458
const newEnv = {
456459
"SYCL_ENABLE_DEFAULT_CONTEXTS": "1",
457460
"SYCL_CACHE_PERSISTENT": "1",
458461
"PYTHONIOENCODING": "utf-8"
459462
};
460463

461-
const options = settings.debug ? {
462-
cwd: wordkDir,
463-
detached: true,
464-
windowsHide: false,
465-
env: Object.assign(process.env, newEnv)
466-
} : {
467-
cwd: wordkDir,
468-
windowsHide: true,
469-
env: Object.assign(process.env, newEnv)
464+
if (settings.debug) {
465+
apiService.webProcess = spawn("cmd.exe", ["/c", pythonExe, "web_api.py", "--port", settings.port.toString()], {
466+
cwd: wordkDir,
467+
detached: true,
468+
windowsHide: false,
469+
env: Object.assign(process.env, newEnv)
470+
});
471+
} else {
472+
apiService.webProcess = spawn(pythonExe, ["web_api.py", "--port", settings.port.toString()], {
473+
cwd: wordkDir,
474+
windowsHide: true,
475+
env: Object.assign(process.env, newEnv)
476+
});
470477
}
471-
472-
apiService.webProcess = spawn("cmd.exe", ["/c", pytonExe, "web_api.py"], options);
473478
}
474479

475480
function closeApiService() {
@@ -584,4 +589,4 @@ app.whenReady().then(async () => {
584589
createWindow();
585590
wakeupApiService();
586591
}
587-
});
592+
});

WebUI/external/settings-dev.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
{
22
"envType": "arc",
3-
"debug":1
3+
"debug":1,
4+
"port":56789
45
}

0 commit comments

Comments
 (0)