Update aiEngine.js

aiEngine.js

@@ -1,26 +1,42 @@
 import { GoogleGenAI } from '@google/genai';
 import fs from 'fs';
+import path from 'path';
+
+// Load prompts safely
+const promptsPath = path.resolve('./prompts.json');
+const prompts = JSON.parse(fs.readFileSync(promptsPath, 'utf8'));
 
 // Initialize SDK
+// Make sure process.env.GEMINI_API_KEY is set in your environment
 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)
+   * Uses High-Reasoning Model
    */
   callPM: async (history, input) => {
-    const modelId = 'gemini-3-pro-preview'; 
+    const modelId = 'gemini-2.0-pro-exp-02-05'; 
+    
     const config = {
       thinkingConfig: { thinkingLevel: 'HIGH' }, 
       tools: [{ googleSearch: {} }],
-      systemInstruction: { parts: [{ text: prompts.pm_system_prompt }] }
+      systemInstruction: {
+        parts: [{ text: prompts.pm_system_prompt }]
+      }
     };
 
-    const contents = [...history, { role: 'user', parts: [{ text: input }] }];
+    const contents = [
+      ...history,
+      { role: 'user', parts: [{ text: input }] }
+    ];
 
     try {
-      const response = await genAI.models.generateContent({ model: modelId, config, contents });
+      const response = await genAI.models.generateContent({
+        model: modelId,
+        config,
+        contents,
+      });
       return response.text(); 
     } catch (error) {
       console.error("PM AI Error:", error);
@@ -30,38 +46,45 @@ export const AIEngine = {
 
   /**
    * 2. WORKER (Coding & Execution)
+   * Uses Flash Model (Fast) + Image Support
    */
   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-3-pro-preview'; 
+    const modelId = 'gemini-2.0-flash';
+    
     const config = {
-      thinkingConfig: { thinkingLevel: 'HIGH' }, 
       tools: [{ googleSearch: {} }],
-      systemInstruction: { parts: [{ text: prompts.worker_system_prompt }] }
+      systemInstruction: {
+        parts: [{ text: prompts.worker_system_prompt }]
+      }
     };
-    
 
     const currentParts = [{ text: input }];
-    
-    // Process incoming images from Roblox to the AI
+
+    // Handle Image Injection (Base64)
     if (images && images.length > 0) {
       images.forEach(base64String => {
+        // Strip prefix if present
         const cleanData = base64String.replace(/^data:image\/\w+;base64,/, "");
         currentParts.push({
-            inlineData: { mimeType: "image/png", data: cleanData }
+            inlineData: {
+                mimeType: "image/png",
+                data: cleanData
+            }
         });
       });
     }
 
-    const contents = [...history, { role: 'user', parts: currentParts }];
+    const contents = [
+      ...history,
+      { role: 'user', parts: currentParts }
+    ];
 
     try {
-      const response = await genAI.models.generateContent({ model: modelId, config, contents });
+      const response = await genAI.models.generateContent({
+        model: modelId,
+        config,
+        contents,
+      });
       return response.text();
     } catch (error) {
       console.error("Worker AI Error:", error);
@@ -70,11 +93,62 @@ export const AIEngine = {
   },
 
   /**
-   * 3. IMAGE GENERATOR (Visual Assets)
-   * Uses Imagen 3 to generate textures/UI assets
+   * 3. ONBOARDING ANALYST (Question Generation)
+   * Returns STRICT JSON for the Frontend
+   */
+  generateEntryQuestions: async (description) => {
+    const modelId = 'gemini-2.0-flash';
+    const input = `[MODE 1: QUESTIONS]\nAnalyze this game idea: "${description}".\nGenerate 3 crucial technical questions to clarify scope. Output ONLY raw JSON array.`;
+    
+    try {
+      const response = await genAI.models.generateContent({
+        model: modelId,
+        config: { 
+            responseMimeType: "application/json",
+            systemInstruction: { parts: [{ text: prompts.analyst_system_prompt }] }
+        },
+        contents: [{ role: 'user', parts: [{ text: input }] }]
+      });
+      
+      const text = response.text();
+      return JSON.parse(text);
+    } catch (e) {
+      console.error("Analyst Error:", e);
+      // Fallback to prevent frontend crash
+      return [{ id: "fallback", label: "Please describe the core gameplay loop in detail.", type: "textarea" }];
+    }
+  },
+
+  /**
+   * 4. PROJECT GRADER (Feasibility Check)
+   * Returns STRICT JSON
+   */
+  gradeProject: async (description, answers) => {
+    const modelId = 'gemini-2.0-flash';
+    const input = `[MODE 2: GRADING]\nIdea: "${description}"\nUser Answers: ${JSON.stringify(answers)}\n\nAssess feasibility for an AI Builder. Output ONLY raw JSON.`;
+
+    try {
+      const response = await genAI.models.generateContent({
+        model: modelId,
+        config: { 
+            responseMimeType: "application/json",
+            systemInstruction: { parts: [{ text: prompts.analyst_system_prompt }] }
+        },
+        contents: [{ role: 'user', parts: [{ text: input }] }]
+      });
+      return JSON.parse(response.text());
+    } catch (e) {
+      console.error("Grading Error:", e);
+      return { feasibility: 80, rating: "B", summary: "Standard project structure detected." };
+    }
+  },
+
+  /**
+   * 5. IMAGE GENERATOR (Visual Assets)
+   * Uses Imagen 3
    */
   generateImage: async (prompt) => {
-    const modelId = 'gemini-3-pro-image-preview';
+    const modelId = 'imagen-3.0-generate-001';
     
     try {
       console.log(`[AIEngine] Generating Image for: "${prompt}"...`);
@@ -83,23 +157,18 @@ export const AIEngine = {
         prompt: prompt,
         config: {
           numberOfImages: 1,
-          // aspectRatio: "1:1", // Standard for textures/icons
-          outputMimeType: "image/png" // Roblox supports PNG
+          aspectRatio: "16:9", // Widescreen for thumbnails
+          outputMimeType: "image/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
+      if (!image || !image.image) throw new Error("No image data returned");
+
       return image.image.toString('base64');
     } catch (error) {
       console.error("Image Gen Error:", error);
-      return null; // Fail gracefully so the server doesn't crash
+      return null; 
     }
   }
 };