Update app.js

app.js

@@ -94,41 +94,76 @@ async function startBackgroundInit(projectId, description) {
 // =======================================================
 // 2. FEEDBACK
 // =======================================================
+
 app.post('/project/feedback', async (req, res) => {
-    const { projectId, prompt, hierarchyContext, scriptContext, logContext } = req.body;
+    // Added 'taskComplete' to the destructured body
+    const { projectId, prompt, hierarchyContext, scriptContext, logContext, taskComplete } = req.body;
     
     const project = await StateManager.getProject(projectId);
-    if (!project) return res.status(404).json({ error: "Project not found. Run /new/project first." });
+    if (!project) return res.status(404).json({ error: "Project not found." });
 
-    const fullInput = `USER: ${prompt || "Automatic Feedback"}` + formatContext({ hierarchyContext, scriptContext, logContext });
+    // 1. CHECK FOR LOOP STOPPER
+    if (taskComplete) {
+        console.log(`[${projectId}] ✅ TASK COMPLETE. Stopping Feedback Loop.`);
+        
+        // We log it to history but DO NOT call the Worker AI
+        project.workerHistory.push({ role: 'user', parts: [{ text: "USER: Task Execution Confirmed. Logs: " + (logContext?.logs || "OK") }] });
+        await StateManager.updateProject(projectId, { workerHistory: project.workerHistory });
+        
+        return res.json({ success: true, message: "Loop Stopped. Waiting for new instructions." });
+    }
 
-    console.log(`[${projectId}] Feedback received. Processing...`);
+    // 2. ERROR DETECTION (Safety Net)
+    let isFailure = false;
+    if (logContext && logContext.logs) {
+        const errorKeywords = ["Infinite yield", "Stack Begin", "Error:", "Exception", "failed"];
+        if (errorKeywords.some(keyword => logContext.logs.includes(keyword))) {
+            isFailure = true;
+            console.log(`[${projectId}] ⚠️ Detected Error in Logs.`);
+        }
+    }
 
-    // Call Worker
-    const response = await AIEngine.callWorker(project.workerHistory, fullInput);
+    // 3. BUILD AI PROMPT
+    const fullInput = `USER: ${prompt || "Automatic Feedback"}` + formatContext({ hierarchyContext, scriptContext, logContext });
     
-    // Check Escalation
-    if (response.includes("STATUS: ESCALATE_TO_PM") || (project.failureCount > 2)) {
-        console.log(`[${projectId}] Escalating to PM...`);
-        const guidance = await AIEngine.callPM(project.pmHistory, `Worker Stuck. Input: ${fullInput}`);
-        const fixed = await AIEngine.callWorker(project.workerHistory, `PM GUIDANCE: ${guidance}`);
+    let finalPrompt = fullInput;
+    if (isFailure && (!prompt || prompt.length < 5)) {
+        finalPrompt += "\n[SYSTEM ALERT]: The logs indicate a Runtime Error or Infinite Yield. Fix the code immediately. Do not use WaitForChild.";
+    }
+
+    console.log(`[${projectId}] Processing Feedback...`);
+
+    // 4. CALL AI (Continue Loop)
+    try {
+        const response = await AIEngine.callWorker(project.workerHistory, finalPrompt);
         
-        await StateManager.updateProject(projectId, { failureCount: 0 });
+        // Check Escalation
+        if (response.includes("STATUS: ESCALATE_TO_PM") || (project.failureCount > 2)) {
+            console.log(`[${projectId}] Escalating to PM...`);
+            const guidance = await AIEngine.callPM(project.pmHistory, `Worker Stuck. Input: ${fullInput}`);
+            const fixed = await AIEngine.callWorker(project.workerHistory, `PM GUIDANCE: ${guidance}`);
+            
+            await StateManager.updateProject(projectId, { failureCount: 0 });
+            project.workerHistory.push({ role: 'user', parts: [{ text: fullInput }] });
+            project.workerHistory.push({ role: 'model', parts: [{ text: fixed }] });
+            await StateManager.queueCommand(projectId, fixed);
+            return res.json({ success: true, message: "Escalated & Fixed." });
+        }
+
         project.workerHistory.push({ role: 'user', parts: [{ text: fullInput }] });
-        project.workerHistory.push({ role: 'model', parts: [{ text: fixed }] });
-        await StateManager.queueCommand(projectId, fixed);
-        return res.json({ success: true, message: "Escalated & Fixed." });
-    }
+        project.workerHistory.push({ role: 'model', parts: [{ text: response }] });
+        await StateManager.updateProject(projectId, { workerHistory: project.workerHistory });
+        await StateManager.queueCommand(projectId, response);
 
-    // Normal Success
-    project.workerHistory.push({ role: 'user', parts: [{ text: fullInput }] });
-    project.workerHistory.push({ role: 'model', parts: [{ text: response }] });
-    await StateManager.updateProject(projectId, { workerHistory: project.workerHistory });
-    await StateManager.queueCommand(projectId, response);
+        res.json({ success: true, message: "Input Processed." });
 
-    res.json({ success: true, message: "Input Processed." });
+    } catch (err) {
+        console.error("AI Error:", err);
+        res.status(500).json({ error: "AI Failed" });
+    }
 });
-
+;
+    
 // =======================================================
 // 3. INTERNAL / PING
 // =======================================================