Update image_processor.py

image_processor.py

@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
 
 def process_zip_and_generate_images(
     zip_path, category, size, quality, user_prompt,
-    sentiment, platform, num_images, demo_mode, existing_images
+    sentiment, platform, num_images, demo_mode, existing_images, blur
 ):
     num_images = 1 if demo_mode else num_images
 
@@ -26,7 +26,7 @@ def process_zip_and_generate_images(
         else:
             image_files = [(os.path.basename(zip_path.name), zip_path.name)]
 
-        results = process_image_files(image_files, category, size, quality, user_prompt, sentiment, platform, num_images)
+        results = process_image_files(image_files, category, size, quality, user_prompt, sentiment, platform, num_images, blur)
 
         all_urls = [url for entry in results for url in entry["urls"]]
 
@@ -69,7 +69,7 @@ def get_valid_image_files(temp_dir):
 
 
 
-def process_image_files(image_files, category, size, quality, user_prompt, sentiment, platform, num_images):
+def process_image_files(image_files, category, size, quality, user_prompt, sentiment, platform, num_images, blur):
     final_results = []
     logs = []
 
@@ -92,6 +92,7 @@ def process_image_files(image_files, category, size, quality, user_prompt, senti
                     sentiment,
                     platform,
                     num_images,
+                    blur,
                     log_id,
                     logs
                 )
@@ -108,7 +109,7 @@ def process_image_files(image_files, category, size, quality, user_prompt, senti
     return final_results
 
 
-def process_single_image(file_name, file_path, category, size, quality, user_prompt, sentiment, platform, num_images, log_id, logs):
+def process_single_image(file_name, file_path, category, size, quality, user_prompt, sentiment, platform, num_images, blur, log_id, logs):
     try:
         # base64_img = encode_image_to_base64(file_path)
         # logs.append(f"Prompts generated for {file_name}")
@@ -116,7 +117,7 @@ def process_single_image(file_name, file_path, category, size, quality, user_pro
         # prompts = get_prompts(base64_img, category, user_prompt, sentiment, negative_prompt)
         # logs.append(f"Generating images for {file_name}")
 
-        image_urls = generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, platform, num_images)
+        image_urls = generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, platform, num_images, blur)
 
         status = "completed" if image_urls else "failed"
         message = "Completed successfully" if image_urls else "No images generated"
@@ -138,12 +139,12 @@ 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):
+def generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, platform, num_images, blur):
     image_urls = []
 
     for _ in range(num_images):
         try:
-            img_url = generate_image(file_path, size, quality, category, sentiment, user_prompt, platform)
+            img_url = generate_image(file_path, size, quality, category, sentiment, user_prompt, platform, blur)
             
             s3_url = upload_image_to_r2(img_url)
             logger.info(f"Generated and uploaded image to: {s3_url}")