Update app.py

app.py

@@ -1,14 +1,18 @@
 import streamlit as st
 import openai
+import requests
+from PIL import Image
+from io import BytesIO
 
-# Access the OpenAI API key
-openai.api_key = st.secrets["YOUR_OPENAI_API_KEY"]
+# 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("Real Estate Video Background Image Generator")
+st.title("Background Image Generator for Real Estate Videos")
 
-# User inputs for image generation
-image_theme = st.text_input("Image Theme", placeholder="Enter a theme for your image (e.g., cozy living room, modern office)")
-color_scheme = st.text_input("Color Scheme", placeholder="Preferred color scheme (e.g., warm, cool, monochromatic)")
+# User inputs
+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?")
 
 if st.button('Generate Image'):
@@ -16,12 +20,22 @@ if st.button('Generate Image'):
     prompt = f"Create an image with the theme: '{image_theme}', color scheme: '{color_scheme}', including elements: '{additional_elements}'."
 
     # Call the DALL-E API
-    response = openai.Image.create(
-        prompt=prompt,
-        n=1,  # Number of images to generate
-        size="1024x1024"  # Image dimensions
-    )
-
-    # Display the image
-    image_data = response.data[0]  # Assuming response contains image data
-    st.image(image_data, caption='Generated Image')
+    try:
+        response = openai.Image.create(
+            prompt=prompt,
+            n=1,  # Number of images to generate
+            size="1024x1024"  # Image dimensions
+        )
+
+        # Assuming the response contains a URL to the image
+        image_url = response['data'][0]['url']
+        
+        # Fetch the image from the URL
+        image_response = requests.get(image_url)
+        image = Image.open(BytesIO(image_response.content))
+
+        # Display the image
+        st.image(image, caption='Generated Image')
+
+    except Exception as e:
+        st.error(f"An error occurred: {e}")