Update app.py

app.py

@@ -1,18 +1,17 @@
 import streamlit as st
 import openai
 from dotenv import load_dotenv
-import os
 
 # Streamlit UI for input
 st.title("Generate an Image with DALL-E 3")
 
 # Input for OpenAI API key
-api_key = st.text_input("Enter your OpenAI API Key:",type="password")
+api_key = st.text_input("Enter your OpenAI API Key:", type="password")
 
 # Check if API key is provided
 if api_key:
-    # Set the API key for OpenAI
-    os.environ["OPENAI_API_KEY"] = api_key
+    # Set the API key directly in the OpenAI library
+    openai.api_key = api_key
     load_dotenv()  # Load environment variables if needed
     
     # Prompt input field
@@ -20,7 +19,7 @@ if api_key:
     
     if prompt:
         st.write("Generating image...")
-        
+
         # Call the OpenAI API to generate the image
         try:
             response = openai.Image.create(