Update generator_function/image_function.py

generator_function/image_function.py

@@ -330,42 +330,39 @@ def handle_image_generation_optimized(
 
 def generate_image(file_path, size, quality, category, sentiment, user_prompt, platform, blur, i=None):
     try:
-        if model == "GPT"
-            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:
-                background = "blurred background." if blur else " not blurred background."
-                result = client.images.edit(
-                    model="gpt-image-1",
-                    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: \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}.{img_size}"
-                    ),
-                    image=img_file,
-                    size=size,
-                    quality=quality,
-                )
-            image_base64 = result.data[0].b64_json
-            image_bytes = base64.b64decode(image_base64)
-            logger.info(f"Successfully generated image for {file_path}")
-            return image_bytes
-    else: 
-        
+        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:
+            background = "blurred background." if blur else " not blurred background."
+            result = client.images.edit(
+                model="gpt-image-1",
+                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: \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}.{img_size}"
+                ),
+                image=img_file,
+                size=size,
+                quality=quality,
+            )
+        image_base64 = result.data[0].b64_json
+        image_bytes = base64.b64decode(image_base64)
+        logger.info(f"Successfully generated image for {file_path}")
+        return image_bytes        
     except Exception as e:
         logger.exception(f"Failed to generate image for {file_path}: {e}")
         raise