Support ComfyUI hosted on relative URLs Closes #3208

5739efc59b2233374d2fbb7ac7be242f1cad0b22

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/scripts/extensions/stable-diffusion/index.js+2 -1 Viewed

@@ -1111,7 +1111,8 @@ function onHrSecondPassStepsInput() {
1111	}	1111	}
1112		1112
1113	function onComfyUrlInput() {	1113	function onComfyUrlInput() {
1114	extension_settings.sd.comfy_url = $('#sd_comfy_url').val();	1114	// Remove trailing slashes
		1115	extension_settings.sd.comfy_url = String($('#sd_comfy_url').val() ?? '').replace(/\/$/, '');
1115	saveSettingsDebounced();	1116	saveSettingsDebounced();
1116	}	1117	}
1117		1118

src/endpoints/stable-diffusion.js+9 -9 Viewed

 comfy.post('/ping', jsonParser, async (request, response) => {
     try {
         const url = new URL(request.body.url);
-        url.pathname = '/system_stats';
+        url.pathname += '/system_stats';
         const result = await fetch(url);
         if (!result.ok) {

 comfy.post('/samplers', jsonParser, async (request, response) => {
     try {
         const url = new URL(request.body.url);
-        url.pathname = '/object_info';
+        url.pathname += '/object_info';
         const result = await fetch(url);
         if (!result.ok) {

 comfy.post('/models', jsonParser, async (request, response) => {
     try {
         const url = new URL(request.body.url);
-        url.pathname = '/object_info';
+        url.pathname += '/object_info';
         const result = await fetch(url);
         if (!result.ok) {

 comfy.post('/schedulers', jsonParser, async (request, response) => {
     try {
         const url = new URL(request.body.url);
-        url.pathname = '/object_info';
+        url.pathname += '/object_info';
         const result = await fetch(url);
         if (!result.ok) {

 comfy.post('/vaes', jsonParser, async (request, response) => {
     try {
         const url = new URL(request.body.url);
-        url.pathname = '/object_info';
+        url.pathname += '/object_info';
         const result = await fetch(url);
         if (!result.ok) {

 comfy.post('/generate', jsonParser, async (request, response) => {
     try {
         const url = new URL(request.body.url);
-        url.pathname = '/prompt';
+        url.pathname += '/prompt';
         const controller = new AbortController();
         request.socket.removeAllListeners('close');
         request.socket.on('close', function () {
             if (!response.writableEnded && !item) {
                 const interruptUrl = new URL(request.body.url);
-                interruptUrl.pathname = '/interrupt';
+                interruptUrl.pathname += '/interrupt';
                 fetch(interruptUrl, { method: 'POST', headers: { 'Authorization': getBasicAuthHeader(request.body.auth) } });
+            }
             controller.abort();

         const id = data.prompt_id;
         let item;
         const historyUrl = new URL(request.body.url);
-        historyUrl.pathname = '/history';
+        historyUrl.pathname += '/history';
         while (true) {
             const result = await fetch(historyUrl);
             if (!result.ok) {

+        }
         const imgInfo = Object.keys(item.outputs).map(it => item.outputs[it].images).flat()[0];
         const imgUrl = new URL(request.body.url);
-        imgUrl.pathname = '/view';
+        imgUrl.pathname += '/view';
         imgUrl.search = `?filename=${imgInfo.filename}&subfolder=${imgInfo.subfolder}&type=${imgInfo.type}`;
         const imgResponse = await fetch(imgUrl);
         if (!imgResponse.ok) {