Update server.js

server.js

@@ -570,13 +570,14 @@ app.post('/api/generate/anime-uc', authenticateToken, async (req, res) => {
     }
 });
 
-// Flux Generation endpoint (now uses UNO-FLUX model)
+// Flux Generation endpoint (now uses UNO-FLUX queue API)
 app.get('/api/generate/flux', authenticateToken, async (req, res) => {
     try {
         const { prompt, aspect } = req.query;
         if (!prompt) {
             return res.status(400).json({ success: false, error: 'Prompt is required' });
         }
+
         // Aspect ratio logic
         let width = 1280, height = 720;
         if (aspect === '9:16') {
@@ -585,39 +586,79 @@ app.get('/api/generate/flux', authenticateToken, async (req, res) => {
             width = 1024; height = 1024;
         }
 
-        // Use a default image for all required image_prompt1-4
-        const defaultImageUrl = 'https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png';
-        async function fetchImageBlob(url) {
-            const response = await fetch(url);
-            if (!response.ok) throw new Error('Failed to fetch image: ' + url);
-            return await response.blob();
-        }
-        const [img1, img2, img3, img4] = await Promise.all([
-            fetchImageBlob(defaultImageUrl),
-            fetchImageBlob(defaultImageUrl),
-            fetchImageBlob(defaultImageUrl),
-            fetchImageBlob(defaultImageUrl)
-        ]);
+        const sessionHash = Math.random().toString(36).substring(2, 15);
 
-        // Connect to UNO-FLUX Gradio client
-        const client = await Client.connect("bytedance-research/UNO-FLUX");
-        const result = await client.predict("/gradio_generate", {
-            prompt,
-            width,
-            height,
-            guidance: 4,
-            num_steps: 25,
-            seed: -1,
-            image_prompt1: img1,
-            image_prompt2: img2,
-            image_prompt3: img3,
-            image_prompt4: img4
-        });
+        // 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/'
+                }
+            }
+        );
+
+        // Poll for results
+        let attempts = 0;
+        const maxAttempts = 60;
+        while (attempts < maxAttempts) {
+            try {
+                const dataResponse = await axios.get(
+                    `https://bytedance-research-uno-flux.hf.space/gradio_api/queue/data?session_hash=${encodeURIComponent(sessionHash)}`,
+                    {
+                        headers: {
+                            'Accept': 'text/event-stream',
+                            'Referer': 'https://bytedance-research-uno-flux.hf.space/'
+                        },
+                        timeout: 10000
+                    }
+                );
 
-        // result.data[0] is the generated image URL
-        res.json({ success: true, imageUrl: result.data[0] });
+                const data = dataResponse.data;
+                // 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);
+                                if (json.msg === 'process_completed' && json.output?.data?.[0]?.url) {
+                                    const imageUrl = json.output.data[0].url;
+                                    return res.json({ success: true, imageUrl });
+                                }
+                            } catch (error) {}
+                            eventData = '';
+                        }
+                    }
+                }
+            } catch (pollError) {}
+            await new Promise(resolve => setTimeout(resolve, 2000));
+            attempts++;
+        }
+        res.status(500).json({ success: false, error: 'Generation timed out' });
     } catch (error) {
-        console.error('[Flux] UNO-FLUX endpoint error:', error);
         res.status(500).json({ success: false, error: error.message || 'Failed to generate image' });
     }
 });