Update app.js

app.js

@@ -290,9 +290,7 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
         let finalResponseToSave = firstTurnResponse;
 
         if (routeTask || pmQuestion) {
-            // --- CRITICAL FIX: Save the Worker's first turn immediately ---
-            // This ensures if we crash or reload, the question/delegation is preserved.
-            // It also fixes the "Amnesia" where the worker forgets it asked a question.
+            // --- Save the Worker's first turn immediately ---
             await StateManager.addHistory(projectId, 'worker', 'user', fullInput);
             await StateManager.addHistory(projectId, 'worker', 'model', firstTurnResponse);
             
@@ -335,12 +333,16 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
                 }
             );
             
-            diamondUsage += 0; // Adjust if tracking PM usage
+            diamondUsage += 0; 
             
             // Save PM History
             await StateManager.addHistory(projectId, 'pm', 'user', pmPrompt);
             await StateManager.addHistory(projectId, 'pm', 'model', pmResponseText);
 
+            // --- CRITICAL FIX: EXECUTE PM CODE ---
+            // If the PM wrote code (backend logic), execute it now.
+            await processAndQueueResponse(projectId, pmResponseText, userId);
+
             // 5. Run Worker Continuation (The "Answer" Phase)
             const nextInstruction = extractWorkerPrompt(pmResponseText) || pmResponseText;
             const workerContinuationPrompt = `${pmContextPrefix} ${nextInstruction}\n\nBased on this, continue the task and output the code.`;
@@ -351,7 +353,7 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
             let secondTurnResponse = "";
             
             await AIEngine.callWorkerStream(
-                currentWorkerHistory, // Uses the updated history including the Question
+                currentWorkerHistory, 
                 workerContinuationPrompt,
                 (thought) => {
                     stopStatus();
@@ -370,7 +372,7 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
             await StateManager.addHistory(projectId, 'worker', 'user', workerContinuationPrompt);
             await StateManager.addHistory(projectId, 'worker', 'model', secondTurnResponse);
             
-            // Update the variable for execution
+            // Update the variable for the FINAL execution (Worker's part)
             finalResponseToSave = secondTurnResponse;
         } else {
             // Standard path (No PM needed), save history normally
@@ -380,8 +382,8 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
 
         StateManager.setStatus(projectId, "Idle");
         
-        // 6. Execute Logic
-        // We execute 'finalResponseToSave' which is either the simple response OR the post-PM complex response
+        // 6. Execute Logic (Worker's part)
+        // If PM ran before this, their code is already queued. Now we queue the Worker's code.
         await processAndQueueResponse(projectId, finalResponseToSave, userId);
 
         // 7. Update Status & Timestamp