Update server.js

server.js

@@ -79,45 +79,85 @@ app.post('/new/project', validateRequest, async (req, res) => {
  * 2. FEEDBACK (The Main Loop Endpoint)
  * Accepts text + optional image. Hits Worker. Checks for escalation.
  */
+
 app.post('/project/feedback', async (req, res) => {
-    const { projectId, prompt, hierarchyContext, scriptContext, logContext } = req.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." });
 
-    // Build Context String based on what was sent
-    let contextStr = "";
-    if (prompt) contextStr += `USER: ${prompt}\n`;
-    
-    // If Plugin sent Hierarchy
-    if (hierarchyContext) {
-        contextStr += `[SYSTEM] HIERARCHY SCAN (${hierarchyContext.rootName}):\n${hierarchyContext.tree}\n`;
+    // 1. MISSION COMPLETE: CALL PM & NUKE WORKER
+    if (taskComplete) {
+        console.log(`[${projectId}] ✅ TASK COMPLETE. Calling PM & Nuking Worker.`);
+        
+        // A. Notify PM (Add to PM History)
+        const summary = `Worker has successfully completed the previous task. Logs: ${logContext?.logs || "Clean"}. Ready for next assignment.`;
+        project.pmHistory.push({ role: 'user', parts: [{ text: summary }] });
+        
+        // B. Nuke Worker (Reset History)
+        project.workerHistory = []; 
+        
+        // C. Update State
+        await StateManager.updateProject(projectId, { 
+            pmHistory: project.pmHistory,
+            workerHistory: [], // THE NUKE
+            status: "IDLE"     // Waiting for user or PM to start next task
+        });
+        
+        return res.json({ success: true, message: "Worker Nuked. PM Notified. Loop Stopped." });
     }
-    // If Plugin sent Script Source
-    if (scriptContext) {
-        contextStr += `[SYSTEM] SCRIPT SOURCE (${scriptContext.targetName}):\n${scriptContext.scriptSource}\n`;
+
+    // 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.`);
+        }
     }
-    // If Plugin sent Logs
-    if (logContext) {
-        contextStr += `[SYSTEM] CONSOLE LOGS:\n${logContext.logs}\n`;
+
+    // 3. BUILD AI PROMPT
+    const fullInput = `USER: ${prompt || "Automatic Feedback"}` + formatContext({ hierarchyContext, scriptContext, logContext });
+    
+    let finalPrompt = fullInput;
+    if (isFailure && (!prompt || prompt.length < 5)) {
+        finalPrompt += "\n[SYSTEM ALERT]: Runtime Error detected. Fix the code immediately. Do not use WaitForChild. Do not yield.";
     }
 
-    console.log(`[${projectId}] Input Received. Processing...`);
+    console.log(`[${projectId}] Processing Feedback...`);
 
-    // Call Worker
-    const response = await AIEngine.callWorker(project.workerHistory, contextStr);
-    
-    // Standard History Update
-    project.workerHistory.push({ role: 'user', parts: [{ text: contextStr }] });
-    project.workerHistory.push({ role: 'model', parts: [{ text: response }] });
-    await StateManager.updateProject(projectId, { workerHistory: project.workerHistory });
-    
-    // Parse and Queue the next move (Execute code or Fetch more info)
-    await StateManager.queueCommand(projectId, response);
+    // 4. CALL AI
+    try {
+        const response = await AIEngine.callWorker(project.workerHistory, finalPrompt);
+        
+        // Escalation Logic
+        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: response }] });
+        await StateManager.updateProject(projectId, { workerHistory: project.workerHistory });
+        await StateManager.queueCommand(projectId, response);
 
-    res.json({ success: true, message: "Context received. Worker processing." });
+        res.json({ success: true, message: "Input Processed." });
+
+    } catch (err) {
+        console.error("AI Error:", err);
+        res.status(500).json({ error: "AI Failed" });
+    }
 });
 
+
 /**
  * 3. INTERNAL PING WORKER
  * PM sends guidance to Worker.