Update pages/api/generate.js

pages/api/generate.js

@@ -2,33 +2,33 @@ import { GoogleGenerativeAI } from "@google/generative-ai";
 
 export default async function handler(req, res) {
   // Only allow POST requests
-  if (req.method !== 'POST') {
-    return res.status(405).json({ error: 'Method not allowed' });
+  if (req.method !== "POST") {
+    return res.status(405).json({ error: "Method not allowed" });
   }
 
   // Get prompt, drawing, and custom API key from request body
   const { prompt, drawingData, customApiKey } = req.body;
-  
+
   // Log request details (truncating drawingData for brevity)
   console.log("API Request:", {
     prompt,
     hasDrawingData: !!drawingData,
     drawingDataLength: drawingData ? drawingData.length : 0,
     drawingDataSample: drawingData ? `${drawingData.substring(0, 50)}... (truncated)` : null,
-    hasCustomApiKey: !!customApiKey
+    hasCustomApiKey: !!customApiKey,
   });
-  
-  if (!prompt) {
-    return res.status(400).json({ error: 'Prompt is required' });
+
+  if (!prompt && !drawingData) {
+    return res.status(400).json({ error: "Either prompt or drawingData is required" });
   }
 
   // Use custom API key if provided, otherwise use the one from environment variables
   const apiKey = customApiKey || process.env.GEMINI_API_KEY;
-  
+
   if (!apiKey) {
-    return res.status(400).json({ 
-      success: false, 
-      error: 'No API key available. Please provide a valid Gemini API key.' 
+    return res.status(400).json({
+      success: false,
+      error: "No API key available. Please provide a valid Gemini API key.",
     });
   }
 
@@ -38,46 +38,50 @@ export default async function handler(req, res) {
   const model = genAI.getGenerativeModel({
     model: "gemini-2.0-flash-exp-image-generation",
     generationConfig: {
-      responseModalities: ['Text', 'Image']
+      responseModalities: ["Text", "Image"],
     },
   });
 
   try {
     let generationContent;
-    
+
     // If drawingData is provided, include it as an image in the request
     if (drawingData) {
       // Create a content part with the base64-encoded image
       const imagePart = {
         inlineData: {
           data: drawingData,
-          mimeType: "image/png"
-        }
+          mimeType: "image/png",
+        },
       };
-      
+
       // Combine drawing with text prompt
       generationContent = [
         imagePart,
-         { text: ${prompt}. From the provided 2D plan layout as the starting point. Convert it into a detailed 3D sketch, including walls, doors, windows, and basic furniture placement. || "Enhance the design with a realistic perspective and improve it to create a stunning 3D representation. Maintain the original style of the layout while adding depth and visual appeal" }
+        {
+          text: `${
+            prompt ? prompt + ". " : ""
+          }From the provided 2D plan layout as the starting point, convert it into a detailed 3D sketch, including walls, doors, windows, and basic furniture placement. Enhance the design with a realistic perspective and improve it to create a stunning 3D representation. Maintain the original style of the layout while adding depth and visual appeal.`,
+        },
       ];
       console.log("Using multipart content with drawing data and prompt");
     } else {
       // Use text-only prompt if no drawing is provided
-      generationContent = prompt;
+      generationContent = prompt || "Convert a basic 2D plan layout into a detailed 3D sketch, including walls, doors, windows, and basic furniture placement. Enhance the design with a realistic perspective and improve it to create a stunning 3D representation.";
       console.log("Using text-only prompt");
     }
-    
+
     console.log("Calling Gemini API...");
     const response = await model.generateContent(generationContent);
     console.log("Gemini API response received");
-    
+
     // Initialize response data
     const result = {
       success: true,
-      message: '',
-      imageData: null
+      message: "",
+      imageData: null,
     };
-    
+
     // Process response parts
     for (const part of response.response.candidates[0].content.parts) {
       // Based on the part type, either get the text or image data
@@ -87,19 +91,19 @@ export default async function handler(req, res) {
       } else if (part.inlineData) {
         const imageData = part.inlineData.data;
         console.log("Received image data, length:", imageData.length);
-        
+
         // Include the base64 data in the response
         result.imageData = imageData;
       }
     }
-    
+
     console.log("Sending successful response");
     return res.status(200).json(result);
   } catch (error) {
     console.error("Error generating content:", error);
-    return res.status(500).json({ 
-      success: false, 
-      error: error.message || 'Failed to generate image' 
+    return res.status(500).json({
+      success: false,
+      error: error.message || "Failed to generate image",
     });
   }
-}
+}