Update app.js

app.js

@@ -15,13 +15,13 @@ app.use(cors());
 app.use(bodyParser.json({ limit: "50mb" }));
 
 // === DEMO CONTROLLER ===
-// Options: 'NONE', 'BUILD_PART', 'FETCH_SCRIPT'
-const DEMO_MODE = 'FETCH_SCRIPT'; 
+// Options: 'NONE', 'BUILD_PART', 'FETCH_SCRIPT', 'FETCH_LOGS'
+const DEMO_MODE = 'FETCH_LOGS'; 
 
 app.post("/api/ai-build", async (req, res) => {
   try {
-    // We expect 'scriptContext' if this is a follow-up request from the plugin
-    const { instruction, state, scriptContext } = req.body;
+    // Destructure specifically to look for follow-up contexts
+    const { instruction, state, scriptContext, logContext } = req.body;
     
     console.log("📥 Received Request");
     console.log("   Instruction:", instruction);
@@ -30,54 +30,63 @@ app.post("/api/ai-build", async (req, res) => {
         console.log(`   Hierarchy Size: ${state.flatList.length} items (Flat Format)`);
     }
 
-    // 1. **FIXED LOOP LOGIC:** IF WE RECEIVED SCRIPT SOURCE (Step 2 of the loop)
+    // 1. **HANDLE SCRIPT SOURCE (Loop Step 2)**
     if (scriptContext) {
         console.log("   📘 Received Script Source for:", scriptContext.targetName);
         console.log("   --- START SOURCE ---\n", scriptContext.scriptSource.substring(0, 100) + "...", "\n   --- END SOURCE ---");
         
-        // **STOP THE LOOP:** Send back a simple JSON message instead of executable code.
         return res.json({ 
             success: true, 
-            message: `Source for ${scriptContext.targetName} received and analyzed. Loop stopped in demo mode.`
+            message: `Source for ${scriptContext.targetName} received. Loop stopped.`
         });
-        
-        // In a real scenario, you would perform AI analysis here and send back new code:
-        /*
+    }
+
+    // 2. **HANDLE LOGS (Loop Step 2)**
+    if (logContext) {
+        console.log("   📝 Received Console Logs:");
+        console.log("   --- LOGS ---\n", logContext.logs);
+        console.log("   --- END LOGS ---");
+
         return res.json({
-            code: `print("I have updated ${scriptContext.targetName}")`
+            success: true,
+            message: "Console logs received and analyzed. Loop stopped."
         });
-        */
     }
 
-    // 2. DEMO RESPONSES (Step 1 of the loop/initial request)
+    // 3. **DEMO TRIGGERS (Loop Step 1)**
+    
     if (DEMO_MODE === 'BUILD_PART') {
         const luaCode = `
         \`\`\`lua
-        -- AI Generated Demo: Part Builder
         local p = Instance.new("Part")
-        p.Name = "AI_Generated_Cube"
-        p.Color = Color3.fromRGB(0, 255, 100)
-        p.Size = Vector3.new(4, 4, 4)
-        p.Position = Vector3.new(0, 10, 0)
+        p.Name = "AI_Cube"
+        p.Color = Color3.new(1,0,0)
+        p.Position = Vector3.new(0, 5, 0)
         p.Anchored = true
         p.Parent = workspace
-        print("Hello from the Server! I made a part.")
+        print("Part created by AI")
         \`\`\`
         `;
         return res.send(luaCode);
     } 
     
     else if (DEMO_MODE === 'FETCH_SCRIPT') {
-        console.log("   👉 Sending Fetch Request Demo (Step 1)");
-        // This tells the plugin to find a script named "BikeLogic" and send it back
+        console.log("   👉 Sending Fetch Script Request");
         return res.json({
             action: "read_script",
-            targetName: "BikeLogic" 
+            targetName: "BikeLogic" // Ensure a script named this exists to test
+        });
+    }
+
+    else if (DEMO_MODE === 'FETCH_LOGS') {
+        console.log("   👉 Sending Fetch Logs Request");
+        return res.json({
+            action: "read_logs"
         });
     }
 
     // Default
-    return res.json({ success: true, assetId: "req_" + Date.now(), message: "Snapshot received. AI is thinking..." });
+    return res.json({ success: true, message: "Snapshot received. Waiting for demo trigger..." });
 
   } catch (err) {
     console.error("Server Error:", err);