Update server.js

server.js

@@ -570,7 +570,7 @@ app.post('/api/generate/anime-uc', authenticateToken, async (req, res) => {
     }
 });
 
-// Flux Generation endpoint (UNO-FLUX direct API with detailed logging)
+// Flux Generation endpoint (UNO-FLUX direct API with node-fetch SSE polling)
 app.get('/api/generate/flux', authenticateToken, async (req, res) => {
     try {
         const { prompt, aspect } = req.query;
@@ -612,71 +612,77 @@ app.get('/api/generate/flux', authenticateToken, async (req, res) => {
         console.log('[Flux] Sending queue/join payload:', JSON.stringify(joinPayload));
         let joinResponse;
         try {
-            joinResponse = await axios.post(
+            joinResponse = await fetch(
                 'https://bytedance-research-uno-flux.hf.space/gradio_api/queue/join?__theme=system',
-                joinPayload,
                 {
+                    method: 'POST',
                     headers: {
                         'Content-Type': 'application/json',
-                        'Referer': 'https://bytedance-research-uno-flux.hf.space/'
-                    }
+                        'Referer': 'https://bytedance-research-uno-flux.hf.space/',
+                        'Origin': 'https://bytedance-research-uno-flux.hf.space'
+                    },
+                    body: JSON.stringify(joinPayload)
                 }
             );
-            console.log('[Flux] queue/join response:', joinResponse.data);
+            const joinData = await joinResponse.json();
+            console.log('[Flux] queue/join response:', joinData);
         } catch (err) {
-            console.error('[Flux] Error joining UNO-FLUX queue:', err.response?.data || err.message);
+            console.error('[Flux] Error joining UNO-FLUX queue:', err);
             return res.status(500).json({ success: false, error: 'Failed to join UNO-FLUX queue' });
         }
 
-        // Poll for results
-        let attempts = 0;
-        const maxAttempts = 60;
+        // Poll for results using node-fetch and streaming
         const pollUrl = `https://bytedance-research-uno-flux.hf.space/gradio_api/queue/data?session_hash=${encodeURIComponent(sessionHash)}`;
         console.log(`[Flux] Polling for results at: ${pollUrl}`);
         let found = false;
-        while (attempts < maxAttempts) {
+        let attempts = 0;
+        const maxAttempts = 60;
+        let buffer = '';
+        while (attempts < maxAttempts && !found) {
             attempts++;
             try {
-                const dataResponse = await axios.get(
-                    pollUrl,
-                    {
-                        headers: {
-                            'Accept': 'text/event-stream',
-                            'Referer': 'https://bytedance-research-uno-flux.hf.space/'
-                        },
-                        timeout: 10000
-                    }
-                );
-                const data = dataResponse.data;
-                console.log(`[Flux] Poll attempt ${attempts}, data length: ${data.length}`);
-                // Parse the event stream data
-                const lines = data.split('\n');
-                let eventData = '';
-                for (const line of lines) {
-                    if (line.startsWith('data: ')) {
-                        eventData += line.substring(6);
-                    } else if (line === '') {
-                        if (eventData) {
-                            try {
-                                const json = JSON.parse(eventData);
-                                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 });
-                                } else if (json.msg === 'process_completed') {
-                                    console.warn('[Flux] process_completed but no image URL:', json);
+                const response = await fetch(pollUrl, {
+                    headers: {
+                        'Accept': 'text/event-stream',
+                        'Referer': 'https://bytedance-research-uno-flux.hf.space/',
+                        'Origin': 'https://bytedance-research-uno-flux.hf.space'
+                    },
+                    timeout: 10000
+                });
+                if (!response.ok) {
+                    console.error(`[Flux] Poll error on attempt ${attempts}:`, response.status, await response.text());
+                    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);
                                 }
-                            } catch (error) {
-                                console.error('[Flux] Error parsing event data:', error, 'Raw data:', eventData);
                             }
-                            eventData = '';
                         }
+                        buffer = lines[lines.length - 1]; // keep incomplete line
                     }
+                    done = streamDone;
                 }
             } catch (pollError) {
-                console.error(`[Flux] Poll error on attempt ${attempts}:`, pollError.message);
+                console.error(`[Flux] Poll error on attempt ${attempts}:`, pollError);
             }
             await new Promise(resolve => setTimeout(resolve, 2000));
         }