Update pages/api/generate.js

pages/api/generate.js

@@ -10,7 +10,7 @@ export default async function handler(req, res) {
   const { prompt, drawingData, customApiKey } = req.body;
 
   // Log request details (truncating drawingData for brevity)
-  console.log("API Request:", {
+  console.log("Fashion Design Conversion Request:", {
     prompt,
     hasDrawingData: !!drawingData,
     drawingDataLength: drawingData ? drawingData.length : 0,
@@ -18,8 +18,8 @@ export default async function handler(req, res) {
     hasCustomApiKey: !!customApiKey,
   });
 
-  if (!prompt && !drawingData) {
-    return res.status(400).json({ error: "Either prompt or drawingData is required" });
+  if (!drawingData) {
+    return res.status(400).json({ error: "Fashion design drawing is required" });
   }
 
   // Use custom API key if provided, otherwise use the one from environment variables
@@ -34,7 +34,7 @@ export default async function handler(req, res) {
 
   const genAI = new GoogleGenerativeAI(apiKey);
 
-  // Set responseModalities to include "Image" so the model can generate an image
+  // Set responseModalities to include "Image" for image generation
   const model = genAI.getGenerativeModel({
     model: "gemini-2.0-flash-exp-image-generation",
     generationConfig: {
@@ -43,37 +43,30 @@ export default async function handler(req, res) {
   });
 
   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",
-        },
-      };
-
-      // Combine drawing with text prompt
-      generationContent = [
-        imagePart,
-        {
-          text: `${
-            prompt ? prompt + ". " : ""
-          }Use the provided hand-drawing as the starting line-art, build it up to a stunning photo. Remeber to keep the original line of 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 || "Convert a basic line into a stunning artwork.";
-      console.log("Using text-only prompt");
-    }
+    // Create a content part with the base64-encoded fashion design
+    const imagePart = {
+      inlineData: {
+        data: drawingData,
+        mimeType: "image/png",
+      },
+    };
+
+    // Fashion-specific instructions to preserve design integrity
+    const fashionPrompt = `Convert this fashion design sketch into a high-quality product image with the following requirements:
+    1. Maintain EXACT proportions, silhouette, and design details from the original sketch
+    2. Use realistic fabrics and textures appropriate for the design
+    3. Add professional lighting and shadows to enhance dimensionality
+    4. Keep the background neutral (white or light gray) unless specified
+    ${prompt ? "Additional instructions: " + prompt : ""}`;
+
+    const generationContent = [
+      imagePart,
+      { text: fashionPrompt },
+    ];
 
-    console.log("Calling Gemini API...");
+    console.log("Processing fashion design conversion...");
     const response = await model.generateContent(generationContent);
-    console.log("Gemini API response received");
+    console.log("Fashion conversion completed");
 
     // Initialize response data
     const result = {
@@ -84,26 +77,29 @@ export default async function handler(req, res) {
 
     // 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
       if (part.text) {
         result.message = part.text;
         console.log("Received text response:", part.text);
       } else if (part.inlineData) {
         const imageData = part.inlineData.data;
-        console.log("Received image data, length:", imageData.length);
-
-        // Include the base64 data in the response
+        console.log("Received product image data, length:", imageData.length);
         result.imageData = imageData;
       }
     }
 
-    console.log("Sending successful response");
+    // Verify the output maintains design integrity
+    if (!result.imageData) {
+      throw new Error("The generated image did not preserve the original design adequately");
+    }
+
+    console.log("Sending successful fashion conversion response");
     return res.status(200).json(result);
   } catch (error) {
-    console.error("Error generating content:", error);
+    console.error("Error in fashion design conversion:", error);
     return res.status(500).json({
       success: false,
-      error: error.message || "Failed to generate image",
+      error: error.message || "Failed to convert fashion design to product image",
+      suggestion: "Please ensure your design sketch has clear lines and distinct elements",
     });
   }
 }