Make generation actually work

5da3921907df255025d18514f4466625480c2518

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

2 files changed, +4 -5Showing whitespace changes
public/scripts/extensions/stable-diffusion/index.js+2 -4
@@ -2725,12 +2725,10 @@ async function generateStabilityImage(prompt, negativePrompt) {
2725 });2725 });
27262726
2727 if (!response.ok) {2727 if (!response.ok) {
2728 const errorText = await response.text();2728 throw new Error(`HTTP ${response.status}: ${response.statusText}`);
2729 throw new Error(`HTTP ${response.status}: ${errorText}`);
2730 }2729 }
27312730
2732 const blob = await response.blob();2731 const base64Image = await response.text();
2733 const base64Image = await getBase64Async(blob);
27342732
2735 return {2733 return {
2736 format: IMAGE_FORMAT,2734 format: IMAGE_FORMAT,
src/endpoints/stable-diffusion.js+2 -1
@@ -836,6 +836,7 @@ stability.post('/generate', jsonParser, async (request, response) => {
836 'Accept': 'image/*',836 'Accept': 'image/*',
837 },837 },
838 body: formData,838 body: formData,
839 timeout: 0,
839 });840 });
840841
841 if (!result.ok) {842 if (!result.ok) {
@@ -845,7 +846,7 @@ stability.post('/generate', jsonParser, async (request, response) => {
845 }846 }
846847
847 const buffer = await result.buffer();848 const buffer = await result.buffer();
848 return response.send(buffer);849 return response.send(buffer.toString('base64'));
849 } catch (error) {850 } catch (error) {
850 console.log(error);851 console.log(error);
851 return response.sendStatus(500);852 return response.sendStatus(500);