update2

image_processor.py

@@ -20,7 +20,7 @@ def process_zip_and_generate_images(
 ):
     num_images = 1 if demo_mode else num_images
 
-    num_images = 4 if not demo_mode and platform == "Google Display Network" else num_images
+    num_images = 3 if not demo_mode and platform == "Google Display Network" else num_images
 
     try:
         if zip_path.endswith('.zip'):
@@ -145,21 +145,20 @@ def process_single_image(file_name, file_path, category, size, quality, user_pro
 def generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, platform, num_images, blur):
     image_urls = []
 
-    def worker(_):
+    def worker(i):
         try:
-            image_bytes = generate_image(file_path, size, quality, category, sentiment, user_prompt, platform, blur)
+            image_bytes = generate_image(file_path, size, quality, category, sentiment, user_prompt, platform, blur, i)
 
             if not image_bytes:
-                log_error("Image generation returned empty bytes.")
+                logger.error("Image generation returned empty bytes.")
                 return None
 
             logger.info("Successfully generated image bytes.")
 
-            logger.info("Injecting EXIF metadata into image.")
             try:
                 image_with_metadata = meta_data_helper_function(image_bytes)
             except Exception as e:
-                log_error(f"Metadata generation failed: {e}")
+                logger.error(f"Metadata generation failed: {e}")
                 return None
 
             s3_url = upload_image_to_r2(image_with_metadata)
@@ -167,7 +166,7 @@ def generate_images_from_prompts(file_path, size, quality, category, sentiment,
             return s3_url
 
         except Exception as e:
-            log_error(f"Image generation failed: {e}")
+            logger.error(f"Image generation failed: {e}")
             return None
 
     with ThreadPoolExecutor(max_workers=min(10, num_images)) as executor: