Update generator_function/multimodel_image_processor.py

generator_function/multimodel_image_processor.py

@@ -165,7 +165,7 @@ def generate_images_from_prompts_multimodel(
                 # Use multimodel approach
                 image_bytes = generate_image_multimodel(
                     file_path, selected_model, category, sentiment, user_prompt, 
-                    platform, size, model_params, i
+                    platform, size, model_params, i, num_images
                 )
             
             if not image_bytes: return None
@@ -185,7 +185,7 @@ def generate_images_from_prompts_multimodel(
 
 def generate_image_multimodel(file_path: str, model_name: str, category: str, sentiment: str, 
                             user_prompt: str, platform: str, size: str, model_params: Optional[dict], 
-                            variation_index: int) -> Optional[bytes]:
+                            variation_index: int, num_images: int) -> Optional[bytes]:
     """Generate image using multimodel approach"""
     try:
         # Convert image to base64
@@ -195,7 +195,7 @@ def generate_image_multimodel(file_path: str, model_name: str, category: str, se
         
         # Use existing prompt service to get prompt variations
         prompt_variations = get_prompts(
-            base64_image, category, user_prompt, sentiment, None
+            base64_image, category, user_prompt, sentiment, None, num_images
         )
         
         # Select a prompt based on variation index