Update app.js

app.js

@@ -14,8 +14,8 @@ const PORT = process.env.PORT || 7860;
 app.use(cors());
 app.use(bodyParser.json({ limit: '50mb' }));
 
-const WORKER_PHASES = ["Worker: Analyzing Request...", "Worker: Reading Context...", "Worker: Checking Hierarchy...", "Worker: Thinking..."];
-const PM_PHASES = ["Manager: Reviewing...", "Manager: Formulating Strategy...", "Manager: Delegating...", "Manager: Thinking..."];
+const WORKER_PHASES = ["Worker: Analyzing Context...", "Worker: Triage Task...", "Worker: Planning Logic...", "Worker: Thinking..."];
+const PM_PHASES = ["Manager: Reviewing Request...", "Manager: Formulating Strategy...", "Manager: Preparing Delegation...", "Manager: Thinking..."];
 
 function startStatusLoop(projectId, type = 'worker') {
     const phases = type === 'pm' ? PM_PHASES : WORKER_PHASES;
@@ -31,8 +31,7 @@ function startStatusLoop(projectId, type = 'worker') {
 
 async function checkMinimumCredits(userId, type = 'basic') {
     const { data } = await supabase.from('users').select('credits').eq('id', userId).single();
-    const credits = data?.credits?.[type] || 0;
-    if (credits < 50) throw new Error(`Insufficient ${type} credits.`);
+    if (!data || (data.credits?.[type] || 0) < 50) throw new Error(`Insufficient credits.`);
 }
 
 async function deductUserCredits(userId, amount, type = 'basic') {
@@ -45,9 +44,9 @@ async function deductUserCredits(userId, amount, type = 'basic') {
 
 const formatContext = ({ hierarchyContext, scriptContext, logContext }) => {
     let out = "";
-    if (scriptContext) out += `\n[SCRIPT]: ${scriptContext.targetName}\n[SOURCE]: ${scriptContext.scriptSource}`;
-    if (logContext) out += `\n[LOGS]: ${logContext.logs}`;
-    if (hierarchyContext) out += `\n[HIERARCHY]: ${hierarchyContext}`;
+    if (scriptContext) out += `\n[TARGET SCRIPT]: ${scriptContext.targetName}\n[SOURCE PREVIEW]: ${scriptContext.scriptSource}`;
+    if (logContext) out += `\n[LAST LOGS]: ${logContext.logs}`;
+    if (hierarchyContext) out += `\n[Hierarchy Context]: ${hierarchyContext}`;
     return out;
 };
 
@@ -62,7 +61,7 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
         const project = await StateManager.getProject(projectId);
         StateManager.clearSnapshot(projectId);
         
-        // --- STEP 1: WORKER ---
+        // --- PASS 1: WORKER ---
         let stopStatus = startStatusLoop(projectId, 'worker');
         let workerPass1 = "";
         const result1 = await AIEngine.callWorkerStream(
@@ -75,12 +74,12 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
 
         const tags = extractTags(workerPass1);
         if (tags.pmQuestion || tags.pmRoute) {
-            // Commit Pass 1
+            // Save Pass 1 immediately
             await StateManager.addHistory(projectId, 'worker', 'user', fullInput);
             await StateManager.addHistory(projectId, 'worker', 'model', workerPass1);
             await StateManager.queueCommand(projectId, workerPass1);
 
-            // --- STEP 2: PROJECT MANAGER ---
+            // --- PASS 2: PROJECT MANAGER ---
             StateManager.clearSnapshot(projectId);
             stopStatus = startStatusLoop(projectId, 'pm');
             const pmInput = tags.pmRoute ? `[ROUTED TASK]: ${tags.pmRoute}` : `[QUESTION]: ${tags.pmQuestion}`;
@@ -96,18 +95,18 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
             await StateManager.addHistory(projectId, 'pm', 'model', pmText);
             await StateManager.queueCommand(projectId, pmText);
 
-            // Notify Worker of PM Answer
+            // Save PM answer into Worker's persistent history
             const pmAnswerVisible = `[PM RESPONSE]:\n${pmText}`;
             await StateManager.addHistory(projectId, 'worker', 'model', pmAnswerVisible);
 
-            // --- STEP 3: WORKER FINAL PASS ---
-            const nextTask = extractTags(pmText).workerPrompt || "Proceed with implementation.";
+            // --- PASS 3: WORKER FINAL PASS ---
+            const nextTask = extractTags(pmText).workerPrompt || "Proceed based on PM answer.";
             StateManager.clearSnapshot(projectId);
             stopStatus = startStatusLoop(projectId, 'worker');
             let workerPass2 = "";
-            const currentHistory = await StateManager.getProject(projectId); // Refresh history
+            const currentProj = await StateManager.getProject(projectId); // Refresh for Pass 2 context
             await AIEngine.callWorkerStream(
-                currentHistory.workerHistory, nextTask,
+                currentProj.workerHistory, nextTask,
                 (thought) => { stopStatus(); StateManager.appendSnapshotOnly(projectId, thought); },
                 (chunk) => { stopStatus(); workerPass2 += chunk; StateManager.appendStream(projectId, chunk); }
             );
@@ -121,9 +120,9 @@ async function runAsyncFeedback(projectId, userId, fullInput, images = []) {
         }
 
         await StateManager.updateProject(projectId, { status: "idle" });
-        await deductUserCredits(userId, 10); // Standard cost
+        await deductUserCredits(userId, 10);
     } catch (err) {
-        console.error("Feedback Loop Failed:", err.message);
+        console.error("Feedback Loop Crashed:", err.message);
         StateManager.setStatus(projectId, "Error: " + err.message);
         await StateManager.updateProject(projectId, { status: "error" });
     }
@@ -138,7 +137,7 @@ app.post('/project/feedback', async (req, res) => {
         await StateManager.updateProject(projectId, { status: "working" });
         res.json({ success: true });
         const context = formatContext({ hierarchyContext, scriptContext, logContext });
-        runAsyncFeedback(projectId, userId, `USER: ${prompt || "Auto Feedback"}${context}`, images || []);
+        runAsyncFeedback(projectId, userId, `USER: ${prompt || "Auto-Correction Loop"}${context}`, images || []);
     } catch (err) { res.status(500).json({ error: err.message }); }
 });
 
@@ -154,12 +153,11 @@ app.post('/project/ping', async (req, res) => {
     res.json(resData);
 });
 
-// Standard Handlers
 app.post('/onboarding/create', async (req, res) => {
     const { userId, description, answers } = req.body;
     const projectId = `proj_${Date.now()}_${crypto.randomBytes(2).toString("hex")}`;
     const grade = await AIEngine.gradeProject(description, answers);
-    await supabase.from('projects').insert({ id: projectId, user_id: userId, info: { title: grade.title, description, status: "idle" } });
+    await supabase.from('projects').insert({ id: projectId, user_id: userId, info: { title: grade.title, status: "idle" } });
     res.json({ success: true, projectId, grade });
 });