Update image_processor.py

image_processor.py

@@ -10,12 +10,12 @@ from mongo_logger import create_log, update_log_status
 from helpers import encode_image_to_base64, is_valid_image
 
 
-def process_zip_and_generate_images(zip_path, category, size, quality, user_prompt, sentiment, negative_prompt):
+def process_zip_and_generate_images(zip_path, category, size, quality, user_prompt, sentiment, platform):
     try:
         temp_dir = extract_zip_file(zip_path)
 
         image_files = get_valid_image_files(temp_dir)
-        results = process_image_files(image_files, category, size, quality, user_prompt, sentiment, negative_prompt)
+        results = process_image_files(image_files, category, size, quality, user_prompt, sentiment, platform)
         all_urls = [url for entry in results for url in entry["urls"]]
 
         new_base = "https://adgen.statics.lookfinity.net/"
@@ -50,7 +50,7 @@ def get_valid_image_files(temp_dir):
     return valid_files
 
 
-def process_image_files(image_files, category, size, quality, user_prompt, sentiment, negative_prompt):
+def process_image_files(image_files, category, size, quality, user_prompt, sentiment, platform):
     final_results = []
     logs = []
 
@@ -71,7 +71,7 @@ def process_image_files(image_files, category, size, quality, user_prompt, senti
                     quality,
                     user_prompt,
                     sentiment,
-                    negative_prompt,
+                    platform,
                     log_id,
                     logs
                 )
@@ -85,7 +85,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, negative_prompt, log_id, logs):
+def process_single_image(file_name, file_path, category, size, quality, user_prompt, sentiment, platform, log_id, logs):
     try:
         # base64_img = encode_image_to_base64(file_path)
         # logs.append(f"Prompts generated for {file_name}")
@@ -93,7 +93,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, negative_prompt)
+        image_urls = generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, platform)
 
         status = "completed" if image_urls else "failed"
         message = "Completed successfully" if image_urls else "No images generated"
@@ -112,12 +112,12 @@ def process_single_image(file_name, file_path, category, size, quality, user_pro
         return None
 
 
-def generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, negative_prompt):
+def generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, platform):
     image_urls = []
 
     for _ in range(10):
         try:
-            img_url = generate_image(file_path, size, quality, category, sentiment, user_prompt, negative_prompt)
+            img_url = generate_image(file_path, size, quality, category, sentiment, user_prompt, platform)
 
             s3_url = upload_image_to_r2(img_url)
             image_urls.append(s3_url)