Update app.js

app.js

@@ -79,59 +79,43 @@ 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', validateRequest, async (req, res) => {
-    const { projectId, prompt, imageBase64 } = req.body;
+app.post('/project/feedback', async (req, res) => {
+    const { projectId, prompt, hierarchyContext, scriptContext, logContext } = req.body;
+    
+    const project = await StateManager.getProject(projectId);
+    if (!project) return res.status(404).json({ error: "Project not found." });
 
-    try {
-        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`;
+    }
+    // If Plugin sent Script Source
+    if (scriptContext) {
+        contextStr += `[SYSTEM] SCRIPT SOURCE (${scriptContext.targetName}):\n${scriptContext.scriptSource}\n`;
+    }
+    // If Plugin sent Logs
+    if (logContext) {
+        contextStr += `[SYSTEM] CONSOLE LOGS:\n${logContext.logs}\n`;
+    }
 
-        let imagePart = null;
-        if (imageBase64) {
-            imagePart = { inlineData: { mimeType: 'image/png', data: imageBase64 } };
-        }
+    console.log(`[${projectId}] Input Received. Processing...`);
 
-        // Call Worker
-        const response = await AIEngine.callWorker(project.workerHistory, prompt, imagePart);
-        
-        // Update History
-        project.workerHistory.push({ role: 'user', parts: [{ text: prompt }] });
-        project.workerHistory.push({ role: 'model', parts: [{ text: response }] });
-
-        // Check for Escalation Condition (Keyword or Failure Count)
-        if (response.includes("STATUS: ESCALATE_TO_PM") || project.failureCount >= 2) {
-            console.log(`Project ${projectId} escalating to PM.`);
-            
-            // Auto-trigger internal PM ping
-            // We reuse the logic from /internal/ping/pm here for efficiency
-            const distressSignal = `Worker failed to solve: ${prompt}. Last response: ${response}`;
-            const pmGuidance = await AIEngine.callPM(
-                project.pmHistory, 
-                sysPrompts.pm_guidance_prompt.replace('{{DISTRESS_SIGNAL}}', distressSignal)
-            );
-
-            // Send guidance back to worker
-            const fixedWorkerResponse = await AIEngine.callWorker(project.workerHistory, `PM GUIDANCE: ${pmGuidance}. Try again.`);
-            
-            project.failureCount = 0; // Reset failure count
-            StateManager.queueCommand(projectId, fixedWorkerResponse);
-            await StateManager.updateProject(projectId, { pmHistory: project.pmHistory, workerHistory: project.workerHistory, failureCount: 0 });
-            
-            return res.json({ success: true, message: "Escalated to PM and Resolved", type: "ESCALATION" });
-        } else {
-            // Standard Success
-            StateManager.queueCommand(projectId, response);
-            await StateManager.updateProject(projectId, { workerHistory: project.workerHistory });
-            return res.json({ success: true, message: "Processed by Worker" });
-        }
+    // 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);
 
-    } catch (err) {
-        const project = await StateManager.getProject(projectId);
-        if (project) {
-            await StateManager.updateProject(projectId, { failureCount: (project.failureCount || 0) + 1 });
-        }
-        res.status(500).json({ error: err.message });
-    }
+    res.json({ success: true, message: "Context received. Worker processing." });
 });
 
 /**
@@ -200,6 +184,7 @@ app.post('/project/reset', validateRequest, async (req, res) => {
  * 6. PROJECT PING (The Roblox Plugin Endpoint)
  * Plugin calls this to fetch executable code.
  */
+/*
 app.post('/project/ping', validateRequest, async (req, res) => {
     const { projectId } = req.body;
     const project = await StateManager.getProject(projectId);
@@ -219,6 +204,7 @@ app.post('/project/ping', validateRequest, async (req, res) => {
         codeSnippet: command.code // The markdown snippet extracted in StateManager
     });
 });
+*/
 
 /**
  * 7. INACTIVITY CLEANUP
@@ -228,6 +214,22 @@ app.delete('/internal/cleanup', async (req, res) => {
     res.json({ success: true, removed: count });
 });
 
+app.post('/project/ping', async (req, res) => {
+    const { projectId } = req.body;
+    const command = await StateManager.popCommand(projectId);
+
+    if (command) {
+        // Sends: { action: "EXECUTE", target: "...", code: "..." }
+        res.json({ 
+            action: command.type, 
+            target: command.payload, 
+            code: command.type === 'EXECUTE' ? command.payload : null 
+        });
+    } else {
+        res.json({ action: "IDLE" });
+    }
+});
+
 app.listen(PORT, () => {
     console.log(`AI Builder Backend running on port ${PORT}`);
 });