Update aiEngine.js

aiEngine.js

@@ -35,12 +35,11 @@ const azureOpenAI = new OpenAI({
 });
 
 // --- HELPER: History Transformer ---
-// Converts your App's { role: 'user', parts: [{text: ''}] } format 
-// to formats acceptable by Azure/Bedrock
 function toAzureHistory(history) {
     return history.map(h => ({
         role: h.role === 'model' ? 'assistant' : 'user',
-        content: h.parts[0].text
+        // We only extract text from history to ensure DB remains lightweight
+        content: h.parts[0].text 
     }));
 }
 
@@ -57,7 +56,6 @@ export const AIEngine = {
      * Uses AWS Bedrock -> Claude 3.5 Sonnet (v2/4.6)
      */
     callPM: async (history, input) => {
-        // Prepare History
         const chatHistory = toBedrockHistory(history);
         const sysPrompt = prompts.pm_system_prompt || "You are a Senior Project Manager.";
 
@@ -76,7 +74,6 @@ export const AIEngine = {
             const response = await bedrockClient.send(command);
             const text = response.output.message.content.find(b => b.text)?.text;
             
-            // Extract usage
             const inputTokens = response.usage?.inputTokens || 0;
             const outputTokens = response.usage?.outputTokens || 0;
 
@@ -95,22 +92,44 @@ export const AIEngine = {
     /**
      * 2. WORKER (Coding & Execution)
      * Uses Azure OpenAI -> GPT-5 Mini
+     * UPDATED: Accepts images but does not persist them in history (App.js handles persistence of text only)
      */
     callWorker: async (history, input, images = []) => {
         const chatHistory = toAzureHistory(history);
         const sysPrompt = prompts.worker_system_prompt || "You are a specialized Worker.";
         
-        // Azure GPT-5 Mini Setup
-        // Note: For now, we append images to the user content if needed, 
-        // but simplified here for text-parity with your script.
-        
+        // Construct the Current User Message
+        let userMessageContent;
+
+        if (images && images.length > 0) {
+            // Multimodal Request
+            userMessageContent = [
+                { type: "text", text: input }
+            ];
+
+            images.forEach(img => {
+                // Ensure the base64 string has the data prefix if missing (or pass as is if client handles it)
+                // Assuming client sends "data:image/png;base64,..."
+                userMessageContent.push({
+                    type: "image_url",
+                    image_url: {
+                        url: img, 
+                        detail: "auto"
+                    }
+                });
+            });
+        } else {
+            // Text-only Request
+            userMessageContent = input;
+        }
+
         try {
             const response = await azureOpenAI.chat.completions.create({
                 model: process.env.AZURE_DEPLOYMENT_NAME, // "gpt-5-mini"
                 messages: [
                     { role: "system", content: sysPrompt },
                     ...chatHistory,
-                    { role: "user", content: input }
+                    { role: "user", content: userMessageContent }
                 ],
                 reasoning_effort: "high"
             });
@@ -132,7 +151,6 @@ export const AIEngine = {
 
     /**
      * 3. ONBOARDING ANALYST (Question Generation)
-     * Returns STRICT JSON
      */
     generateEntryQuestions: async (description) => {
         const sysPrompt = prompts.analyst_system_prompt || "Return JSON only.";
@@ -155,7 +173,6 @@ export const AIEngine = {
             };
         } catch (e) {
             console.error("Analyst Error:", e);
-            // Fallback structure
             return { 
                 status: "ACCEPTED", 
                 questions: [{ id: "fallback", label: "Elaborate?", type: "textarea" }], 
@@ -166,7 +183,6 @@ export const AIEngine = {
 
     /**
      * 4. PROJECT GRADER (Feasibility Check)
-     * Returns STRICT JSON
      */
     gradeProject: async (description, answers) => {
         const sysPrompt = prompts.analyst_system_prompt || "Return JSON only.";
@@ -199,12 +215,9 @@ export const AIEngine = {
     },
 
     /**
-     * 5. IMAGE GENERATOR 
-     * Kept as placeholder since you said "not being used" for now,
-     * but prevents crashes if App.js calls it.
+     * 5. IMAGE GENERATOR (Placeholder)
      */
     generateImage: async (prompt) => {
-        console.log("Image generation currently skipped in Azure/Bedrock migration.");
         return null;
     }
 };