Create app.py

app.py

@@ -0,0 +1,34 @@
+import streamlit as st
+from openai import OpenAI
+import os
+
+# Fetch the API key from an environment variable
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+st.title("DALL-E Image Generator")
+
+# User inputs a text description for the image
+prompt = st.text_input("Enter a description for the image you want to generate:")
+
+if prompt and st.button('Generate Image'):
+    if not openai_api_key:
+        st.error("No OpenAI API key found. Please set the OPENAI_API_KEY environment variable.")
+        st.stop()
+
+    client = OpenAI(api_key=openai_api_key)
+
+    # Use a spinner to indicate that the image is being generated
+    with st.spinner('Generating image...'):
+        response = client.Image.create(
+            model="dall-e-2",
+            prompt=prompt,
+            n=1,
+            size="1024x1024"
+        )
+
+    # Get the image URL from the response
+    image_url = response.data[0].url
+
+    # Display the generated image
+    st.image(image_url, caption="Generated Image")
+