Update app.py

app.py

@@ -1,23 +1,32 @@
 import os
+from dotenv import load_dotenv
 import openai
 import gradio as gr
 
-# Load API key from Hugging Face Secrets (not .env)
-api_key = os.environ.get("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")
+# Load environment variables
+load_dotenv()
+
+# Get the API key from Hugging Face Secrets
+api_key = os.getenv("OPENAI_API_KEY")
+
+# Ensure API key is set
+if not api_key:
+    raise ValueError("OPENAI_API_KEY is not set. Please check your Hugging Face Secrets.")
+
 openai.api_key = api_key
 
 def generate_image(prompt):
-    response = openai.images.generate(
-        model="dall-e-3",
+    response = openai.Image.create(
         prompt=prompt,
-        size="1024x1024",
         n=1,
+        size="1024x1024"
     )
-    return response.data[0].url
+    return response["data"][0]["url"]
 
+# Create Gradio Interface
 gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Describe your image"),
     outputs=gr.Image(label="Generated Image"),
     title="DALL-E Image Generator"
-).launch()
+).launch()