Update app.js

app.js

@@ -1,4 +1,4 @@
-// server.js
+// server.js (Only the app.post function and surrounding context needed)
 import express from "express";
 import bodyParser from "body-parser";
 import cors from "cors";
@@ -12,13 +12,11 @@ const PORT = process.env.PORT || 7860;
 
 const app = express();
 app.use(cors());
-// Increased limit to 50mb to handle large complex workspaces
 app.use(bodyParser.json({ limit: "50mb" }));
 
 // === DEMO CONTROLLER ===
-// CHANGE THIS TO TEST DIFFERENT INTERACTIONS
-// Options: 'NONE' (Just acknowledge), 'BUILD_PART' (Send Lua Code), 'FETCH_SCRIPT' (Ask for Source)
-const DEMO_MODE = "FETCH_SCRIPT"; //'BUILD_PART'; 
+// Options: 'NONE', 'BUILD_PART', 'FETCH_SCRIPT'
+const DEMO_MODE = 'FETCH_SCRIPT'; // Keeping this set for the test
 
 app.post("/api/ai-build", async (req, res) => {
   try {
@@ -27,40 +25,38 @@ app.post("/api/ai-build", async (req, res) => {
     console.log("📥 Received Request");
     console.log("   Instruction:", instruction);
     
-    // 1. Log Payload Size (Check token efficiency)
     if(state && state.flatList) {
         console.log(`   Hierarchy Size: ${state.flatList.length} items (Flat Format)`);
- // console.log(`Flatlist : ${JSON.stringify(state.flatList).length}`);
- 
+    //console.log(`Flatlist : ${JSON.stringify(state.flatList).length}`);
     }
 
-    // 2. IF WE RECEIVED SCRIPT SOURCE (From a previous fetch request)
+    // 1. **FIXED LOOP LOGIC:** IF WE RECEIVED SCRIPT SOURCE (Step 2 of the loop)
     if (scriptContext) {
         console.log("   📘 Received Script Source for:", scriptContext.targetName);
         console.log("   --- START SOURCE ---\n", scriptContext.scriptSource.substring(0, 100) + "...", "\n   --- END SOURCE ---");
-        // In a real app, you would now feed this to the AI to edit.
-        // For demo purposes, we can acknowledge and then send code back:
         
-        const returnCode = `
+        // **STOP THE LOOP:** Send back a simple JSON message instead of executable code.
+        // In a real application, the AI would generate the final Lua code here.
+        return res.json({ 
+            success: true, 
+            message: `Source for ${scriptContext.targetName} received and analyzed. Loop stopped in demo mode.`
+        });
+        
+        // If you were testing the final code output, you would send the code here:
+        /*
+        const finalCode = `
         \`\`\`lua
-        -- AI analyzed script ${scriptContext.targetName}. Sending back a confirmation part.
-        local p = Instance.new("Part")
-        p.Name = "AI_SCRIPT_ACKNOWLEDGED"
-        p.Color = Color3.fromRGB(255, 255, 0)
-        p.Size = Vector3.new(2, 2, 2)
-        p.Position = Vector3.new(5, 5, 0)
-        p.Anchored = true
-        p.Parent = workspace
-        print("Script analysis complete. Confirmation part created.")
+        -- Final AI-modified code for ${scriptContext.targetName}
+        print("Final Code Executed!")
         \`\`\`
         `;
-        return res.send(returnCode);
+        return res.send(finalCode);
+        */
     }
 
-    // 3. DEMO RESPONSES
+    // 2. DEMO RESPONSES (Step 1 of the loop/initial request)
     if (DEMO_MODE === 'BUILD_PART') {
-        console.log("   👉 Sending Code Demo");
-        // FIX: Using template literals (backticks `) for multi-line string
+        // ... (BUILD_PART logic remains the same)
         const luaCode = `
         \`\`\`lua
         -- AI Generated Demo: Part Builder
@@ -74,19 +70,19 @@ app.post("/api/ai-build", async (req, res) => {
         print("Hello from the Server! I made a part.")
         \`\`\`
         `;
-        return res.send(luaCode); // Sending raw markdown/text
+        return res.send(luaCode);
     } 
     
     else if (DEMO_MODE === 'FETCH_SCRIPT') {
-        console.log("   👉 Sending Fetch Request Demo");
+        console.log("   👉 Sending Fetch Request Demo (Step 1)");
         // This tells the plugin to find a script named "BikeLogic" and send it back
         return res.json({
             action: "read_script",
-            targetName: "BikeLogic" // CHANGE THIS to a script name that actually exists in your studio
+            targetName: "BikeLogic" 
         });
     }
 
-    // Default: Just acknowledge
+    // Default
     return res.json({ success: true, assetId: "req_" + Date.now(), message: "Snapshot received. AI is thinking..." });
 
   } catch (err) {