Update server.js

server.js

@@ -210,6 +210,9 @@ app.post('/api/generate', async (req, res) => {
 
 // Flux model generation endpoint using black-forest-labs/FLUX.1-dev Gradio client
 app.post('/api/generate/flux', async (req, res) => {
+    res.setHeader('Content-Type', 'text/event-stream');
+    res.setHeader('Cache-Control', 'no-cache');
+    res.setHeader('Connection', 'keep-alive');
     try {
         const {
             prompt = '',
@@ -222,11 +225,18 @@ app.post('/api/generate/flux', async (req, res) => {
         } = req.body;
 
         if (!prompt) {
-            return res.status(400).json({ success: false, error: 'Prompt is required' });
+            res.write(`data: ${JSON.stringify({ type: 'error', message: 'Prompt is required' })}\n\n`);
+            res.end();
+            return;
         }
 
+        // Simulate estimation and processing events for UI feedback
+        res.write(`data: ${JSON.stringify({ type: 'estimation', queueSize: 1, eta: 5 })}\n\n`);
+        await new Promise(resolve => setTimeout(resolve, 1000));
+        res.write(`data: ${JSON.stringify({ type: 'processing' })}\n\n`);
+
         // Connect to FLUX.1-dev Gradio client
-        const client = await Client.connect("black-forest-labs/FLUX.1-schnell");
+        const client = await Client.connect("black-forest-labs/FLUX.1-dev");
         const result = await client.predict("/infer", {
             prompt,
             seed,
@@ -237,10 +247,18 @@ app.post('/api/generate/flux', async (req, res) => {
             num_inference_steps
         });
 
-        // result.data: [imageUrl, seed]
-        res.json({ success: true, image: result.data[0], seed: result.data[1] });
+        // result.data: [imageObj, seed]
+        const imageObj = result.data[0];
+        const imageUrl = (imageObj && imageObj.url) ? imageObj.url : (typeof imageObj === 'string' ? imageObj : null);
+        if (imageUrl) {
+            res.write(`data: ${JSON.stringify({ type: 'success', imageUrl, seed: result.data[1] })}\n\n`);
+        } else {
+            res.write(`data: ${JSON.stringify({ type: 'error', message: 'No image URL returned' })}\n\n`);
+        }
+        res.end();
     } catch (err) {
-        res.status(500).json({ success: false, error: err.message });
+        res.write(`data: ${JSON.stringify({ type: 'error', message: err.message })}\n\n`);
+        res.end();
     }
 });