use DALLE-3

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
-from langchain.llms import OpenAI
+from langchain.llms import OpenAI as LangChainOpenAI
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
-import openai
+from openai import OpenAI
 import os
 import re
 
-# Set your OpenAI API key
-openai.api_key = os.environ["OPENAI_API_KEY"]
+# Initialize OpenAI client
+client = OpenAI()
 
 # Create a prompt template
 template = """
@@ -22,17 +22,19 @@ Story:
 prompt = PromptTemplate(template=template, input_variables=["topic"])
 
 # Create an LLMChain
-llm = OpenAI(temperature=0.7)
+llm = LangChainOpenAI(temperature=0.7)
 story_chain = LLMChain(llm=llm, prompt=prompt)
 
 def generate_image(prompt):
     try:
-        response = openai.Image.create(
+        response = client.images.generate(
+            model="dall-e-3",
             prompt=prompt,
+            size="1024x1024",
+            quality="standard",
             n=1,
-            size="512x512"
         )
-        return response['data'][0]['url']
+        return response.data[0].url
     except Exception as e:
         print(f"Error generating image: {e}")
         return None
@@ -49,26 +51,26 @@ def generate_story_with_images(topic):
     while len(paragraphs) < 3:
         paragraphs.append("...")
 
-    # Generate an image for each paragraph
-    result = []
+    # Generate images for each paragraph
+    image_urls = []
     for paragraph in paragraphs:
         image_url = generate_image(paragraph)
-        result.append((paragraph, image_url))
+        image_urls.append(image_url)
     
-    return result
+    return paragraphs, image_urls
 
 # Create the Gradio interface
 def gradio_interface(topic):
     try:
-        story_with_images = generate_story_with_images(topic)
-        output = []
-        for paragraph, image_url in story_with_images:
-            output.append(paragraph)
-            output.append(image_url if image_url else gr.Markdown("Image generation failed"))
-        return output
+        paragraphs, image_urls = generate_story_with_images(topic)
+        return [
+            paragraphs[0], image_urls[0],
+            paragraphs[1], image_urls[1],
+            paragraphs[2], image_urls[2]
+        ]
     except Exception as e:
         print(f"Error in gradio_interface: {e}")
-        return ["An error occurred while generating the story."] + [""] * 5
+        return ["An error occurred while generating the story and images."] + [""] * 5
 
 iface = gr.Interface(
     fn=gradio_interface,
@@ -82,7 +84,7 @@ iface = gr.Interface(
         gr.Image(label="Image 3", type="filepath"),
     ],
     title="Story Generator with Images",
-    description="Enter a topic, and the AI will generate a short story with images for each paragraph. If image generation fails, you'll still see the story."
+    description="Enter a topic, and the AI will generate a short story with an image for each paragraph."
 )
 
 # Launch the Gradio app