Update app.js

app.js

@@ -1,7 +1,7 @@
 // server.js
 // Node + Express server to accept Snapshotter payloads from the Roblox plugin.
 // Save as server.js and run: node server.js
-// Installs: npm install express body-parser cors
+// Installs: npm install express body-parser cors node-fetch
 
 import express from "express";
 import bodyParser from "body-parser";
@@ -9,11 +9,7 @@ import cors from "cors";
 import fs from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
-import fetch from "node-fetch"; // Node 18+ has fetch globally; if not, install node-fetch
-
-// If you're running Node 18+, you can use global fetch and remove node-fetch import.
-// If you get "fetch is not defined", run: npm install node-fetch
-// and uncomment the import above (already present).
+import fetch from "node-fetch";
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
@@ -25,77 +21,39 @@ if (!fs.existsSync(STORAGE_DIR)) fs.mkdirSync(STORAGE_DIR, { recursive: true });
 
 const app = express();
 app.use(cors());
-app.use(bodyParser.json({ limit: "10mb" })); // Allow larger payloads (tweak as needed)
+// Increase limit to 50mb to handle large complex workspaces
+app.use(bodyParser.json({ limit: "50mb" }));
 
-// Simple logging middleware
+// Logging middleware
 app.use((req, res, next) => {
   console.log(new Date().toISOString(), req.method, req.url);
   next();
 });
 
-// Simple health/ping
+// Health check
 app.get("/ping", (req, res) => res.status(200).send("pong"));
 
-// Compatibility endpoint from your older plugin code
-app.post("/capture-and-upload", async (req, res) => {
+// Main endpoint for the Plugin
+app.post("/api/ai-build", async (req, res) => {
   try {
     const payload = req.body;
     if (!payload) return res.status(400).json({ success: false, error: "empty payload" });
 
-    // generate an id and save payload to disk
+    // Generate ID
     const id = "asset_" + Math.random().toString(36).slice(2, 10);
     const filename = path.join(STORAGE_DIR, `${id}.json`);
-    fs.writeFileSync(filename, JSON.stringify({ receivedAt: Date.now(), payload }, null, 2), "utf8");
 
-    // Optionally perform immediate processing here (call AI service, upload images...).
-    // For this demo we simply accept and return an assetId.
+    // Log what we received
+    console.log(`📥 Received snapshot from: ${payload.sender || "Unknown"}`);
+    console.log(`📝 Instruction: "${payload.instruction || "No instruction provided"}"`);
+    console.log(`Target Root: ${payload.selectionName}`);
 
-    // If client asked for a callbackUrl, try to POST back a small completion message
-    if (payload.callbackUrl && typeof payload.callbackUrl === "string") {
-      const cbBody = { assetId: id, status: "processed", ts: Date.now() };
-      // fire-and-forget; do not block the main response on callback success
-      (async () => {
-        try {
-          // Use fetch if available
-          if (typeof fetch === "function") {
-            const r = await fetch(payload.callbackUrl, {
-              method: "POST",
-              headers: { "Content-Type": "application/json" },
-              body: JSON.stringify(cbBody),
-              timeout: 10000
-            });
-            console.log("Callback posted, status:", r.status);
-          } else {
-            console.log("No fetch available to call callback URL.");
-          }
-        } catch (err) {
-          console.warn("Callback POST failed:", err && err.message ? err.message : err);
-        }
-      })();
-    }
-
-    return res.json({ success: true, assetId: id });
-  } catch (err) {
-    console.error("capture-and-upload error:", err);
-    return res.status(500).json({ success: false, error: String(err) });
-  }
-});
-
-// New route name used in the plugin snippet: /api/ai-build
-app.post("/api/ai-build", async (req, res) => {
-  try {
-    const payload = req.body;
-    if (!payload) return res.status(400).json({ success: false, error: "empty payload" });
-
-    // Save payload to disk
-    const id = "asset_" + Math.random().toString(36).slice(2, 10);
-    const filename = path.join(STORAGE_DIR, `${id}.json`);
+    // Save payload to disk (Game State + Instruction)
     fs.writeFileSync(filename, JSON.stringify({ receivedAt: Date.now(), payload }, null, 2), "utf8");
 
-    // (Here you could call the Gemini API, save images, run processing, etc.)
-    // For now we immediately return success with the generated asset id.
-
-    // If payload provides callbackUrl, POST a callback (fire-and-forget)
+    // FUTURE: Here is where you would trigger the Gemini API call using 'payload.instruction' and 'payload.state'
+    
+    // If client provided a callback URL, fire-and-forget
     if (payload.callbackUrl && typeof payload.callbackUrl === "string") {
       const cbBody = { assetId: id, status: "processed", ts: Date.now() };
       (async () => {
@@ -109,19 +67,20 @@ app.post("/api/ai-build", async (req, res) => {
             console.log("Callback sent to", payload.callbackUrl);
           }
         } catch (err) {
-          console.warn("Callback failed:", err && err.message ? err.message : err);
+          console.warn("Callback failed:", err.message);
         }
       })();
     }
 
-    return res.json({ success: true, assetId: id, message: "received" });
+    return res.json({ success: true, assetId: id, message: "Snapshot received successfully" });
+
   } catch (err) {
     console.error("/api/ai-build error:", err);
     return res.status(500).json({ success: false, error: String(err) });
   }
 });
 
-// Optional: status endpoint for an asset (reads saved JSON)
+// Retrieve stored assets
 app.get("/asset/:id", (req, res) => {
   try {
     const id = req.params.id;
@@ -135,8 +94,7 @@ app.get("/asset/:id", (req, res) => {
   }
 });
 
-// Start server
 app.listen(PORT, () => {
-  console.log(`Plugin backend listening on http://localhost:${PORT}`);
-  console.log("Saved payloads will be stored in:", STORAGE_DIR);
+  console.log(`🚀 Plugin backend listening on http://localhost:${PORT}`);
+  console.log(`📂 Storage: ${STORAGE_DIR}`);
 });