Stability Fix: Prevent HF Space restart loop by disabling internal git pushes, and improve image recognition data flow

backend/controllers/ai.js

@@ -185,13 +185,18 @@ exports.chat = asyncHandler(async (req, res, next) => {
     });
 
     response.data.on('end', async () => {
-      // Persist Messages after stream ends
-      await Message.create({ sessionId: session._id, sender: 'user', content: message || "[BINARY_UPLOAD]" });
-      await Message.create({ sessionId: session._id, sender: 'ai', content: fullAIResponse, modelUsed: model });
-      
-      user.usage.requestsToday += 1;
-      user.usage.lastRequestDate = new Date();
-      await user.save();
+      try {
+        // Persist Messages after stream ends
+        const userContent = message || (req.file ? `[Attached Image: ${req.file.originalname}]` : "[SIGNAL]");
+        await Message.create({ sessionId: session._id, sender: 'user', content: userContent });
+        await Message.create({ sessionId: session._id, sender: 'ai', content: fullAIResponse, modelUsed: model });
+        
+        user.usage.requestsToday += 1;
+        user.usage.lastRequestDate = new Date();
+        await user.save();
+      } catch (dbErr) {
+        console.error("[DATABASE_ERROR] Persistence failed:", dbErr.message);
+      }
       
       res.end();
     });

backend/services/vaultService.js

@@ -44,9 +44,13 @@ exports.uploadToVault = async (filePath, fileName) => {
         execSync(`git ${gitIdent} commit -m "Vault: Archived ${fileName} [Auto-Sync]"`, { cwd: rootDir });
         
         console.log("Vault: Pushing to remote...");
-        execSync(`git push "${remoteUrl}" main`, { cwd: rootDir });
+        if (process.env.SPACE_ID) {
+          console.log("Vault: [HF_SPACE_DETECTED] Skipping remote push to prevent restart loop.");
+        } else {
+          execSync(`git push "${remoteUrl}" main`, { cwd: rootDir });
+        }
         
-        console.log(`Vault: Sync Complete -> GitHub`);
+        console.log(`Vault: Sync Complete -> ${process.env.SPACE_ID ? 'Local Archive Only' : 'GitHub'}`);
     } catch (gitErr) {
         console.error('Vault: Git Command Failed:', gitErr.message);
         // We don't throw here to avoid crashing the chat, but we log the error