Update app.py

app.py

@@ -5,7 +5,6 @@ from PIL import Image
 from io import BytesIO
 
 # Set up the OpenAI API key from Hugging Face Spaces secrets
-# Ensure that you have added your OpenAI API key in the Hugging Face Spaces secrets under the name "OPENAI_API_KEY"
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
 st.title("Background Image Generator for Real Estate Videos")
@@ -14,6 +13,8 @@ st.title("Background Image Generator for Real Estate Videos")
 image_theme = st.text_input("Image Theme", placeholder="Enter a theme for your image (e.g., modern living room)")
 color_scheme = st.text_input("Color Scheme", placeholder="Preferred color scheme (e.g., warm, cool)")
 additional_elements = st.text_input("Additional Elements", placeholder="Any specific elements to include in the image?")
+image_style = st.selectbox("Image Style", ["vivid", "natural"], index=0)  # Select between vivid and natural styles
+image_quality = st.selectbox("Image Quality", ["standard", "hd"], index=0)  # Select between standard and HD quality
 
 if st.button('Generate Image'):
     # Construct the prompt
@@ -22,9 +23,12 @@ if st.button('Generate Image'):
     # Call the DALL-E API
     try:
         response = openai.Image.create(
+            model="dall-e-3",  # Specify DALL-E 3 model
             prompt=prompt,
             n=1,  # Number of images to generate
-            size="1024x1024"  # Image dimensions
+            size="1024x1024",  # Image dimensions
+            quality=image_quality,  # Image quality parameter
+            style=image_style  # Image style parameter
         )
 
         # Assuming the response contains a URL to the image