Rename server.js to app.py

app.py

@@ -0,0 +1,264 @@
+import random
+import string
+import requests
+from flask import Flask, request, Response, stream_with_context, send_file
+from io import BytesIO
+
+app = Flask(__name__)
+
+# --- Utility functions ---
+def random_user_agent():
+    agents = [
+        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
+        'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Safari/605.1.15',
+        'Mozilla/5.0 (Linux; Android 13; SM-G991B) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Mobile Safari/537.36',
+        'Mozilla/5.0 (iPhone; CPU iPhone OS 17_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Mobile/15E148 Safari/604.1',
+        'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36'
+    ]
+    return random.choice(agents)
+
+def random_ip():
+    return '.'.join(str(random.randint(0, 255)) for _ in range(4))
+
+def random_session_hash():
+    return ''.join(random.choices(string.ascii_lowercase + string.digits, k=12))
+
+# --- SSE Streaming Helper ---
+def sse_stream(generator):
+    return Response(stream_with_context(generator), mimetype='text/event-stream')
+
+# --- UNO-FLUX Endpoint ---
+@app.route('/api/generate/flux', methods=['POST'])
+def generate_flux():
+    data = request.json or {}
+    prompt = data.get('prompt', '')
+    width = int(data.get('width', 1280))
+    height = int(data.get('height', 720))
+    guidance = data.get('guidance', 4)
+    num_steps = data.get('num_steps', 25)
+    seed = data.get('seed', -1)
+    image_prompt1 = data.get('image_prompt1')
+    image_prompt2 = data.get('image_prompt2')
+    image_prompt3 = data.get('image_prompt3')
+    image_prompt4 = data.get('image_prompt4')
+    aspect = data.get('aspect', '16:9')
+    if not prompt:
+        return sse_stream(lambda: iter([f"data: {{\"type\":\"error\",\"message\":\"Prompt is required\"}}\n\n"]))
+    # Aspect ratio logic
+    if aspect == '9:16':
+        width, height = 720, 1280
+    elif aspect == '1:1':
+        width, height = 1024, 1024
+    session_hash = random_session_hash()
+    join_payload = {
+        "data": [
+            prompt,
+            width,
+            height,
+            guidance,
+            num_steps,
+            seed,
+            image_prompt1,
+            image_prompt2,
+            image_prompt3,
+            image_prompt4
+        ],
+        "event_data": None,
+        "fn_index": 0,
+        "trigger_id": 25,
+        "session_hash": session_hash
+    }
+    def event_gen():
+        # Join queue
+        try:
+            join_resp = requests.post(
+                'https://bytedance-research-uno-flux.hf.space/gradio_api/queue/join?__theme=system',
+                json=join_payload,
+                headers={
+                    'Content-Type': 'application/json',
+                    'Referer': 'https://bytedance-research-uno-flux.hf.space/',
+                    'Origin': 'https://bytedance-research-uno-flux.hf.space',
+                    'User-Agent': random_user_agent(),
+                    'X-Forwarded-For': random_ip()
+                },
+                timeout=30
+            )
+            if join_resp.status_code != 200:
+                yield f"data: {{\"type\":\"error\",\"message\":\"Failed to join UNO-FLUX queue\",\"status\":{join_resp.status_code}}}\n\n"
+                return
+        except Exception as e:
+            yield f"data: {{\"type\":\"error\",\"message\":\"Failed to join UNO-FLUX queue\",\"details\":\"{str(e)}\"}}\n\n"
+            return
+        # Poll for results
+        max_attempts = 90
+        for attempt in range(max_attempts):
+            try:
+                poll_resp = requests.get(
+                    f'https://bytedance-research-uno-flux.hf.space/gradio_api/queue/data?session_hash={session_hash}',
+                    headers={
+                        'Accept': 'text/event-stream',
+                        'Referer': 'https://bytedance-research-uno-flux.hf.space/',
+                        'Origin': 'https://bytedance-research-uno-flux.hf.space',
+                        'User-Agent': random_user_agent(),
+                        'X-Forwarded-For': random_ip()
+                    },
+                    timeout=30
+                )
+                if poll_resp.status_code != 200:
+                    yield f"data: {{\"type\":\"error\",\"message\":\"Polling failed\",\"status\":{poll_resp.status_code}}}\n\n"
+                    return
+                lines = poll_resp.text.split('\n')
+                event_data = ''
+                for line in lines:
+                    if line.startswith('data: '):
+                        event_data += line[6:]
+                    elif line == '':
+                        if event_data:
+                            try:
+                                json_event = eval(event_data, {}, {}) if event_data.strip().startswith('{') else None
+                            except Exception:
+                                json_event = None
+                            if json_event:
+                                if json_event.get('msg') == 'estimation':
+                                    yield f"data: {{\"type\":\"estimation\",\"queueSize\":{json_event.get('queue_size')},\"eta\":{json_event.get('rank_eta')}}}\n\n"
+                                elif json_event.get('msg') == 'process_starts':
+                                    yield f"data: {{\"type\":\"processing\"}}\n\n"
+                                elif json_event.get('msg') == 'process_generating':
+                                    yield f"data: {{\"type\":\"generating\",\"output\":{json_event.get('output')}}}\n\n"
+                                elif json_event.get('msg') == 'process_failed':
+                                    yield f"data: {{\"type\":\"error\",\"message\":\"Generation failed on server\"}}\n\n"
+                                    return
+                                elif json_event.get('msg') == 'process_completed' and json_event.get('output', {}).get('data', [{}])[0].get('url'):
+                                    image_url = json_event['output']['data'][0]['url']
+                                    proxies = f"/api/proxy-image?url={image_url}"
+                                    yield f"data: {{\"type\":\"success\",\"imageUrl\":\"{image_url}\",\"proxies\":\"{proxies}\"}}\n\n"
+                                    return
+                                elif json_event.get('msg') == 'process_completed' and json_event.get('output', {}).get('error'):
+                                    yield f"data: {{\"type\":\"error\",\"message\":\"{json_event['output']['error']}\"}}\n\n"
+                                    return
+                            event_data = ''
+                    elif line.startswith(':'):
+                        continue
+                    elif line != '':
+                        event_data += line
+            except Exception as e:
+                yield f"data: {{\"type\":\"error\",\"message\":\"Polling error\",\"details\":\"{str(e)}\"}}\n\n"
+                return
+        yield f"data: {{\"type\":\"error\",\"message\":\"Generation timed out\"}}\n\n"
+    return sse_stream(event_gen())
+
+# --- Heartsync Endpoint ---
+@app.route('/api/generate', methods=['POST'])
+def generate_heartsync():
+    data = request.json or {}
+    prompt = data.get('prompt', '')
+    if not prompt:
+        return sse_stream(lambda: iter([f"data: {{\"type\":\"error\",\"message\":\"Prompt is required\"}}\n\n"]))
+    session_hash = random_session_hash()
+    join_payload = {
+        "data": [
+            prompt,
+            "text, talk bubble, low quality, watermark, signature",
+            0,
+            True,
+            1024,
+            1024,
+            7,
+            28
+        ],
+        "event_data": None,
+        "fn_index": 2,
+        "trigger_id": 14,
+        "session_hash": session_hash
+    }
+    def event_gen():
+        try:
+            join_resp = requests.post(
+                'https://heartsync-nsfw-uncensored.hf.space/gradio_api/queue/join',
+                json=join_payload,
+                headers={
+                    'Content-Type': 'application/json',
+                    'Referer': 'https://heartsync-nsfw-uncensored.hf.space/',
+                    'User-Agent': random_user_agent(),
+                    'X-Forwarded-For': random_ip()
+                },
+                timeout=30
+            )
+            if join_resp.status_code != 200:
+                yield f"data: {{\"type\":\"error\",\"message\":\"Failed to join Heartsync queue\",\"status\":{join_resp.status_code}}}\n\n"
+                return
+        except Exception as e:
+            yield f"data: {{\"type\":\"error\",\"message\":\"Failed to join Heartsync queue\",\"details\":\"{str(e)}\"}}\n\n"
+            return
+        max_attempts = 60
+        for attempt in range(max_attempts):
+            try:
+                poll_resp = requests.get(
+                    f'https://heartsync-nsfw-uncensored.hf.space/gradio_api/queue/data?session_hash={session_hash}',
+                    headers={
+                        'Accept': 'text/event-stream',
+                        'Referer': 'https://heartsync-nsfw-uncensored.hf.space/',
+                        'User-Agent': random_user_agent(),
+                        'X-Forwarded-For': random_ip()
+                    },
+                    timeout=30
+                )
+                if poll_resp.status_code != 200:
+                    yield f"data: {{\"type\":\"error\",\"message\":\"Polling failed\",\"status\":{poll_resp.status_code}}}\n\n"
+                    return
+                lines = poll_resp.text.split('\n')
+                event_data = ''
+                for line in lines:
+                    if line.startswith('data: '):
+                        event_data += line[6:]
+                    elif line == '':
+                        if event_data:
+                            try:
+                                json_event = eval(event_data, {}, {}) if event_data.strip().startswith('{') else None
+                            except Exception:
+                                json_event = None
+                            if json_event:
+                                if json_event.get('msg') == 'estimation':
+                                    yield f"data: {{\"type\":\"estimation\",\"queueSize\":{json_event.get('queue_size')},\"eta\":{json_event.get('rank_eta')}}}\n\n"
+                                elif json_event.get('msg') == 'process_starts':
+                                    yield f"data: {{\"type\":\"processing\"}}\n\n"
+                                elif json_event.get('msg') == 'process_generating':
+                                    yield f"data: {{\"type\":\"generating\",\"output\":{json_event.get('output')}}}\n\n"
+                                elif json_event.get('msg') == 'process_failed':
+                                    yield f"data: {{\"type\":\"error\",\"message\":\"Generation failed on server\"}}\n\n"
+                                    return
+                                elif json_event.get('msg') == 'process_completed' and json_event.get('output', {}).get('data', [{}])[0].get('url'):
+                                    image_url = json_event['output']['data'][0]['url']
+                                    proxies = f"/api/proxy-image?url={image_url}"
+                                    yield f"data: {{\"type\":\"success\",\"originalUrl\":\"{image_url}\"}}\n\n"
+                                    return
+                                elif json_event.get('msg') == 'process_completed' and json_event.get('output', {}).get('error'):
+                                    yield f"data: {{\"type\":\"error\",\"message\":\"{json_event['output']['error']}\"}}\n\n"
+                                    return
+                            event_data = ''
+                    elif line.startswith(':'):
+                        continue
+                    elif line != '':
+                        event_data += line
+            except Exception as e:
+                yield f"data: {{\"type\":\"error\",\"message\":\"Polling error\",\"details\":\"{str(e)}\"}}\n\n"
+                return
+        yield f"data: {{\"type\":\"error\",\"message\":\"Generation timed out\"}}\n\n"
+    return sse_stream(event_gen())
+
+# --- Proxy Image Endpoint ---
+@app.route('/api/proxy-image')
+def proxy_image():
+    url = request.args.get('url')
+    if not url:
+        return 'Missing url parameter', 400
+    try:
+        resp = requests.get(url, stream=True, timeout=30)
+        if resp.status_code != 200:
+            return 'Failed to fetch image', resp.status_code
+        return Response(resp.raw, content_type=resp.headers.get('content-type', 'image/png'))
+    except Exception as e:
+        return f'Proxy error: {str(e)}', 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True) 

