Update app.js

app.js

@@ -284,7 +284,10 @@ async function runBackgroundInitialization(projectId, userId, description) {
         await processAndQueueResponse(projectId, workerResponse);
         
         // 5. Update Status to IDLE (Ready for user input)
-        if(db) await db.ref(`projects/${projectId}/info/status`).set("IDLE");
+        if(db) await db.ref(`projects/${projectId}/info`).update({
+            status: "IDLE",
+            lastUpdated: Date.now()
+        });
         
         console.log(`[Background] Initialization complete for ${projectId}`);
 
@@ -336,7 +339,12 @@ app.post('/project/feedback', async (req, res) => {
         const nextInstruction = extractWorkerPrompt(pmResponse);
         if (!nextInstruction) {
              await StateManager.updateProject(projectId, { pmHistory: project.pmHistory, status: "IDLE" });
-             if(db) await db.ref(`projects/${projectId}/info/status`).set("IDLE");
+             if(db) {
+                 await db.ref(`projects/${projectId}/info`).update({
+                     status: "IDLE",
+                     lastUpdated: Date.now()
+                 });
+             }
              return res.json({ success: true, message: "No further tasks. Project Idle." });
         }
 
@@ -355,7 +363,14 @@ app.post('/project/feedback', async (req, res) => {
 
         StateManager.queueCommand(projectId, { type: "EXECUTE", payload: "warn('Starting Next Task...')" });
         await processAndQueueResponse(projectId, workerResponse);
-        if(db) await db.ref(`projects/${projectId}/info/status`).set("working"); // Keep working while plugin picks it up? Or IDLE to wait for plugin?
+        
+        // Still working on next task
+        if(db) {
+            await db.ref(`projects/${projectId}/info`).update({
+                status: "working",
+                lastUpdated: Date.now()
+            });
+        }
         return res.json({ success: true, message: "Next Task Assigned" });
     }
 
@@ -429,11 +444,14 @@ app.post('/project/feedback', async (req, res) => {
         });
         
         await processAndQueueResponse(projectId, response);
-        // Ensure status goes back to working or IDLE depending on your loop preference.
-        // Usually, if we just gave a command, we are 'working' until the plugin polls and finishes.
-        // But for UI feedback, 'working' usually means "AI is generating". 
-        // Once generated (here), we might want to set it to 'IDLE' or 'waiting_for_plugin'.
-        if(db) await db.ref(`projects/${projectId}/info/status`).set("idle");
+        
+        // --- UPDATED: Set status to IDLE and update lastUpdated ---
+        if(db) {
+            await db.ref(`projects/${projectId}/info`).update({
+                status: "idle",
+                lastUpdated: Date.now()
+            });
+        }
       
         res.json({ success: true });
     } catch (err) {