Update

Author: amitpuri

images/generated_image_2366.png

@@ -46,14 +46,13 @@ def create_image_from_prompt(prompt: str, imagesize: str ="256x256", num_images:
 
 def main():
    prompt = 'While traveling through a dense forest, you stumble upon an ancient, overgrown path veering off from the main trail. Do you dare to explore its mysteries?'
-   output = text_to_image(openai_api_key,openai_api_image_generation, prompt)
+   output = text_to_image(openai_api_key,openai_api_image_generation, prompt) 
+   image_paths1 = write_image([output])
 
-   image_paths = write_image([output])
-   for image_path in image_paths:
-      image = Image.open(image_path)
-      image.show()
    image_urls = create_image_from_prompt(prompt, num_images=2)
-   image_paths = write_image(image_urls)
+   image_paths2 = write_image(image_urls)
+
+   image_paths = [*image_paths1, *image_paths2]
    for image_path in image_paths:
       image = Image.open(image_path)
       image.show()