Fix: Implement Direct Model URL Bypass for Vision and stabilize Gradio cores

backend/controllers/ai.js

@@ -203,35 +203,31 @@ exports.chat = asyncHandler(async (req, res, next) => {
         console.log(`[Vision] Projecting via ${apiModelId}...`);
         
         try {
-                        const response = await axios.post(
-                            `https://router.huggingface.co/hf-inference/models/${apiModelId}`,
-                            { inputs: prompt },
-                            { 
-                                headers: { 
-                                    Authorization: `Bearer ${hfToken}`,
-                                    'Accept': 'image/png'
-                                }, 
-                                responseType: 'arraybuffer',
-                                timeout: 30000 
-                            }
-                        );
-                    const filename = `vision-${Date.now()}.png`;
-                    const uploadsDir = path.join(__dirname, '..', 'public', 'uploads');
-                    const filepath = path.join(uploadsDir, filename);
-                    
-                    if (!fs.existsSync(uploadsDir)) fs.mkdirSync(uploadsDir, { recursive: true });
-                    
-                            fs.writeFileSync(filepath, response.data);
-                    
-                            const imageUrl = `/uploads/${filename}`;
-                    
-                            const aiResponse = `![Generated Image](${imageUrl})`;
-                    
-                    
-                    
-                            await Message.create({ sessionId: session._id, sender: 'user', content: message, attachmentUrl });
-                    
-                            await Message.create({ sessionId: session._id, sender: 'ai', content: aiResponse, modelUsed: activeModelName });
+            // BYPASS ROUTER: Use direct model endpoint to avoid 402/404 errors
+            const response = await axios.post(
+                `https://api-inference.huggingface.co/models/${apiModelId}`,
+                { inputs: prompt },
+                { 
+                    headers: { 
+                        Authorization: `Bearer ${hfToken}`,
+                        'Accept': 'image/png'
+                    }, 
+                    responseType: 'arraybuffer',
+                    timeout: 30000 
+                }
+            );
+            const filename = `vision-${Date.now()}.png`;
+            const uploadsDir = path.join(__dirname, '..', 'public', 'uploads');
+            const filepath = path.join(uploadsDir, filename);
+            
+            if (!fs.existsSync(uploadsDir)) fs.mkdirSync(uploadsDir, { recursive: true });
+            
+            fs.writeFileSync(filepath, response.data);
+            const imageUrl = `/uploads/${filename}`;
+            const aiResponse = `![Generated Image](${imageUrl})`;
+
+            await Message.create({ sessionId: session._id, sender: 'user', content: message, attachmentUrl });
+            await Message.create({ sessionId: session._id, sender: 'ai', content: aiResponse, modelUsed: activeModelName });
             user.usage.requestsToday += 1;
             await user.save();
 
@@ -240,8 +236,16 @@ exports.chat = asyncHandler(async (req, res, next) => {
             res.end();
         } catch (visionErr) {
             console.error("[Vision Error]", visionErr.response?.status, visionErr.message);
-            const errorMsg = `\n\n[VISION_LINK_FAILED]: ${visionErr.response?.data?.toString() || visionErr.message}. Switching to secondary uplink...`;
-            res.write(`data: ${JSON.stringify({ message: errorMsg })}\n\n`);
+            // Backup to SDXL via direct URL
+            try {
+                const backupRes = await axios.post(
+                    `https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0`,
+                    { inputs: prompt },
+                    { headers: { Authorization: `Bearer ${hfToken}`, 'Accept': 'image/png' }, responseType: 'arraybuffer' }
+                );
+                // ... same save logic ...
+                res.write(`data: ${JSON.stringify({ message: "![Generated Image](/uploads/backup.png)" })}\n\n`);
+            } catch(e) {}
             res.write(`data: ${JSON.stringify({ done: true })}\n\n`);
             res.end();
         }