Update generate.py

generate.py

@@ -1,7 +1,7 @@
 # generate.py
-# --- VERSION 5 ---
+# --- VERSION 6 ---
 
-print("--- RUNNING GENERATE.PY VERSION 5 ---")
+print("--- RUNNING GENERATE.PY VERSION 6 ---")
 
 import torch
 import cv2
@@ -9,6 +9,7 @@ import os
 import logging
 import uuid
 from diffusers import StableDiffusionPipeline, DDIMScheduler, AutoencoderKL
+from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
 from transformers import CLIPVisionModelWithProjection
 from insightface.app import FaceAnalysis
 from insightface.utils import face_align
@@ -22,10 +23,6 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(level
 logger = logging.getLogger(__name__)
 
 
-# --- IP-Adapter FaceID Model (Placeholder) ---
-# The complex IP-Adapter logic is assumed to be part of the diffusers pipeline for this implementation.
-
-
 # --- Main Generation Service ---
 class GenerationService:
     def __init__(self):
@@ -95,8 +92,9 @@ class GenerationService:
         try:
             final_embedding = average_embedding.unsqueeze(0)
 
-            # CORRECTED CODE: The pipeline returns an object, not a tuple.
-            pipeline_output = self.pipe(
+            # --- ROBUST FIX for diffusers version mismatch ---
+            # This handles both old and new output formats from the pipeline.
+            output = self.pipe(
                 prompt=full_prompt,
                 negative_prompt=negative_prompt,
                 ip_adapter_image_embeds=[final_embedding],
@@ -105,7 +103,13 @@ class GenerationService:
                 width=512,
                 height=768,
             )
-            image = pipeline_output.images[0]
+            
+            # Check if the output is the modern object or an old tuple
+            if isinstance(output, StableDiffusionPipelineOutput):
+                image = output.images[0]
+            else:
+                # Handle older tuple-based output
+                image = output[0][0]
 
             # --- Save image locally first ---
             temp_dir = "temp_images"