Update server.js

server.js

@@ -6,10 +6,12 @@ import { Readable } from 'stream';
 const app = express();
 const PORT = 7860;
 
+// Configuración CRÍTICA para que express-rate-limit lea la IP real del usuario
+// cuando la app está detrás de proxies inversos (ej. Vercel, Render, HF Spaces, Nginx)
 app.set('trust proxy', 1);
 app.use(cors());
 
-// CRÍTICO: Aumentar límite a 50mb para soportar imágenes en Base64 (Visión / Multimodal)
+// Aumentar límite a 50mb para soportar imágenes en Base64 (Visión / Multimodal)
 app.use(express.json({ limit: '50mb' }));
 app.use(express.urlencoded({ limit: '50mb', extended: true }));
 
@@ -29,12 +31,12 @@ const PROVIDERS = [
         id: "pollinations", 
         url: "https://text.pollinations.ai/openai", 
         modelsUrl: "https://text.pollinations.ai/models",
-        imageUrl: "https://image.pollinations.ai/prompt" // Endpoint nativo de imágenes
+        imageUrl: "https://image.pollinations.ai/prompt"
     },
     { 
         id: "airforce", 
         url: "https://api.airforce/v1/chat/completions",
-        imageUrl: "https://api.airforce/v1/images/generations" // Soporte de imágenes
+        imageUrl: "https://api.airforce/v1/images/generations"
     },
     { 
         id: "ventarys-mirror", 
@@ -48,9 +50,11 @@ const QUEUE_TIMEOUT = 25000;
 
 let currentLoad = { "llm7": 0, "pollinations": 0, "airforce": 0, "ventarys-mirror": 0 };
 
+// --- RATE LIMITING (Basado en la IP real del usuario) ---
 const limiter = rateLimit({
     windowMs: 60 * 1000, 
     max: 25, 
+    keyGenerator: (req) => req.ip, // Extrae la IP real verificada por 'trust proxy'
     message: { error: { message: "Límite alcanzado. Espera 1 minuto entre mensajes.", code: 429 } },
     standardHeaders: true,
     legacyHeaders: false,
@@ -60,17 +64,19 @@ const limiter = rateLimit({
 const IMAGE_KEYWORDS = ["flux", "dall", "midjourney", "sdxl", "stable-diffusion", "image", "vision"];
 const AUDIO_KEYWORDS = ["suno", "udio", "music", "audio", "song", "voice", "tts"];
 
-function isImageModel(id) {
-    if (!id) return false;
-    const lowerId = id.toLowerCase();
-    return IMAGE_KEYWORDS.some(kw => lowerId.includes(kw));
+function isImageModel(model) {
+    if (!model) return false;
+    // Soporte directo para el flag de api.airforce
+    if (model.type === 'image' || model.supports_images === true) return true;
+    
+    const id = (model.id || model.name || "").toLowerCase();
+    return IMAGE_KEYWORDS.some(kw => id.includes(kw));
 }
 
-function isAudioModel(id, type) {
-    if (type === 'audio' || type === 'music') return true;
-    if (!id) return false;
-    const lowerId = id.toLowerCase();
-    return AUDIO_KEYWORDS.some(kw => lowerId.includes(kw));
+function isAudioModel(model) {
+    if (model.type === 'audio' || model.type === 'music') return true;
+    const id = (model.id || model.name || "").toLowerCase();
+    return AUDIO_KEYWORDS.some(kw => id.includes(kw));
 }
 
 async function fetchAllModels() {
@@ -93,8 +99,9 @@ async function fetchAllModels() {
 
             if (modelsArray.length > 0) {
                 return modelsArray
-                    // Eliminamos los modelos de música de raíz para que no ensucien ninguna lista
-                    .filter(model => !isAudioModel(model.id || model.name, model.type))
+                    // Filtrar modelos de audio/música
+                    .filter(model => !isAudioModel(model))
+                    // Procesar y formatear campos
                     .map(model => ({
                         ...model,
                         id: model.id || model.name,
@@ -129,11 +136,14 @@ app.get('/health', (req, res) => {
 app.get('/v1/models', async (req, res) => {
     try {
         const allModels = await fetchAllModels();
-        // Filtrar modelos asegurando que NO sean de imágenes
-        const textModels = allModels.filter(m => !isImageModel(m.id) && m.type !== 'image').map(m => {
-            m.type = 'text'; // Forzar etiqueta
-            return m;
-        });
+        
+        const textModels = allModels
+            // Excluir imágenes y asegurar que soporte chat (filtro estricto para api.airforce)
+            .filter(m => !isImageModel(m) && m.supports_chat !== false)
+            .map(m => {
+                m.type = 'text'; // Forzar etiqueta
+                return m;
+            });
 
         res.json({ object: "list", data: textModels });
     } catch (error) {
@@ -146,13 +156,16 @@ app.get('/v1/models', async (req, res) => {
 app.get('/v1/images/models', async (req, res) => {
     try {
         const allModels = await fetchAllModels();
-        // Filtrar modelos asegurando que SÍ sean de imágenes
-        let imageModels = allModels.filter(m => isImageModel(m.id) || m.type === 'image').map(m => {
-            m.type = 'image'; // Forzar etiqueta
-            return m;
-        });
+        
+        let imageModels = allModels
+            // Incluir explícitamente modelos de imagen (o que supports_images sea true en airforce)
+            .filter(m => isImageModel(m))
+            .map(m => {
+                m.type = 'image'; // Forzar etiqueta
+                return m;
+            });
 
-        // Garantizar que la App siempre reconozca los modelos base aunque la API origen no los liste
+        // Garantizar que la App siempre reconozca los modelos base
         const baseImages = [
             { id: "flux", object: "model", type: "image", owned_by: "system", tier: "standard" },
             { id: "dall-e-3", object: "model", type: "image", owned_by: "system", tier: "standard" }
@@ -220,14 +233,12 @@ app.post(['/v1/chat/completions', '/v1/images/generations'], limiter, async (req
             const prompt = req.body.prompt || "A random image";
             const seed = Math.floor(Math.random() * 10000000);
             
-            // Extraer resolución del body (por defecto cuadrado si no se provee)
             let width = 1024, height = 1024;
             if (req.body.size) {
                 const parts = req.body.size.split('x');
                 if (parts.length === 2) { width = parseInt(parts[0]); height = parseInt(parts[1]); }
             }
             
-            // Ajustar el modelo si el usuario pidió uno en específico
             const pollModel = req.body.model && req.body.model.includes('flux') ? 'flux' : 'dall-e-3';
 
             targetUrl = `${targetUrl}/${encodeURIComponent(prompt)}?seed=${seed}&width=${width}&height=${height}&model=${pollModel}&nologo=true`;
@@ -253,9 +264,15 @@ app.post(['/v1/chat/completions', '/v1/images/generations'], limiter, async (req
             if (contentType.includes("application/json")) {
                 const jsonResp = await response.json();
                 
-                // Si la API remota nos responde con JSON, extraemos las URL para transformarlas a Base64
-                if (jsonResp.data && Array.isArray(jsonResp.data)) {
-                    for (let item of jsonResp.data) {
+                // Extraer el array de datos o crearlo si la API (ej. airforce) responde con { url: "..." } directo
+                let dataArray = jsonResp.data;
+                if (!dataArray && jsonResp.url) {
+                    dataArray = [{ url: jsonResp.url }];
+                }
+                
+                // Convertir URLs de imágenes a base64_json (Formato esperado idéntico a Aqua AI)
+                if (dataArray && Array.isArray(dataArray)) {
+                    for (let item of dataArray) {
                         if (item.url && !item.b64_json) {
                             try {
                                 const imgRes = await fetch(item.url);
@@ -267,7 +284,14 @@ app.post(['/v1/chat/completions', '/v1/images/generations'], limiter, async (req
                             }
                         }
                     }
+                    releaseSlot();
+                    return res.status(response.status).json({
+                        created: Math.floor(Date.now() / 1000),
+                        data: dataArray
+                    });
                 }
+                
+                // Respaldo de seguridad si no encontró formato extraíble
                 releaseSlot();
                 return res.status(response.status).json(jsonResp);
             } 
@@ -277,14 +301,13 @@ app.post(['/v1/chat/completions', '/v1/images/generations'], limiter, async (req
                 const b64 = Buffer.from(arrayBuffer).toString('base64');
                 releaseSlot();
                 
-                // Estandarizar respuesta para tu frontend (Aqua AI / OpenAI compatible)
+                // Formato Aqua AI Strict:
                 return res.status(200).json({
                     created: Math.floor(Date.now() / 1000),
                     data: [{ b64_json: b64 }]
                 });
             } 
             else {
-                // Caída libre para errores o comportamientos inesperados de los proveedores
                 const textResp = await response.text();
                 releaseSlot();
                 return res.status(response.status).type(contentType).send(textResp);