Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
-from openai import OpenAI
+import openai
 import os
 
 # Fetch the API key from an environment variable
 openai_api_key = os.getenv("OPENAI_API_KEY")
+openai.api_key = openai_api_key
 
 st.title("DALL-E Image Generator")
 
@@ -15,20 +16,18 @@ if prompt and st.button('Generate Image'):
         st.error("No OpenAI API key found. Please set the OPENAI_API_KEY environment variable.")
         st.stop()
 
-    client = OpenAI(api_key=openai_api_key)
-
     # Use a spinner to indicate that the image is being generated
     with st.spinner('Generating image...'):
-        response = client.Image.create(
-            model="dall-e-2",
-            prompt=prompt,
-            n=1,
-            size="1024x1024"
-        )
-
-    # Get the image URL from the response
-    image_url = response.data[0].url
-
-    # Display the generated image
-    st.image(image_url, caption="Generated Image")
+        # Ensure you use the correct function based on your API documentation. Below is for demonstration.
+        try:
+            response = openai.Image.create(
+                model="image-dalle-002", # Use the correct model identifier
+                prompt=prompt,
+                n=1,
+                size="1024x1024"
+            )
+            image_url = response['data'][0]['url']  # Make sure response structure is correct
+            st.image(image_url, caption="Generated Image")
+        except Exception as e:
+            st.error(f"Failed to generate an image: {e}")