Update app.js

app.js

@@ -11,85 +11,65 @@ const PORT = process.env.PORT || 7860;
 
 const app = express();
 app.use(cors());
-// Increased limit for large snapshot payloads
 app.use(bodyParser.json({ limit: "50mb" }));
 
-// === DEMO CONTROLLER ===
-// Options: 'NONE', 'BUILD_PART', 'FETCH_SCRIPT', 'FETCH_LOGS'
-const DEMO_MODE = 'FETCH_SCRIPT'; // 'FETCH_LOGS'; 
+// Options: 'NONE', 'BUILD_PART', 'FETCH_SCRIPT', 'FETCH_LOGS', 'SCAN_HIERARCHY'
+const DEMO_MODE = 'SCAN_HIERARCHY'; 
 
 app.post("/api/ai-build", async (req, res) => {
   try {
-    // Destructure specifically to look for follow-up contexts
-    const { instruction, state, scriptContext, logContext } = req.body;
+    const { instruction, selection, scriptContext, logContext, hierarchyContext } = req.body;
     
-    console.log("📥 Received Request");
-    console.log("   Instruction:", instruction);
+    console.log("------------------------------------------------");
+    console.log("📥 INSTRUCTION:", instruction);
+    if(selection) console.log("   Selected:", selection.name, `(${selection.className})`);
     
-    if(state && state.flatList) {
-        console.log(`   Hierarchy Size: ${state.flatList.length} items (Flat Format)`);
-      const lll = JSON.stringify(state.flatList); 
-      console.log(` Flat list: ${lll.length}`);
-     
-    }
-
-    // 1. **HANDLE SCRIPT SOURCE (Loop Step 2)**
+    // 1. Script Loop
     if (scriptContext) {
-        console.log("   📘 Received Script Source for:", scriptContext.targetName);
-        console.log("   --- START SOURCE ---\n", scriptContext.scriptSource.substring(0, 100) + "...", "\n   --- END SOURCE ---");
-        
-        return res.json({ 
-            success: true, 
-            message: `Source for ${scriptContext.targetName} received. Loop stopped.`
-        });
+        console.log(`📘 SCRIPT [${scriptContext.targetName}]:`);
+        console.log(scriptContext.scriptSource.substring(0, 150) + "...\n");
+        return res.json({ success: true, message: `Source received.` });
     }
 
-    // 2. **HANDLE LOGS (Loop Step 2)**
+    // 2. Log Loop
     if (logContext) {
-        console.log("   📝 Received Console Logs:");
-        console.log("   --- LOGS ---\n", logContext.logs);
-        console.log("   --- END LOGS ---");
+        console.log("📝 LOGS RECEIVED:\n", logContext.logs);
+        return res.json({ success: true, message: "Logs analyzed." });
+    }
 
-        return res.json({
-            success: true,
-            message: "Console logs received and analyzed. Loop stopped."
-        });
+    // 3. Hierarchy Loop (New)
+    if (hierarchyContext) {
+        console.log("🌳 HIERARCHY RECEIVED:");
+        console.log(hierarchyContext.tree);
+        return res.json({ success: true, message: "Hierarchy analyzed." });
     }
 
-    // 3. **DEMO TRIGGERS (Loop Step 1)**
-    
+    // 4. Demo Triggers (Initial Request)
     if (DEMO_MODE === 'BUILD_PART') {
         const luaCode = `
         \`\`\`lua
         local p = Instance.new("Part")
-        p.Name = "AI_Cube"
-        p.Color = Color3.new(1,0,0)
-        p.Position = Vector3.new(0, 5, 0)
-        p.Anchored = true
+        p.Name = "AI_Part"
+        p.Color = Color3.new(0,1,0)
+        p.Position = Vector3.new(0, 10, 0)
         p.Parent = workspace
-        print("Part created by AI")
         \`\`\`
         `;
         return res.send(luaCode);
     } 
-    
     else if (DEMO_MODE === 'FETCH_SCRIPT') {
-        console.log("   👉 Sending Fetch Script Request");
-        return res.json({
-            action: "read_script",
-            targetName: "BikeLogic" // Ensure a script named this exists to test
-        });
+        return res.json({ action: "read_script", targetName: "BikeLogic" });
     }
-
     else if (DEMO_MODE === 'FETCH_LOGS') {
-        console.log("   👉 Sending Fetch Logs Request");
-        return res.json({
-            action: "read_logs"
-        });
+        return res.json({ action: "read_logs" });
+    }
+    else if (DEMO_MODE === 'SCAN_HIERARCHY') {
+        // AI decides it needs to see the world before acting
+        console.log("   👉 Requesting Hierarchy Scan...");
+        return res.json({ action: "read_hierarchy" });
     }
 
-    // Default
-    return res.json({ success: true, message: "Snapshot received. Waiting for demo trigger..." });
+    return res.json({ success: true, message: "Thinking..." });
 
   } catch (err) {
     console.error("Server Error:", err);
@@ -97,4 +77,4 @@ app.post("/api/ai-build", async (req, res) => {
   }
 });
 
-app.listen(PORT, () => console.log(`🚀 Server running on port ${PORT}. Demo Mode: ${DEMO_MODE}`));
+app.listen(PORT, () => console.log(`🚀 Server running on port ${PORT}. Mode: ${DEMO_MODE}`));