Update generate.py

generate.py

@@ -1,13 +1,14 @@
 # generate.py
-# --- VERSION 6 ---
+# --- VERSION 7 ---
 
-print("--- RUNNING GENERATE.PY VERSION 6 ---")
+print("--- RUNNING GENERATE.PY VERSION 7 (WITH TRACEBACK) ---")
 
 import torch
 import cv2
 import os
 import logging
 import uuid
+import traceback # <-- ADDED FOR DETAILED ERROR LOGGING
 from diffusers import StableDiffusionPipeline, DDIMScheduler, AutoencoderKL
 from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
 from transformers import CLIPVisionModelWithProjection
@@ -16,7 +17,7 @@ from insightface.utils import face_align
 from huggingface_hub import hf_hub_download
 from storage3.utils import StorageException
 import config
-from database import supabase # Import the initialized supabase client
+from database import supabase
 
 # --- Setup Logging ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -60,9 +61,6 @@ class GenerationService:
             raise RuntimeError(f"Could not initialize GenerationService: {e}") from e
 
     def generate_magic_image(self, face_images: list, gender: str, prompt: str, plan: str = 'free') -> str | None:
-        """
-        Generates an image, uploads it to cloud storage, and returns the public URL.
-        """
         logger.info("Starting image generation process...")
         
         full_prompt = f"{prompt}, 4k, high-resolution, photorealistic, masterpiece, single person, solo portrait, centered composition"
@@ -92,8 +90,6 @@ class GenerationService:
         try:
             final_embedding = average_embedding.unsqueeze(0)
 
-            # --- 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,
@@ -104,20 +100,16 @@ class GenerationService:
                 height=768,
             )
             
-            # 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"
             os.makedirs(temp_dir, exist_ok=True)
             local_path = os.path.join(temp_dir, f"{uuid.uuid4()}.png")
             image.save(local_path)
             
-            # --- Upload to Supabase Storage ---
             storage_path = f"public/{os.path.basename(local_path)}"
             logger.info(f"Uploading {local_path} to Supabase bucket '{config.SUPABASE_BUCKET_NAME}' at path '{storage_path}'")
             
@@ -131,7 +123,6 @@ class GenerationService:
             public_url = supabase.storage.from_(config.SUPABASE_BUCKET_NAME).get_public_url(storage_path)
             logger.info(f"Upload successful. Public URL: {public_url}")
 
-            # --- Clean up local file ---
             os.remove(local_path)
             
             return public_url
@@ -140,7 +131,13 @@ class GenerationService:
             logger.error(f"Supabase Storage Error: {e}")
             return None
         except Exception as e:
-            logger.error(f"An error occurred during image generation or upload: {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)
             return None
@@ -158,6 +155,6 @@ if __name__ == '__main__':
         if result_url:
             print(f"\n✅ Test successful! Image URL: {result_url}")
         else:
-            print("\n❌ Test failed. Check logs for details.")
+            print(f"\n❌ Test failed. Please check the new traceback logs for details.")
     else:
         print("To run a test, place an image named 'test_face.jpg' in the root directory.")