Update app.js

app.js

@@ -1,4 +1,4 @@
-// server.js - ALIGNED WITH PLUGIN
+// server.js - ALIGNED WITH PLUGIN & FIXED SYNTAX
 // Run: GEMINI_API_KEY=your_key node server.js
 
 import express from "express";
@@ -25,7 +25,6 @@ const simpleId = (prefix = "") => prefix + Math.random().toString(36).slice(2, 9
 /**
  * STRICT VOCABULARY
  * Only actions that exist in the Plugin's ACTION_HANDLERS are allowed here.
- * Removed: create_union, create_rig, apply_animation, unweld, apply_texture
  */
 const VALID_ACTIONS = [
   "create_part",
@@ -40,7 +39,7 @@ const VALID_ACTIONS = [
   "set_parent",
   "delete",
   "weld",
-  "list_workspace", // Plugin calls this workspaceSnapshot
+  "list_workspace",
   "finish"
 ];
 
@@ -69,12 +68,12 @@ const ACTION_SYNONYMS = {
   "group": "create_model", 
   "parent": "set_parent",
   "set_parent_to": "set_parent",
-  "join": "weld", -- "Join" usually implies welding in this simple context
+  "join": "weld", // Fixed: JS comment syntax
   
   // Unsupported fallbacks (Safety)
-  "union": "create_model", -- Fallback: Group them instead of Unioning (prevents crash)
+  "union": "create_model", // Fixed: JS comment syntax
   "create_union": "create_model",
-  "create_rig": "create_model" -- Fallback: Just make a model container
+  "create_rig": "create_model" // Fixed: JS comment syntax
 };
 
 function validateAndPrepareTasks(arr) {
@@ -100,8 +99,6 @@ function validateAndPrepareTasks(arr) {
     }
 
     if (!canonicalAction) {
-      // If AI hallucinates an action, we force it to a safe fallback or error
-      // Instead of crashing, let's log it and skip or error.
       throw new Error(`Invalid action '${actionRaw}'. Supported: ${VALID_ACTIONS.join(", ")}`);
     }
 
@@ -119,7 +116,6 @@ function validateAndPrepareTasks(arr) {
 }
 
 async function generateTasks(project) {
-  // Summarize project state
   const historySample = (project.history || []).slice(-15).map(h => ({
     status: h.status,
     action: h.originalAction || "unknown",
@@ -128,7 +124,6 @@ async function generateTasks(project) {
   
   const status = project.status || "unknown";
 
-  // We explicitly tell the AI about the Registry logic in the prompt
   const systemPrompt = `
 You are a Roblox Studio Assistant. Convert user descriptions into JSON tasks.
 
@@ -152,17 +147,18 @@ User Request:
 ${JSON.stringify({ description: project.description })}
 `;
 
+  // Use gemini-1.5-flash or 2.0-flash depending on what you have access to.
+  // 1.5-flash is very stable.
   const resp = await ai.models.generateContent({
-    model: "gemini-2.0-flash", // Updated to latest flash if available, or use 1.5-flash
+    model: "gemini-1.5-flash", 
     contents: systemPrompt,
-    generationConfig: { responseMimeType: "application/json" } // Force JSON mode
+    generationConfig: { responseMimeType: "application/json" }
   });
 
   const text = resp?.text ? resp.text() : (resp?.response?.text() ?? "");
   
   let arr;
   try {
-    // Clean markdown code blocks if present
     const cleanText = text.replace(/```json/g, "").replace(/```/g, "").trim();
     arr = JSON.parse(cleanText);
   } catch (err) {
@@ -173,7 +169,7 @@ ${JSON.stringify({ description: project.description })}
   return validateAndPrepareTasks(arr);
 }
 
-// --- EXPRESS ROUTES (Same as before, simplified) ---
+// --- EXPRESS ROUTES ---
 
 app.post("/projects", async (req, res) => {
   try {
@@ -189,7 +185,7 @@ app.post("/projects", async (req, res) => {
       _refs: {}
     };
     projects.set(projectId, project);
-    runProject(project); // Background run
+    runProject(project); 
     return res.status(202).json({ projectId, status: "accepted" });
   } catch (err) {
     return res.status(500).json({ error: err.message });
@@ -201,14 +197,12 @@ app.post("/projects/:projectId/prompt", async (req, res) => {
   if (!project) return res.status(404).json({ error: "project not found" });
   
   const prompt = req.body.prompt;
-  // Append new prompt to description acts as "memory" for this simple prototype
   project.description += `\nUser Update: ${prompt}`;
   
-  // Re-run generation logic for the new prompt
   try {
-    const newTasks = await generateTasks({ ...project, description: prompt }); // Only send the *new* prompt as the task trigger
+    const newTasks = await generateTasks({ ...project, description: prompt });
     project.tasks.push(...newTasks);
-    project.commandQueue.push(...newTasks); // Push directly to queue if already running
+    project.commandQueue.push(...newTasks);
     return res.json({ tasks: newTasks });
   } catch(e) {
     return res.status(500).json({error: e.message});
@@ -230,10 +224,8 @@ app.post("/projects/:projectId/result", (req, res) => {
   
   const { commandId, status, message, target_ref } = req.body;
   
-  // Log history
   project.history.push({ commandId, status, message, timestamp: Date.now() });
   
-  // Resolve waiting promise (if runProject is waiting)
   if (project.pendingResults.has(commandId)) {
     project.pendingResults.get(commandId).resolve({ status, message, target_ref });
     project.pendingResults.delete(commandId);
@@ -252,9 +244,6 @@ async function runProject(project) {
        return;
      }
   }
-
-  // In this simplified version, we just let the /next endpoint drain the queue.
-  // We don't strictly "block" here, making it more robust for connection drops.
 }
 
 app.listen(PORT, () => console.log(`BloxBuddy Brain running on ${PORT}`));