Update server.js

server.js

@@ -653,32 +653,28 @@ app.get('/api/generate/flux', authenticateToken, async (req, res) => {
                     await new Promise(resolve => setTimeout(resolve, 2000));
                     continue;
                 }
-                const reader = response.body.getReader();
-                let done = false;
-                while (!done) {
-                    const { value, done: streamDone } = await reader.read();
-                    if (value) {
-                        buffer += Buffer.from(value).toString('utf8');
-                        const lines = buffer.split('\n');
-                        for (const line of lines) {
-                            if (line.startsWith('data: ')) {
-                                try {
-                                    const json = JSON.parse(line.slice(6));
-                                    console.log('[Flux] Event:', json);
-                                    if (json.msg === 'process_completed' && json.output?.data?.[0]?.url) {
-                                        const imageUrl = json.output.data[0].url;
-                                        console.log(`[Flux] Success! Image URL: ${imageUrl}`);
-                                        found = true;
-                                        return res.json({ success: true, imageUrl });
-                                    }
-                                } catch (e) {
-                                    console.error('[Flux] Parse error:', e, line);
+                // Use Node.js stream reading
+                for await (const chunk of response.body) {
+                    buffer += chunk.toString('utf8');
+                    const lines = buffer.split('\n');
+                    for (let i = 0; i < lines.length - 1; i++) {
+                        const line = lines[i];
+                        if (line.startsWith('data: ')) {
+                            try {
+                                const json = JSON.parse(line.slice(6));
+                                console.log('[Flux] Event:', json);
+                                if (json.msg === 'process_completed' && json.output?.data?.[0]?.url) {
+                                    const imageUrl = json.output.data[0].url;
+                                    console.log(`[Flux] Success! Image URL: ${imageUrl}`);
+                                    found = true;
+                                    return res.json({ success: true, imageUrl });
                                 }
+                            } catch (e) {
+                                console.error('[Flux] Parse error:', e, line);
                             }
                         }
-                        buffer = lines[lines.length - 1]; // keep incomplete line
                     }
-                    done = streamDone;
+                    buffer = lines[lines.length - 1]; // keep incomplete line
                 }
             } catch (pollError) {
                 console.error(`[Flux] Poll error on attempt ${attempts}:`, pollError);