Update image_processor.py

image_processor.py

@@ -87,13 +87,13 @@ def process_image_files(image_files, category, size, quality, user_prompt, senti
 
 def process_single_image(file_name, file_path, category, size, quality, user_prompt, sentiment, negative_prompt, log_id, logs):
     try:
-        base64_img = encode_image_to_base64(file_path)
-        logs.append(f"Prompts generated for {file_name}")
+        # base64_img = encode_image_to_base64(file_path)
+        # logs.append(f"Prompts generated for {file_name}")
+        #
+        # prompts = get_prompts(base64_img, category, user_prompt, sentiment, negative_prompt)
+        # logs.append(f"Generating images for {file_name}")
 
-        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(prompts, size, quality)
+        image_urls = generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, negative_prompt)
 
         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(prompts, size, quality):
+def generate_images_from_prompts(file_path, size, quality, category, sentiment, user_prompt, negative_prompt):
     image_urls = []
 
-    for prompt in prompts:
+    for _ in range(10):
         try:
-            img_url = generate_image(prompt, size, quality)
+            img_url = generate_image(file_path, size, quality, category, sentiment, user_prompt, negative_prompt)
 
             s3_url = upload_image_to_r2(img_url)
             image_urls.append(s3_url)