main_config changes - WIP

lib/gpt_providers/image_generation/gen_dali2_images.py

@@ -0,0 +1,56 @@
+from openai import OpenAI
+from loguru import logger
+import sys
+
+from .save_image import save_generated_image
+
+from tenacity import (
+    retry,
+    stop_after_attempt,
+    wait_random_exponential,
+)  # for exponential backoff
+
+
+@retry(wait=wait_random_exponential(min=1, max=120), stop=stop_after_attempt(6))
+def generate_dalle3_images(img_prompt, image_dir, size="1024x1024", quality="hd", n=1):
+    """
+    Generates images using the DALL-E 3 model based on a given text prompt.
+
+    Args:
+        img_prompt (str): Text prompt to generate the image.
+        image_dir (str): Directory where the generated image will be saved.
+        size (str, optional): Size of the generated images. Defaults to "1024x1024".
+        quality (str, optional): Quality of the generated images. Defaults to "hd".
+        n (int, optional): Number of images to generate. Defaults to 1.
+
+    Returns:
+        str: Path to the saved image.
+
+    Raises:
+        SystemExit: If an error occurs in image generation or saving.
+    """
+    try:
+        logger.info("Generating Dall-e-3 image for the blog.")
+        client = OpenAI()
+
+        img_generation_response = client.images.generate(
+            model="dall-e-3",
+            prompt=img_prompt,
+            size=size,
+            quality=quality,
+            n=n
+        )
+        # Save the generated image locally.
+        try:
+            img_path = save_generated_image(img_generation_response, image_dir)
+            return img_path
+        except Exception as err:
+            logger.error(f"Failed to Save generated image: {err}")
+
+    except openai.OpenAIError as e:
+        logger.error(f"Dalle-3 image generation error: HTTP Status {e.http_status}, Error: {e.error}")
+        sys.exit("Exiting due to Dalle-3 image generation error.")
+
+    except Exception as e:
+        logger.error(f"Failed to generate images with Dalle3: {e}")
+        sys.exit("Exiting due to a general error in image generation.")

lib/gpt_providers/image_generation/gen_dali3_images.py

@@ -0,0 +1,53 @@
+from openai import OpenAI
+from loguru import logger
+import sys
+
+from tenacity import (
+    retry,
+    stop_after_attempt,
+    wait_random_exponential,
+)  # for exponential backoff
+
+from .save_image import save_generated_image
+
+
+@retry(wait=wait_random_exponential(min=1, max=120), stop=stop_after_attempt(6))
+def generate_dalle3_images(img_prompt, image_dir, size="1024x1024", quality="hd", n=1):
+    """
+    Generates images using the DALL-E 3 model based on a given text prompt.
+
+    Args:
+        img_prompt (str): Text prompt to generate the image.
+        image_dir (str): Directory where the generated image will be saved.
+        size (str, optional): Size of the generated images. Defaults to "1024x1024".
+        quality (str, optional): Quality of the generated images. Defaults to "hd".
+        n (int, optional): Number of images to generate. Defaults to 1.
+
+    Returns:
+        str: Path to the saved image.
+
+    Raises:
+        SystemExit: If an error occurs in image generation or saving.
+    """
+    try:
+        logger.info("Generating Dall-e-3 image for the blog.")
+        client = OpenAI()
+
+        img_generation_response = client.images.generate(
+            model="dall-e-3",
+            prompt=img_prompt,
+            size=size,
+            quality=quality,
+            n=n
+        )
+
+        img_path = save_generated_image(img_generation_response, image_dir)
+        return img_path
+
+    except openai.OpenAIError as e:
+        logger.error(f"Dalle-3 image generation error: HTTP Status {e.http_status}, Error: {e.error}")
+        sys.exit("Exiting due to Dalle-3 image generation error.")
+
+    except Exception as e:
+        logger.error(f"Failed to generate images with Dalle3: {e}")
+        sys.exit("Exiting due to a general error in image generation.")

lib/gpt_providers/image_generation/gen_variation_img.py

@@ -0,0 +1,51 @@
+from loguru import logger
+import sys
+from PIL import Image
+from openai import OpenAI
+
+def gen_new_from_given_img(img_path, image_dir, num_img=1, img_size="1024x1024", response_format="url"):
+    """
+    Generates variations of a given image using OpenAI's image variation API.
+
+    This function takes an existing image, processes it, and generates a specified number of new images based on it. 
+    These generated images are variations of the original, providing creative flexibility.
+
+    Args:
+        img_path (str): Path to the original image file.
+        image_dir (str): Directory where the generated images will be saved.
+        num_img (int, optional): Number of image variations to generate. Defaults to 1.
+        img_size (str, optional): Size of the generated images. Defaults to "1024x1024".
+        response_format (str, optional): Format in which the generated images are returned. Defaults to "url".
+
+    Returns:
+        str: Path to the saved image variation.
+
+    Raises:
+        SystemExit: If a critical error occurs that prevents successful execution.
+    """
+    try:
+        logger.info(f"Starting image variation generation for: {img_path}")
+
+        # Convert and prepare the image
+        png = Image.open(img_path).convert('RGBA')
+        background = Image.new('RGBA', png.size, (255, 255, 255))
+        alpha_composite = Image.alpha_composite(background, png)
+        alpha_composite.save(img_path, 'PNG', quality=80)
+        logger.info("Image prepared for variation generation.")
+
+        client = OpenAI()
+        variation_response = client.images.create_variation(
+            image=open(img_path, "rb"),
+            n=num_img,
+            size=img_size,
+            response_format=response_format
+        )
+
+        # Saving the generated image
+        generated_image_path = save_generated_image(variation_response, image_dir)
+        logger.info(f"Image variation generated and saved to: {generated_image_path}")
+        return generated_image_path
+
+    except Exception as e:
+        logger.error(f"Error occurred during image variation generation: {e}")
+        sys.exit(f"Exiting due to critical error: {e}")