Update generate.py

generate.py

@@ -1,14 +1,14 @@
 # generate.py
-# --- VERSION 7 ---
+# --- VERSION 8 (FINAL) ---
 
-print("--- RUNNING GENERATE.PY VERSION 7 (WITH TRACEBACK) ---")
+print("--- RUNNING GENERATE.PY VERSION 8 (FINAL) ---")
 
 import torch
 import cv2
 import os
 import logging
 import uuid
-import traceback # <-- ADDED FOR DETAILED ERROR LOGGING
+import traceback
 from diffusers import StableDiffusionPipeline, DDIMScheduler, AutoencoderKL
 from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
 from transformers import CLIPVisionModelWithProjection
@@ -88,7 +88,14 @@ class GenerationService:
 
         logger.info("Calling the generation pipeline...")
         try:
-            final_embedding = average_embedding.unsqueeze(0)
+            # --- FINAL FIX: Create negative embeddings ---
+            # The IP-Adapter expects both positive and negative image embeddings.
+            # We create a zeroed-out tensor for the negative part.
+            positive_embedding = average_embedding.unsqueeze(0)
+            negative_embedding = torch.zeros_like(positive_embedding)
+            
+            # Combine them for the pipeline
+            final_embedding = torch.cat([negative_embedding, positive_embedding], dim=0)
 
             output = self.pipe(
                 prompt=full_prompt,
@@ -127,16 +134,9 @@ class GenerationService:
             
             return public_url
 
-        except StorageException as e:
-            logger.error(f"Supabase Storage Error: {e}")
-            return None
         except Exception as e:
-            # --- THIS IS THE IMPORTANT PART ---
-            # It will print the full error details to the log.
             logger.error("An unexpected error occurred. Full traceback below:")
             traceback.print_exc()
-            # ---
-            
             logger.error(f"Error summary: {e}")
             if 'local_path' in locals() and os.path.exists(local_path):
                 os.remove(local_path)