Create aiEngine.js

aiEngine.js

@@ -0,0 +1,105 @@
+import { GoogleGenAI } from '@google/genai';
+import fs from 'fs';
+
+// Initialize SDK
+const genAI = new GoogleGenAI({ apiKey: process.env.GEMINI_API_KEY });
+const prompts = JSON.parse(fs.readFileSync('./prompts.json', 'utf8'));
+
+export const AIEngine = {
+  /**
+   * 1. PROJECT MANAGER (Reasoning & Delegation)
+   */
+  callPM: async (history, input) => {
+    const modelId = 'gemini-2.0-pro-exp-02-05'; 
+    const config = {
+      thinkingConfig: { thinkingLevel: 'HIGH' }, 
+      tools: [{ googleSearch: {} }],
+      systemInstruction: { parts: [{ text: prompts.pm_system_prompt }] }
+    };
+
+    const contents = [...history, { role: 'user', parts: [{ text: input }] }];
+
+    try {
+      const response = await genAI.models.generateContent({ model: modelId, config, contents });
+      return response.text(); 
+    } catch (error) {
+      console.error("PM AI Error:", error);
+      throw error;
+    }
+  },
+
+  /**
+   * 2. WORKER (Coding & Execution)
+   */
+  callWorker: async (history, input, images = []) => {
+   /* const modelId = 'gemini-2.0-flash';
+    const config = {
+      tools: [{ googleSearch: {} }],
+      systemInstruction: { parts: [{ text: prompts.worker_system_prompt }] }
+    };
+    */
+    const modelId = 'gemini-2.0-pro-exp-02-05'; 
+    const config = {
+      thinkingConfig: { thinkingLevel: 'HIGH' }, 
+      tools: [{ googleSearch: {} }],
+      systemInstruction: { parts: [{ text: prompts.worker_system_prompt }] }
+    };
+    
+
+    const currentParts = [{ text: input }];
+    
+    // Process incoming images from Roblox to the AI
+    if (images && images.length > 0) {
+      images.forEach(base64String => {
+        const cleanData = base64String.replace(/^data:image\/\w+;base64,/, "");
+        currentParts.push({
+            inlineData: { mimeType: "image/png", data: cleanData }
+        });
+      });
+    }
+
+    const contents = [...history, { role: 'user', parts: currentParts }];
+
+    try {
+      const response = await genAI.models.generateContent({ model: modelId, config, contents });
+      return response.text();
+    } catch (error) {
+      console.error("Worker AI Error:", error);
+      throw error;
+    }
+  },
+
+  /**
+   * 3. IMAGE GENERATOR (Visual Assets)
+   * Uses Imagen 3 to generate textures/UI assets
+   */
+  generateImage: async (prompt) => {
+    const modelId = 'imagen-3.0-generate-001';
+    
+    try {
+      console.log(`[AIEngine] Generating Image for: "${prompt}"...`);
+      const response = await genAI.models.generateImages({
+        model: modelId,
+        prompt: prompt,
+        config: {
+          numberOfImages: 1,
+          aspectRatio: "1:1", // Standard for textures/icons
+          outputMimeType: "image/png" // Roblox supports PNG
+        }
+      });
+
+      // The SDK returns the image data directly in the response
+      const image = response.generatedImages[0];
+      
+      if (!image || !image.image) {
+        throw new Error("No image data received from Imagen.");
+      }
+       console.log("Yay, image generated")
+      // Return Base64 string
+      return image.image.toString('base64');
+    } catch (error) {
+      console.error("Image Gen Error:", error);
+      return null; // Fail gracefully so the server doesn't crash
+    }
+  }
+};