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	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	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';
         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';
         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';
         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';
         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';
         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';
         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';
                 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';
         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.search = `?filename=${imgInfo.filename}&subfolder=${imgInfo.subfolder}&type=${imgInfo.type}`;
         const imgResponse = await fetch(imgUrl);
         if (!imgResponse.ok) {