Update app.js

app.js

@@ -122,6 +122,7 @@ function formatContext({ hierarchyContext, scriptContext, logContext }) {
 
 // --- CORE BACKGROUND LOGIC ---
 
+/*
 async function runBackgroundInitialization(projectId, userId, description) {
     console.log(`[Background] Starting initialization for ${projectId}`);
     let diamondUsage = 0; 
@@ -201,6 +202,97 @@ async function runBackgroundInitialization(projectId, userId, description) {
         await StateManager.updateProject(projectId, { status: "error" });
     }
 }
+*/
+
+// app.js - runBackgroundInitialization update
+
+async function runBackgroundInitialization(projectId, userId, description) {
+    // GUARD: Prevent double execution
+    if (StateManager.isLocked(projectId)) {
+        console.log(`[Init Guard] Project ${projectId} is already initializing. Skipping.`);
+        return;
+    }
+
+    StateManager.lock(projectId);
+    console.log(`[Background] Starting initialization for ${projectId}`);
+    
+    let diamondUsage = 0; 
+    let basicUsage = 0;   
+
+    try {
+        // --- 1. SET STATUS TO WORKING IMMEDIATELY ---
+        await StateManager.updateProject(projectId, { 
+            status: "WORKING",
+            gdd: "",
+            failureCount: 0 
+        });
+
+        const pmHistory = [];
+        
+        // --- 2. GENERATE GDD ---
+        const gddPrompt = `Create a comprehensive Game Design Document (GDD) for: ${description}`;
+        const gddResult = await AIEngine.callPM(pmHistory, gddPrompt);
+        
+        if (!gddResult?.text) throw new Error("PM failed to generate GDD");
+        
+        const gddText = gddResult.text;
+        diamondUsage += (gddResult.usage?.totalTokenCount || 0);
+
+        // Always back up to DB immediately
+        await StateManager.addHistory(projectId, 'pm', 'user', gddPrompt);
+        await StateManager.addHistory(projectId, 'pm', 'model', gddText);
+
+        // Update PM History for the next step context
+        pmHistory.push({ role: 'user', parts: [{ text: gddPrompt }] });
+        pmHistory.push({ role: 'model', parts: [{ text: gddText }] });
+
+        // --- 3. GENERATE FIRST TASK ---
+        const taskPrompt = "Based on the GDD, generate the first technical milestone.\nOutput format:\nTASK_NAME: <Name>\nWORKER_PROMPT: <Specific, isolated instructions for the worker>";
+        const taskResult = await AIEngine.callPM(pmHistory, taskPrompt);
+        
+        if (!taskResult?.text) throw new Error("PM failed to generate Task");
+
+        const taskText = taskResult.text;
+        diamondUsage += (taskResult.usage?.totalTokenCount || 0);
+
+        await StateManager.addHistory(projectId, 'pm', 'user', taskPrompt);
+        await StateManager.addHistory(projectId, 'pm', 'model', taskText);
+
+        // --- 4. INITIALIZE WORKER ---
+        const initialWorkerInstruction = extractWorkerPrompt(taskText) || `Initialize structure for: ${description}`;
+        const workerHistory = [];
+        const initialWorkerPrompt = `CONTEXT: New Project. \nINSTRUCTION: ${initialWorkerInstruction}`;
+        
+        const workerResult = await AIEngine.callWorker(workerHistory, initialWorkerPrompt, []);
+        if (!workerResult?.text) throw new Error("Worker failed to initialize");
+
+        basicUsage += (workerResult.usage?.totalTokenCount || 0);
+        const workerText = workerResult.text;
+
+        await StateManager.addHistory(projectId, 'worker', 'user', initialWorkerPrompt);
+        await StateManager.addHistory(projectId, 'worker', 'model', workerText);
+
+        // --- 5. FINISH & CLEANUP ---
+        await StateManager.updateProject(projectId, {
+            gdd: gddText,
+            status: "IDLE" // Set back to IDLE when done
+        });
+
+        await processAndQueueResponse(projectId, workerText, userId); 
+        
+        if (diamondUsage > 0) await deductUserCredits(userId, diamondUsage, 'diamond');
+        if (basicUsage > 0) await deductUserCredits(userId, basicUsage, 'basic');
+        
+        console.log(`[Background] Init complete for ${projectId}`);
+
+    } catch (err) {
+        console.error(`[Background] Init Error for ${projectId}:`, err.message);
+        await StateManager.updateProject(projectId, { status: "ERROR" });
+    } finally {
+        // ALWAYS unlock even if it fails
+        StateManager.unlock(projectId);
+    }
+}
 
 // --- ONBOARDING ENDPOINTS ---