Update server.js

server.js

@@ -570,11 +570,12 @@ app.post('/api/generate/anime-uc', authenticateToken, async (req, res) => {
     }
 });
 
-// Flux Generation endpoint (now uses UNO-FLUX queue API)
+// Flux Generation endpoint (UNO-FLUX direct API with detailed logging)
 app.get('/api/generate/flux', authenticateToken, async (req, res) => {
     try {
         const { prompt, aspect } = req.query;
         if (!prompt) {
+            console.error('[Flux] No prompt provided');
             return res.status(400).json({ success: false, error: 'Prompt is required' });
         }
 
@@ -587,43 +588,57 @@ app.get('/api/generate/flux', authenticateToken, async (req, res) => {
         }
 
         const sessionHash = Math.random().toString(36).substring(2, 15);
+        console.log(`[Flux] Starting UNO-FLUX generation for prompt: "${prompt}" | aspect: ${aspect} | width: ${width} | height: ${height} | session: ${sessionHash}`);
 
         // Join the queue using UNO-FLUX API
-        const joinResponse = await axios.post(
-            'https://bytedance-research-uno-flux.hf.space/gradio_api/queue/join?__theme=system',
-            {
-                data: [
-                    prompt,
-                    width,
-                    height,
-                    4,      // guidance
-                    25,     // num_steps
-                    -1,     // seed
-                    null,   // image_prompt1
-                    null,   // image_prompt2
-                    null,   // image_prompt3
-                    null    // image_prompt4
-                ],
-                event_data: null,
-                fn_index: 0,
-                trigger_id: 25,
-                session_hash: sessionHash
-            },
-            {
-                headers: {
-                    'Content-Type': 'application/json',
-                    'Referer': 'https://bytedance-research-uno-flux.hf.space/'
+        const joinPayload = {
+            data: [
+                prompt,
+                width,
+                height,
+                4,      // guidance
+                25,     // num_steps
+                -1,     // seed
+                null,   // image_prompt1
+                null,   // image_prompt2
+                null,   // image_prompt3
+                null    // image_prompt4
+            ],
+            event_data: null,
+            fn_index: 0,
+            trigger_id: 25,
+            session_hash: sessionHash
+        };
+        console.log('[Flux] Sending queue/join payload:', JSON.stringify(joinPayload));
+        let joinResponse;
+        try {
+            joinResponse = await axios.post(
+                'https://bytedance-research-uno-flux.hf.space/gradio_api/queue/join?__theme=system',
+                joinPayload,
+                {
+                    headers: {
+                        'Content-Type': 'application/json',
+                        'Referer': 'https://bytedance-research-uno-flux.hf.space/'
+                    }
                 }
-            }
-        );
+            );
+            console.log('[Flux] queue/join response:', joinResponse.data);
+        } catch (err) {
+            console.error('[Flux] Error joining UNO-FLUX queue:', err.response?.data || err.message);
+            return res.status(500).json({ success: false, error: 'Failed to join UNO-FLUX queue' });
+        }
 
         // Poll for results
         let attempts = 0;
         const maxAttempts = 60;
+        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) {
+            attempts++;
             try {
                 const dataResponse = await axios.get(
-                    `https://bytedance-research-uno-flux.hf.space/gradio_api/queue/data?session_hash=${encodeURIComponent(sessionHash)}`,
+                    pollUrl,
                     {
                         headers: {
                             'Accept': 'text/event-stream',
@@ -632,8 +647,8 @@ app.get('/api/generate/flux', authenticateToken, async (req, res) => {
                         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 = '';
@@ -644,21 +659,33 @@ app.get('/api/generate/flux', authenticateToken, async (req, res) => {
                         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);
                                 }
-                            } catch (error) {}
+                            } catch (error) {
+                                console.error('[Flux] Error parsing event data:', error, 'Raw data:', eventData);
+                            }
                             eventData = '';
                         }
                     }
                 }
-            } catch (pollError) {}
+            } catch (pollError) {
+                console.error(`[Flux] Poll error on attempt ${attempts}:`, pollError.message);
+            }
             await new Promise(resolve => setTimeout(resolve, 2000));
-            attempts++;
         }
-        res.status(500).json({ success: false, error: 'Generation timed out' });
+        if (!found) {
+            console.error(`[Flux] Generation timed out after ${maxAttempts} attempts`);
+            res.status(500).json({ success: false, error: 'Generation timed out' });
+        }
     } catch (error) {
+        console.error('[Flux] Fatal error in endpoint:', error);
         res.status(500).json({ success: false, error: error.message || 'Failed to generate image' });
     }
 });