fix: image editor

src/server/plugins/ai/ai_image_editor.js

@@ -4,6 +4,13 @@ import multer from "multer";
 import axios from "axios";
 import path from "path"; 
 
+const ModelMap = {
+  nano_banana: 2,
+  seed_dream: 5,
+  flux: 8,
+  qwen_image: 9
+};
+
 const storage = multer.memoryStorage();
 const upload = multer({
   storage: storage,
@@ -20,13 +27,6 @@ const upload = multer({
   }
 });
 
-const ModelMap = {
-  nano_banana: 2,
-  seed_dream: 5,
-  flux: 8,
-  qwen_image: 9
-};
-
 /** @type {import("../../types/plugin.ts").ApiPluginHandler}*/
 const handler = {
   name: "AI Image Editor",
@@ -36,9 +36,6 @@ const handler = {
   category: ["ai"],
   alias: ["aiImageEditor"],
   tags: ["ai", "image"],
-  disabled: true,
-  disabledReason: "Temporarily disabled",
-  
   parameters: {
     body: [
       {
@@ -137,14 +134,8 @@ const handler = {
     }
   },
   exec: async (req, res) => {
-    console.log("🔵 [1] Request received at AI Image Editor");
-    console.log("🔵 [1] Headers:", req.headers);
-    
     upload.single('image')(req, res, async (err) => {
-      console.log("🔵 [2] Inside multer middleware");
-      
       if (err) {
-        console.error("❌ [2] Multer error:", err);
         if (err instanceof multer.MulterError) {
           if (err.code === 'LIMIT_FILE_SIZE') {
             return res.status(413).json({
@@ -159,65 +150,43 @@ const handler = {
         });
       }
       
-      console.log("🔵 [3] Multer successful");
-      console.log("🔵 [3] Body:", req.body);
-      console.log("🔵 [3] File:", req.file ? `${req.file.originalname} (${req.file.size} bytes)` : "No file");
-      
       try {
-        const { prompt, output_size, disable_safety_checker, model } = req.body;
+        const { prompt, output_size, disable_safety_checker, model } = req.body;       
         
-        console.log("🔵 [4] Extracted params:", { prompt, output_size, model, disable_safety_checker });
-        
-        // Validate required parameters
         if (!prompt) {
-          console.log("❌ [4] Missing prompt");
           return ErrorResponses.missingParameter(res, "prompt");
         }
         if (!output_size) {
-          console.log("❌ [4] Missing output_size");
           return ErrorResponses.missingParameter(res, "output_size");
         }
         if (!model) {
-          console.log("❌ [4] Missing model");
           return ErrorResponses.missingParameter(res, "model");
         }
         
-        // Validate model
         if (!ModelMap[model]) {
-          console.log("❌ [4] Invalid model:", model);
           return res.status(400).json({
             success: false,
             message: `Invalid model. Must be one of: ${Object.keys(ModelMap).join(', ')}`
           });
         }
         
-        console.log("🔵 [5] Creating AIEnhancer instance");
         const ai = new AIEnhancer();
-        console.log("✅ [5] AIEnhancer created");
-        
         let imageBuffer;
         let filename;
-        
-        // Handle uploaded file
+
         if (req.file) {
-          console.log("🔵 [6] Using uploaded file");
           imageBuffer = req.file.buffer;
           filename = req.file.originalname;
         } 
-        // Handle URL
         else if (req.body.image && typeof req.body.image === 'string') {
-          console.log("🔵 [6] Downloading from URL:", req.body.image);
           const imageUrl = req.body.image;
           
-          // Validate URL
           try {
             new URL(imageUrl);
           } catch {
-            console.log("❌ [6] Invalid URL");
             return ErrorResponses.invalidUrl(res, "Invalid image URL");
           }
-          
-          // Download image from URL
+
           const response = await axios.get(imageUrl, {
             responseType: 'arraybuffer',
             maxContentLength: 5 * 1024 * 1024,
@@ -229,21 +198,11 @@ const handler = {
 
           imageBuffer = Buffer.from(response.data);
           filename = path.basename(new URL(imageUrl).pathname) || 'image';
-          console.log("✅ [6] Downloaded from URL:", filename);
         } else {
-          console.log("❌ [6] No image provided");
           return ErrorResponses.missingParameter(res, "image (file or URL)");
         }
         
-        console.log("🔵 [7] Image buffer size:", imageBuffer.length);
-        
-        // Get model ID from map
-        const modelId = ModelMap[model];
-        console.log("🔵 [8] Model ID:", modelId);
-
-        // Call AI Image Editor
-        console.log("🔵 [9] Calling ImageAIEditor...");
-        const edit = await ai.ImageAIEditor(imageBuffer, modelId, {
+        const edit = await ai.ImageAIEditor(imageBuffer, model, {
           size: output_size,
           aspect_ratio: "match_input_image",
           go_fast: true,
@@ -251,13 +210,10 @@ const handler = {
           output_quality: 100,
           disable_safety_checker: disable_safety_checker === 'true' || disable_safety_checker === true,
         });
-        
-        console.log("✅ [9] ImageAIEditor response:", edit);
   
-        return sendSuccess(res, edit.results || edit);
+        return sendSuccess(res, edit.results);
       } catch (e) {
-        console.error("❌ [ERROR] Exception caught:", e);
-        console.error("❌ [ERROR] Stack trace:", e.stack);
+        console.error("Exception caught:", e);
         return ErrorResponses.serverError(res, "An error occurred, try again later.");
       }
     });

src/server/plugins/ai/ai_image_utils.js

@@ -1,3 +1,8 @@
+import axios from "axios";
+import { fileTypeFromBuffer } from "file-type";
+import { readFileSync } from "fs";
+import CryptoJS from "crypto-js";
+
 const ModelMap = {
   nano_banana: 2,
   seed_dream: 5,