Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from openai import OpenAI
+import openai
 from dotenv import load_dotenv
 import os
 
@@ -14,7 +14,6 @@ if api_key:
     # Set the API key for OpenAI
     os.environ["OPENAI_API_KEY"] = api_key
     load_dotenv()  # Load environment variables if needed
-    client = OpenAI(api_key=api_key)
     
     # Prompt input field
     prompt = st.text_area("Enter your prompt:")
@@ -24,15 +23,13 @@ if api_key:
         
         # Call the OpenAI API to generate the image
         try:
-            response = client.images.generate(
-                model="dall-e-3",
+            response = openai.Image.create(
                 prompt=prompt,
-                size="1024x1024",  # Available size: 1024x1024, 1024x1792, or 1792x1024
-                quality="standard",  # Available quality: "standard" or "hd"
-                n=1,  # Number of images to generate (limit is 1)
+                n=1,  # Number of images to generate
+                size="1024x1024"  # Available sizes: 256x256, 512x512, or 1024x1024
             )
             
-            image_url = response.data[0].url
+            image_url = response['data'][0]['url']
             
             # Display the image
             st.image(image_url, caption="Generated Image", use_column_width=True)