Create app.py

app.py

@@ -0,0 +1,32 @@
+import streamlit as st
+import openai
+import os
+
+openai.api_key = os.getenv("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")
+
+st.title("DALL-E 2 API Image Generation Demo with OpenAI SDK")
+
+st.write("Enter a prompt to generate an image")
+
+prompt = st.text_area("Prompt", "An eco-friendly computer from the 90s in the style of vaporwave")
+
+num_images = st.slider("Number of images to generate", min_value=1, max_value=10, value=1)
+
+image_size = st.selectbox(
+    "Select an image size",
+    ["256x256", "512x512", "1024x1024"]
+)
+
+def generate_images(prompt, num_images, image_size):
+    response = openai.Image.create(
+        prompt=prompt,
+        n=num_images,
+        size=image_size,
+    )
+    return response["data"]
+
+if st.button("Generate Images"):
+    with st.spinner("Generating images..."):
+        image_data = generate_images(prompt, num_images, image_size)
+        for idx, image in enumerate(image_data):
+            st.image(image['url'], caption=f"Image {idx+1}", width=400)