Update src/image_functions.py

src/image_functions.py

@@ -73,13 +73,24 @@ def _encode_image_to_base64(image_path):
         return ""
 
 
-def generate_image(file_path, size, quality, category, sentiment, user_prompt, platform, blur):
+def generate_image(file_path, size, quality, category, sentiment, user_prompt, platform, blur, i=None):
     try:
         api_key = os.getenv("OPENAI_API_KEY")
         if not api_key:
             logger.critical("OPENAI_API_KEY is not set.")
             raise RuntimeError("OPENAI_API_KEY is missing")
         client = OpenAI(api_key=api_key)
+
+        if platform == "Google Display Network":
+            size_messages = {
+                0: "The aspect ratio of the image should be 1024x1024",
+                1: "The aspect ratio of the image should be 1536x1024.",
+                2: "The aspect ratio of the image should be 1024x1536.",
+            }
+            
+            img_size = size_messages.get(i, "")
+        else:
+            img_size = ""
         
         with open(file_path, "rb") as img_file:
             if blur:
@@ -91,9 +102,9 @@ def generate_image(file_path, size, quality, category, sentiment, user_prompt, p
                 prompt=(
                     f"You are a top-tier performance digital marketer and creative strategist with 15+ years of expertise in affiliate marketing.\n"
                     f"Your objective is to analyze the provided winning ad image, deconstruct its concept, visual composition, and color scheme, and generate a fresh, conversion-focused ad visual tailored for the {category} niche.\n"
-                    f"The new design should convey a {sentiment} sentiment and incorporate the user instruction: {user_prompt}.\n"
+                    f"The new design should convey a {sentiment} sentiment and incorporate the user instruction: \n {user_prompt}.\n If user has given multple choices or options to be include in the image so choose randomly relevant to the reference image."
                     f"Create a visually compelling ad optimized for {platform} Ads that is scroll-stopping, pattern-interrupting, and designed to drive high CTR and Conversion Rate. Utilize striking color combinations, dynamic contrast levels, and strategic layout compositions to command attention while aligning with the target audience avatar.\n"
-                    f"Make sure the images should be realistic, not be stocky at all and raw which should look like they are shot from an iPhone with {background}."
+                    f"Make sure the images should be realistic, not be stocky at all and raw which should look like they are shot from an iPhone with {background}.{img_size}"
                 ),
                 image=img_file,
                 size=size,