Update index.js

index.js

@@ -6,6 +6,7 @@ import axios from "axios";
 import { FormData, Blob } from "formdata-node";
 import { fileTypeFromBuffer } from "file-type";
 import { Client } from "@gradio/client";
+import { stablediff } from "./lib/diffusion.js"
 
 const app = express();
 app.set('json spaces', 4);
@@ -95,6 +96,26 @@ app.post('/api/openai/gpt4', async (req, res) => {
 	}
 })
 
+app.post('/api/stabeldiff', async (req, res) => {
+	try {
+		console.log(req.body)
+		const { prompt, status } = req.body
+        if (!prompt) return res.json({ succese: false, message: 'Require an Promot text!'})
+        if (!status) return res.json({ success: false, message: 'Required an status text!' })
+
+        if(status !== apikey) return res.json({ success: false, message: 'Invalid status!' })
+        const response = await stablediff(prompt);
+        const type_img = await fileTypeFromBuffer(response[0]);
+        
+        res.setHeader('Content-Type', type_img.mime);
+        res.send(response[0]);
+	} catch (e) {
+		console.log(e)
+		e = String(e)
+		res.json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
+	}
+})
+
 const PORT = process.env.PORT || 7860
 app.listen(PORT, () => {
   console.log('App running on port', PORT)