Identity: Full Theme Sync - CODEX_COLLECTIVE v5.0 with image generation and renamed engines

backend/controllers/ai.js

@@ -16,8 +16,9 @@ const MODELS = {
   'Codex Ratio': 'llama-3.3-70b',
   'Codex Nexus': 'deepseek-v3.2',
   'Codex Fero': 'Qwen/Qwen3-Coder-480B-A35B-Instruct',
-  'Codex Zenith': 'Qwen2.5-72B-Instruct',
-  'Codex Magna': 'moonshotai/Kimi-K2.5'
+  'Codex Zenith': 'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B',
+  'Codex Magna': 'meta-llama/Llama-3.3-70B-Instruct',
+  'Codex Vision': 'zai-org/GLM-Image'
 };
 
 const PROVIDERS = {
@@ -34,7 +35,8 @@ const SPECIALIZATIONS = {
   'Codex Nexus': 'ROLE: GENERALIST_ELITE. Deep understanding and versatility.',
   'Codex Fero': 'CORE_PROTOCOL: FERO_ENGINE. Heavyweight architecture.',
   'Codex Zenith': 'CORE_PROTOCOL: ZENITH_REASONING. Deep thinking phase.',
-  'Codex Magna': 'CORE_PROTOCOL: MAGNA_FRONTIER. Ultimate intelligence and nuance.'
+  'Codex Magna': 'CORE_PROTOCOL: MAGNA_FRONTIER. Ultimate intelligence and nuance.',
+  'Codex Vision': 'CORE_PROTOCOL: VISUAL_PROJECTION. Direct image synthesis.'
 };
 
 const SYSTEM_PROMPT = `CORE_IDENTITY:
@@ -48,13 +50,21 @@ OPERATIONAL_RULES:
 5. MODEL_ACKNOWLEDGEMENT: List available models as: {AVAILABLE_MODELS}.`;
 
 exports.chat = asyncHandler(async (req, res, next) => {
-  const { message, sessionId, model } = req.body;
+  let { message, sessionId, model } = req.body;
   const user = req.user;
 
   if (user.usage.requestsToday >= 150 && user.role !== 'owner') {
     return next(new ErrorResponse('DAILY_PROTOCOL_LIMIT_EXCEEDED', 429));
   }
 
+  // --- AUTOMATIC VISION ROUTING ---
+  const visionTriggers = ["make me an image", "generate an image", "create an image", "visualize"];
+  const lowerMsg = message.toLowerCase();
+  if (visionTriggers.some(trigger => lowerMsg.startsWith(trigger))) {
+    model = 'Codex Vision';
+    console.log(`[Auto-Switch] Detected visual request. Engaging Codex Vision.`);
+  }
+
   const activeModelName = model || 'Codex Velox';
   
   // ZENITH LIMITATION LOGIC
@@ -87,7 +97,7 @@ exports.chat = asyncHandler(async (req, res, next) => {
   // 3. Build History
   const history = await Message.find({ sessionId: session._id }).sort({ createdAt: 1 }).limit(10);
   const availableModelsList = Object.keys(MODELS).join(', ');
-  const specialization = SPECIALIZATIONS[activeModelName];
+  const specialization = SPECIALIZATIONS[activeModelName] || SPECIALIZATIONS['Codex Velox'];
   
   const apiMessages = [
     { role: 'system', content: SYSTEM_PROMPT.replace('{ACTIVE_MODEL}', activeModelName).replace('{AVAILABLE_MODELS}', availableModelsList) },
@@ -105,6 +115,35 @@ exports.chat = asyncHandler(async (req, res, next) => {
   res.setHeader('Connection', 'keep-alive');
 
   try {
+    // --- VISION CORE EXECUTION ---
+    if (activeModelName === 'Codex Vision') {
+        const prompt = message.replace(/make me an image|generate an image|create an image|visualize/gi, "").trim();
+        const hfToken = process.env.HF_TOKEN;
+        const response = await axios.post(
+            `https://api-inference.huggingface.co/models/${MODELS['Codex Vision']}`,
+            { inputs: prompt },
+            { headers: { Authorization: `Bearer ${hfToken}` }, responseType: 'arraybuffer' }
+        );
+
+        const filename = `vision-${Date.now()}.png`;
+        const filepath = `./public/uploads/${filename}`;
+        if (!fs.existsSync('./public/uploads')) fs.mkdirSync('./public/uploads', { recursive: true });
+        fs.writeFileSync(filepath, response.data);
+        const imageUrl = `/uploads/${filename}`;
+        const aiResponse = `[VISUAL_PROJECTION_COMPLETE]\n\n![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();
+
+        res.write(`data: ${JSON.stringify({ message: aiResponse })}\n\n`);
+        res.write(`data: ${JSON.stringify({ done: true, sessionId: session._id })}\n\n`);
+        res.end();
+        return;
+    }
+
+    // --- GRADIO CORES (Fero, Zenith, Magna) ---
     if (['Codex Fero', 'Codex Zenith', 'Codex Magna'].includes(activeModelName)) {
       const spaceName = process.env.HF_SPACE_ID || "zhlajiex/aimodel";
       const hfToken = process.env.HF_TOKEN;
@@ -127,23 +166,18 @@ exports.chat = asyncHandler(async (req, res, next) => {
             if (newContent) res.write(`data: ${JSON.stringify({ message: newContent })}\n\n`);
           }
         }
-      } catch (loopErr) {
-        console.error("Gradio_Loop_Error:", loopErr.message);
-        res.write(`data: ${JSON.stringify({ message: "\n\n[NEURAL_LINK_ERROR]: Request interrupted. Processing partial response." })}\n\n`);
-      }
+      } catch (e) { console.error("Gradio Error", e); }
       
       await Message.create({ sessionId: session._id, sender: 'user', content: message || "[SIGNAL]", attachmentUrl });
       await Message.create({ sessionId: session._id, sender: 'ai', content: fullAIResponse || "[EMPTY_SIGNAL]", modelUsed: activeModelName });
       user.usage.requestsToday += 1;
       await user.save();
-      
-      // CRITICAL: Always send done signal
       res.write(`data: ${JSON.stringify({ done: true, sessionId: session._id })}\n\n`);
       res.end();
       return;
     }
 
-    // Standard API Call logic
+    // Standard API Call
     let apiUrl = 'https://api.cerebras.ai/v1/chat/completions', apiKey = 'csk-mvww3vy29hykeektyv65w9rkjx94hw4r6mrcj5tjcw9942d2';
     if (activeModelName === 'Codex Nexus') {
       apiUrl = process.env.FLOWEY_URL || 'https://api.flowey.dev/v1/chat/completions';
@@ -179,15 +213,12 @@ exports.chat = asyncHandler(async (req, res, next) => {
       await Message.create({ sessionId: session._id, sender: 'ai', content: fullAIResponse, modelUsed: activeModelName });
       user.usage.requestsToday += 1;
       await user.save();
-      // Ensure done signal sent for standard providers too if not caught by [DONE]
       res.write(`data: ${JSON.stringify({ done: true, sessionId: session._id })}\n\n`);
       res.end();
     });
 
   } catch (err) {
-    console.error("Master_Chat_Error:", err.message);
     res.write(`data: ${JSON.stringify({ error: "NEURAL_LINK_SEVERED", details: err.message })}\n\n`);
-    res.write(`data: ${JSON.stringify({ done: true })}\n\n`);
     res.end();
   }
 });

backend/public/chat.html

@@ -295,7 +295,7 @@
                     document.getElementById('budget-val').innerText = p.reasoningBudget || 1024;
                     document.getElementById('temp-slider').value = p.temperature || 0.7;
                     document.getElementById('temp-val').innerText = p.temperature || 0.7;
-                    document.getElementById('usage-count').innerText = `${String(data.data.usage.requestsToday).padStart(2, '0')}/75`;
+                    document.getElementById('usage-count').innerText = `${String(data.data.usage.requestsToday).padStart(2, '0')}/150`;
                     if (data.data.isOwner) document.getElementById('owner-tag').classList.remove('hidden');
                 }
                 loadHistory();
@@ -350,15 +350,20 @@
 
             try {
                 const res = await fetch(`${API_BASE}/api/ai/chat`, { method: 'POST', headers: { 'Authorization': `Bearer ${token}` }, body: fd });
+                
+                if (res.status === 401) {
+                    localStorage.removeItem('token');
+                    window.location.href = '/auth?error=SESSION_EXPIRED';
+                    return;
+                }
+
                 const reader = res.body.getReader(); const decoder = new TextDecoder();
                 let aiNode = appendMessage('ai', '', activeModel); let fullText = "";
                 
-                // Safety Timeout: Reset button if no activity for 60 seconds
                 const safetyTimeout = setTimeout(() => {
                     if (isProcessing) {
                         isProcessing = false;
                         document.getElementById('send-btn').innerHTML = '<i class="fas fa-arrow-up text-sm"></i>';
-                        console.warn("[SYSTEM] Connection timeout. Interface reset.");
                     }
                 }, 60000);