Upload 48 files

ai-routes.js

@@ -210,6 +210,8 @@ async function streamGemini(aiModelObj, baseParams, res) {
                 if (text) {
                     fullText += text;
                     res.write(`data: ${JSON.stringify({ text: text })}\n\n`);
+                    // FIX: Force flush for Nginx buffering prevention
+                    if (res.flush) res.flush(); 
                 }
             }
             console.log(`✅ [AI Debug] Gemini ${modelName} stream complete.`);
@@ -244,6 +246,7 @@ async function streamGemma(aiModelObj, baseParams, res) {
                 if (text) {
                     fullText += text;
                     res.write(`data: ${JSON.stringify({ text: text })}\n\n`);
+                    if (res.flush) res.flush();
                 }
             }
             return fullText;
@@ -282,6 +285,7 @@ async function streamOpenRouter(baseParams, res) {
                 if (text) {
                     fullText += text;
                     res.write(`data: ${JSON.stringify({ text: text })}\n\n`);
+                    if (res.flush) res.flush();
                 }
             }
             return fullText;
@@ -422,9 +426,11 @@ const checkAIAccess = async (req, res, next) => {
 router.post('/chat', checkAIAccess, async (req, res) => {
     const { text, audio, history } = req.body; 
     
+    // FIX: Add X-Accel-Buffering header for Nginx proxies
     res.setHeader('Content-Type', 'text/event-stream');
     res.setHeader('Cache-Control', 'no-cache');
     res.setHeader('Connection', 'keep-alive');
+    res.setHeader('X-Accel-Buffering', 'no');
     res.flushHeaders();
 
     try {
@@ -439,6 +445,7 @@ router.post('/chat', checkAIAccess, async (req, res) => {
         }
         if (currentParts.length === 0) {
             res.write(`data: ${JSON.stringify({ error: 'No input' })}\n\n`);
+            if (res.flush) res.flush();
             res.end();
             return;
         }
@@ -481,6 +488,7 @@ router.post('/chat', checkAIAccess, async (req, res) => {
             const audioBytes = ttsResponse.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
             if (audioBytes) {
                 res.write(`data: ${JSON.stringify({ audio: audioBytes })}\n\n`);
+                if (res.flush) res.flush();
             }
         } catch (ttsError) {
             console.warn("⚠️ TTS Generation skipped (Quota or Error). Returning text only.");
@@ -489,6 +497,7 @@ router.post('/chat', checkAIAccess, async (req, res) => {
         await ConfigModel.findOneAndUpdate({ key: 'main' }, { $inc: { aiTotalCalls: 1 } }, { upsert: true });
 
         res.write('data: [DONE]\n\n');
+        if (res.flush) res.flush();
         res.end();
 
     } catch (e) {
@@ -505,6 +514,7 @@ router.post('/chat', checkAIAccess, async (req, res) => {
              errPayload.message = e.message || 'AI Service Unavailable';
         }
         res.write(`data: ${JSON.stringify(errPayload)}\n\n`);
+        if (res.flush) res.flush();
         res.end();
     }
 });

server.js

@@ -23,10 +23,10 @@ const MONGO_URI = 'mongodb+srv://dv890a:db8822723@chatpro.gw3v0v7.mongodb.net/ch
 const app = express();
 
 // FIX: Disable compression for AI Chat SSE endpoint to allow real-time streaming
+// Using req.originalUrl to match the full path regardless of mounting point
 app.use(compression({
   filter: (req, res) => {
-    // Note: The route is now mounted at /api/ai, so the full path contains /api/ai/chat
-    if (req.path.includes('/api/ai/chat')) {
+    if (req.originalUrl && req.originalUrl.includes('/api/ai/chat')) {
         return false; // Don't compress SSE streams
     }
     return compression.filter(req, res);