Update index.js

index.js

@@ -135,6 +135,51 @@ app.get('/yt/convert', async (req, res) => {
 	}
 })
 
+app.all('/stablediff/illusion', async (req, res) => {
+	if (!['GET', 'POST'].includes(req.method)) return res.json({ success: false, message: 'Method Not Allowed' })
+	try {
+		const {
+			prompt,
+			negative_prompt = 'low quality',
+			image_url,
+			num_inference_steps = 25,
+			controlnet_conditioning_scale = 1
+		} = req.method !== 'GET' ? req.body : req.query
+		
+		if (!(prompt && image_url)) return res.json({ success: false, message: 'Required parameter prompt & image_url' })
+		
+		const headers = {
+			'Content-Type': 'application/json',
+			Authorization: `Key ${process.env.falAIKey}`
+		}
+		
+		const { detail, response_url } = await fetch('https://54285744-illusion-diffusion.gateway.alpha.fal.ai/fal/queue/submit', {
+			method: 'POST',
+			body: JSON.stringify({
+				prompt, negative_prompt, image_url,
+				num_inference_steps, controlnet_conditioning_scale
+			}),
+			headers
+		})
+		if (detail) return res.json({ success: false, message: detail })
+		
+		let retry = 0
+		while (true) {
+			await new Promise(resolve => setTimeout(resolve, 2500))
+			const prediction = await fetch(response_url, { headers })
+			console.log(prediction)
+			if (retry > 10) return res.json({ success: false, message: prediction.detail || 'Max retry has reached' })
+			if (prediction.image) return res.json({ success: true, result: prediction })
+			retry += 1
+		}
+		
+	} catch (e) {
+		console.log(e)
+		e = String(e)
+		res.status(500).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))