server.js

@@ -1,429 +0,0 @@
-import express from 'express';
-import axios from 'axios';
-import { Client } from '@gradio/client';
-import dotenv from 'dotenv';
-import https from 'https';
-import http from 'http';
-
-dotenv.config();
-
-// --- Hugging Face Space Auto-Restarter ---
-const HF_TOKEN = process.env.HF_TOKEN;
-const SPACE_ID = process.env.HF_SPACE_ID || 'shashwatIDR/vision';
-const HF_API_BASE = 'https://huggingface.co/api';
-
-async function restartSpace() {
-  console.log(`[🔄 Attempting restart] ${SPACE_ID}`);
-  try {
-    const response = await axios.post(`${HF_API_BASE}/spaces/${SPACE_ID}/restart`, {}, {
-      headers: {
-        Authorization: `Bearer ${HF_TOKEN}`,
-        'Content-Type': 'application/json'
-      },
-      timeout: 30000
-    });
-    console.log(`[✅ Restarted] ${SPACE_ID} at ${new Date().toLocaleString()}`);
-    if (response.data) {
-      console.log('[📊 Response]', response.data);
-    }
-  } catch (err) {
-    if (err.code === 'ENOTFOUND') {
-      console.error('[❌ DNS Error] Cannot resolve huggingface.co - check your internet connection');
-    } else if (err.code === 'ETIMEDOUT') {
-      console.error('[❌ Timeout] Request timed out - try again later');
-    } else if (err.response) {
-      console.error(`[❌ HTTP ${err.response.status}]`, err.response.data || err.response.statusText);
-      if (err.response.status === 401) {
-        console.error('[🔑 Auth Error] Check your Hugging Face token');
-      } else if (err.response.status === 404) {
-        console.error('[🔍 Not Found] Space may not exist or token lacks permissions');
-      }
-    } else {
-      console.error('[❌ Failed to Restart]', err.message);
-    }
-  }
-}
-
-async function testHFAPI() {
-  try {
-    console.log('[🔍 Testing Hugging Face API endpoint...]');
-    const response = await axios.get(`${HF_API_BASE}/spaces`, {
-      headers: { Authorization: `Bearer ${HF_TOKEN}` },
-      timeout: 10000,
-      params: { limit: 1 }
-    });
-    console.log('[✅ API Test] Endpoint is working');
-    return true;
-  } catch (err) {
-    console.error('[❌ API Test Failed]', err.response?.status || err.code || err.message);
-    return false;
-  }
-}
-
-// On server startup, test API and schedule restarts
-(async () => {
-  if (!HF_TOKEN) {
-    console.error('[❌ No HF_TOKEN] Hugging Face token not set. Skipping auto-restart scheduling.');
-    return;
-  }
-  console.log('[🚀 Starting] HuggingFace Space Auto-Restarter');
-  const apiWorking = await testHFAPI();
-  if (!apiWorking) {
-    console.log('[⚠️ Warning] API test failed, but proceeding anyway...');
-  }
-  // Only schedule periodic restarts (no immediate restart on startup)
-  setInterval(restartSpace, 5 * 60 * 1000);
-  console.log('[⏰ Scheduled] Space will restart every 5 minutes');
-})();
-
-const app = express();
-const port = process.env.PORT || 3000;
-app.use(express.json());
-
-// Serve static files from public
-app.use(express.static('public'));
-
-// Serve index.html at root
-app.get('/', (req, res) => {
-  res.sendFile(__dirname + '/public/index.html');
-});
-
-// Image Generation endpoint (Heartsync)
-app.post('/api/generate', async (req, res) => {
-    res.setHeader('Content-Type', 'text/event-stream');
-    res.setHeader('Cache-Control', 'no-cache');
-    res.setHeader('Connection', 'keep-alive');
-
-    try {
-        const { prompt } = req.body;
-        const sessionHash = Math.random().toString(36).substring(2, 15);
-
-        // First, join the queue
-        const joinResponse = await axios.post('https://api.allorigins.win/raw?url=https://heartsync-nsfw-uncensored.hf.space/gradio_api/queue/join', {
-            data: [
-                prompt,
-                "text, talk bubble, low quality, watermark, signature",
-                0,
-                true,
-                1024,
-                1024,
-                7,
-                28
-            ],
-            event_data: null,
-            fn_index: 2,
-            trigger_id: 14,
-            session_hash: sessionHash
-        }, {
-            headers: {
-                'Content-Type': 'application/json',
-                'Referer': 'https://heartsync-nsfw-uncensored.hf.space/'
-            }
-        });
-
-        // Poll for results
-        let attempts = 0;
-        const maxAttempts = 60; // Increased attempts for potentially longer queues
-        
-        while (attempts < maxAttempts) {
-            const dataResponse = await axios.get(
-                `https://api.allorigins.win/raw?url=https://heartsync-nsfw-uncensored.hf.space/gradio_api/queue/data?session_hash=${encodeURIComponent(sessionHash)}`,
-                {
-                    headers: {
-                        'Accept': 'text/event-stream',
-                        'Referer': 'https://heartsync-nsfw-uncensored.hf.space/'
-                    }
-                }
-            );
-
-            const data = dataResponse.data;
-            
-            // Manually parse the event stream data
-            const lines = data.split('\n');
-            let eventData = '';
-            let foundEvent = false;
-            for (const line of lines) {
-                if (line.startsWith('data: ')) {
-                    eventData += line.substring(6);
-                } else if (line === '') {
-                    // Process eventData when an empty line is encountered
-                    if (eventData) {
-                        try {
-                            const json = JSON.parse(eventData);
-                            console.log(`[Image] Parsed event: ${json.msg}`);
-                            
-                            if (json.msg === 'process_completed' && json.output?.data?.[0]?.url) {
-                                const imageUrl = json.output.data[0].url;
-                                
-                                // Send the original image URL back to the client as an SSE
-                                res.write(`data: ${JSON.stringify({
-                                    type: 'success',
-                                    originalUrl: imageUrl
-                                })}\n\n`);
-                                res.end(); // End the connection after sending the final event
-                                return; // Exit the function
-                            }
-
-                            if (json.msg === 'estimation') {
-                                res.write(`data: ${JSON.stringify({ type: 'estimation', queueSize: json.queue_size, eta: json.rank_eta })}\n\n`);
-                                foundEvent = true;
-                            } else if (json.msg === 'process_starts') {
-                                res.write(`data: ${JSON.stringify({ type: 'processing' })}\n\n`);
-                                foundEvent = true;
-                            } else if (json.msg === 'process_failed') {
-                                console.log(`[Image] Process failed:`, json);
-                                res.write(`data: ${JSON.stringify({ type: 'error', message: 'Generation failed on server' })}\n\n`);
-                                res.end();
-                                return;
-                            }
-
-                        } catch (error) {
-                            console.error('[Image] Error parsing event data:', error, 'Raw data:', eventData);
-                        }
-                        eventData = ''; // Reset for the next event
-                    }
-                } else if (line.startsWith(':')) {
-                    // Ignore comment lines
-                    continue;
-                } else if (line !== '') {
-                     // If line is not data or comment, it might be part of a multi-line data chunk, append to eventData
-                     eventData += line;
-                }
-            }
-
-            // If loop finishes without 'process_completed', wait and try again
-            await new Promise(resolve => setTimeout(resolve, 2000)); // Wait longer before next poll attempt
-            attempts++;
-        }
-
-        // If the loop times out without success
-        if (!foundEvent) {
-            console.log(`[Image] No events found in response, raw data:`, data.substring(0, 500));
-        }
-        res.status(500).write(`data: ${JSON.stringify({ type: 'error', message: 'Generation timed out' })}\n\n`);
-        res.end();
-
-    } catch (error) {
-        console.error('Generation endpoint error:', error);
-        res.status(500).write(`data: ${JSON.stringify({ type: 'error', message: error.message || 'Failed to generate image' })}\n\n`);
-        res.end();
-    }
-});
-
-// Proxy image endpoint
-app.get('/api/proxy-image', async (req, res) => {
-    const { url } = req.query;
-    if (!url) {
-        return res.status(400).send('Missing url parameter');
-    }
-    try {
-        // Support both http and https
-        const client = url.startsWith('https') ? https : http;
-        client.get(url, (imageRes) => {
-            if (imageRes.statusCode !== 200) {
-                res.status(imageRes.statusCode).send('Failed to fetch image');
-                return;
-            }
-            res.setHeader('Content-Type', imageRes.headers['content-type'] || 'image/png');
-            imageRes.pipe(res);
-        }).on('error', (err) => {
-            res.status(500).send('Proxy error: ' + err.message);
-        });
-    } catch (err) {
-        res.status(500).send('Proxy error: ' + err.message);
-    }
-});
-
-// Flux model generation endpoint using UNO-FLUX (bytedance-research-uno-flux.hf.space)
-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');
-    // No data is stored on the server for this endpoint
-    function randomUserAgent() {
-        // Simple pool of common user agents
-        const agents = [
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
-            'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Safari/605.1.15',
-            'Mozilla/5.0 (Linux; Android 13; SM-G991B) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Mobile Safari/537.36',
-            'Mozilla/5.0 (iPhone; CPU iPhone OS 17_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Mobile/15E148 Safari/604.1',
-            'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36'
-        ];
-        return agents[Math.floor(Math.random() * agents.length)];
-    }
-    function randomIP() {
-        return Array(4).fill(0).map(() => Math.floor(Math.random() * 256)).join('.');
-    }
-    try {
-        const {
-            prompt = '',
-            width = 1280,
-            height = 720,
-            guidance = 4,
-            num_steps = 25,
-            seed = -1,
-            image_prompt1 = null,
-            image_prompt2 = null,
-            image_prompt3 = null,
-            image_prompt4 = null,
-            aspect = '16:9'
-        } = req.body;
-
-        if (!prompt) {
-            res.write(`data: ${JSON.stringify({ type: 'error', message: 'Prompt is required' })}\n\n`);
-            res.end();
-            return;
-        }
-
-        // Aspect ratio logic
-        let w = width, h = height;
-        if (aspect === '9:16') {
-            w = 720; h = 1280;
-        } else if (aspect === '1:1') {
-            w = 1024; h = 1024;
-        }
-
-        const sessionHash = Math.random().toString(36).substring(2, 15);
-        const joinPayload = {
-            data: [
-                prompt,
-                w,
-                h,
-                guidance,
-                num_steps,
-                seed,
-                image_prompt1,
-                image_prompt2,
-                image_prompt3,
-                image_prompt4
-            ],
-            event_data: null,
-            fn_index: 0,
-            trigger_id: 25,
-            session_hash: sessionHash
-        };
-
-        // Step 1: Join the queue
-        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/',
-                        'Origin': 'https://bytedance-research-uno-flux.hf.space',
-                        'User-Agent': randomUserAgent(),
-                        'X-Forwarded-For': randomIP()
-                    },
-                    validateStatus: () => true // Accept all status codes
-                }
-            );
-        } catch (err) {
-            res.write(`data: ${JSON.stringify({ type: 'error', message: 'Failed to join UNO-FLUX queue', details: err.message })}\n\n`);
-            res.end();
-            return;
-        }
-        if (!joinResponse || joinResponse.status !== 200) {
-            res.write(`data: ${JSON.stringify({ type: 'error', message: 'Failed to join UNO-FLUX queue', status: joinResponse?.status, data: joinResponse?.data })}\n\n`);
-            res.end();
-            return;
-        }
-
-        // Step 2: Poll for results (Axios does not support true SSE streaming, so we poll every 2s)
-        let attempts = 0;
-        const maxAttempts = 90; // Increased for slow generations
-        let foundEvent = false;
-        while (attempts < maxAttempts) {
-            let dataResponse;
-            try {
-                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/',
-                            'Origin': 'https://bytedance-research-uno-flux.hf.space',
-                            'User-Agent': randomUserAgent(),
-                            'X-Forwarded-For': randomIP()
-                        },
-                        validateStatus: () => true // Accept all status codes
-                    }
-                );
-            } catch (pollErr) {
-                res.write(`data: ${JSON.stringify({ type: 'error', message: 'Polling error', details: pollErr.message })}\n\n`);
-                res.end();
-                return;
-            }
-            if (!dataResponse || dataResponse.status !== 200) {
-                res.write(`data: ${JSON.stringify({ type: 'error', message: 'Polling failed', status: dataResponse?.status, data: dataResponse?.data })}\n\n`);
-                res.end();
-                return;
-            }
-            const data = dataResponse.data;
-            // Manually 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('[UNO-FLUX Event]', json); // Log every event
-                            if (json.msg === 'estimation') {
-                                res.write(`data: ${JSON.stringify({ type: 'estimation', queueSize: json.queue_size, eta: json.rank_eta })}\n\n`);
-                                foundEvent = true;
-                            } else if (json.msg === 'process_starts') {
-                                res.write(`data: ${JSON.stringify({ type: 'processing' })}\n\n`);
-                                foundEvent = true;
-                            } else if (json.msg === 'process_generating') {
-                                // Forward process_generating events to client for progress feedback
-                                res.write(`data: ${JSON.stringify({ type: 'generating', output: json.output })}\n\n`);
-                                foundEvent = true;
-                            } else if (json.msg === 'process_failed') {
-                                res.write(`data: ${JSON.stringify({ type: 'error', message: 'Generation failed on server' })}\n\n`);
-                                res.end();
-                                return;
-                            } else if (json.msg === 'process_completed' && json.output?.data?.[0]?.url) {
-                                const imageUrl = json.output.data[0].url;
-                                // Build proxy URL
-                                const base = req.protocol + '://' + req.get('host');
-                                const proxies = base + '/api/proxy-image?url=' + encodeURIComponent(imageUrl);
-                                res.write(`data: ${JSON.stringify({ type: 'success', imageUrl, proxies })}\n\n`);
-                                res.end();
-                                return;
-                            } else if (json.msg === 'process_completed' && json.output?.error) {
-                                res.write(`data: ${JSON.stringify({ type: 'error', message: json.output.error })}\n\n`);
-                                res.end();
-                                return;
-                            }
-                        } catch (error) {
-                            console.error('[UNO-FLUX] Error parsing event data:', error, 'Raw data:', eventData);
-                        }
-                        eventData = '';
-                    }
-                } else if (line.startsWith(':')) {
-                    continue;
-                } else if (line !== '') {
-                    eventData += line;
-                }
-            }
-            await new Promise(resolve => setTimeout(resolve, 2000));
-            attempts++;
-        }
-        if (!foundEvent) {
-            res.write(`data: ${JSON.stringify({ type: 'error', message: 'Generation timed out' })}\n\n`);
-        }
-        res.end();
-    } catch (err) {
-        res.write(`data: ${JSON.stringify({ type: 'error', message: err.message })}\n\n`);
-        res.end();
-    }
-});
-
-app.listen(port, () => {
-  console.log(`Server running on port ${port}`);
-});