API attemped fix

79a1b15600fa027527979d26003fac6046b9343a

WBlair1 <wolfgangblair1@gmail.com>

Parents: 38da3c22c0239a42110d72e94372d9834c67fa77

public/scripts/extensions/stable-diffusion/index.js+54 -21 Viewed

         $('#sd_model').val(extension_settings.sd.model).trigger('change');
+    }
+}
++
 async function generateStabilityImage(prompt, negativePrompt) {
 -    let endpoint;
 +    const payload = {
 +        prompt: prompt,
 +        negative_prompt: negativePrompt,
 +        width: extension_settings.sd.width,
 +        height: extension_settings.sd.height,
 +        seed: extension_settings.sd.seed >= 0 ? extension_settings.sd.seed : undefined,
 +        style_preset: extension_settings.sd.stability_style_preset,
 +        output_format: extension_settings.sd.stability_output_format,
 +    };
++
 +    const formData = new FormData();
 +    for (const [key, value] of Object.entries(payload)) {
 +        if (value !== undefined) {
 +            formData.append(key, String(value));
 +        }
 +    }
++
 +    let apiUrl;
     switch (extension_settings.sd.model) {
         case 'stable-image-ultra':
             endpointapiUrl = 'https://api.stability.ai/v2beta/stable-image/generate/ultra';
             break;
         case 'stable-image-core':
             endpointapiUrl = 'https://api.stability.ai/v2beta/stable-image/generate/core';
             break;
         case 'stable-diffusion-3':
             endpointapiUrl = 'https://api.stability.ai/v2beta/stable-image/generate/sd3';
             break;
         default:
             throw new Error('Invalid Stability AI model selected');
+    }
 -    const result = await fetch('/api/sd/stability/generate', {
 +    try {
 +        const response = await fetch(apiUrl, {
             method: 'POST',
 -        headers: getRequestHeaders(),
 +            headers: {
 -        body: JSON.stringify({
 +                'Authorization': `Bearer ${extension_settings.sd.stability_key}`,
 -            key: extension_settings.sd.stability_key,
 +                'Accept': 'image/*',
 -            endpoint: endpoint,
 +            },
             promptbody: promptformData,
 -            negative_prompt: negativePrompt,
 -            seed: extension_settings.sd.seed >= 0 ? extension_settings.sd.seed : undefined,
 -            style_preset: extension_settings.sd.stability_style_preset,
 -            aspect_ratio: extension_settings.sd.stability_aspect_ratio,
 -        }),
         });
             if (result!response.ok) {
                     const dataerrorText = await resultresponse.jsontext();
 -        return { format: extension_settings.sd.stability_output_format, data: data.image };
 +            throw new Error(`HTTP ${response.status}: ${errorText}`);
 -    } else {
 +        }
 -        const text = await result.text();
++
 -        throw new Error(text);
 +        const arrayBuffer = await response.arrayBuffer();
 +        const base64Image = arrayBufferToBase64(arrayBuffer);
++
 +        return {
 +            format: extension_settings.sd.stability_output_format,
 +            data: base64Image,
 +        };
 +    } catch (error) {
 +        console.error('Error generating image with Stability AI:', error);
 +        throw error;
 +    }
 +}
++
 +function arrayBufferToBase64(buffer) {
 +    let binary = '';
 +    const bytes = new Uint8Array(buffer);
 +    const len = bytes.byteLength;
 +    for (let i = 0; i < len; i++) {
 +        binary += String.fromCharCode(bytes[i]);
+    }
 +    return btoa(binary);
+}