Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import openai
+from openai import OpenAI
 import os
 
 # List of popular styles
@@ -10,12 +10,10 @@ STYLES = [
     "Art Nouveau", "Pop Art", "Minimalist"
 ]
 
-def enhance_prompt(prompt, style):
-    openai.api_key = os.environ.get("OPENAI_API_KEY")
-    
+def enhance_prompt(client, prompt, style):
     enhanced_prompt_request = f"Enhance the following prompt for DALL-E 3 image generation in the style of {style}. Make it more detailed and vivid, while keeping the original intent: '{prompt}'"
     
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4",
         messages=[
             {"role": "system", "content": "You are an expert at creating detailed, vivid prompts for image generation."},
@@ -26,11 +24,11 @@ def enhance_prompt(prompt, style):
     return response.choices[0].message.content.strip()
 
 def generate_image(api_key, prompt, style):
-    openai.api_key = api_key
+    client = OpenAI(api_key=api_key)
     
-    enhanced_prompt = enhance_prompt(prompt, style)
+    enhanced_prompt = enhance_prompt(client, prompt, style)
     
-    response = openai.Image.create(
+    response = client.images.generate(
         model="dall-e-3",
         prompt=enhanced_prompt,
         size="1024x1024",