Create app.py

app.py

@@ -0,0 +1,50 @@
+import streamlit as st
+from openai import OpenAI
+
+# Initialize OpenAI client with secret key from Hugging Face space settings
+client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
+
+# Sample prompt
+sample_prompt = (
+    "A highly detailed, realistic architectural floor plan for a modern tiny home. "
+    "The design features an open-concept layout with a multi-functional living space "
+    "that combines a compact living area, dining space, and efficient kitchen with smart storage. "
+    "A cozy loft bedroom is accessible via a sleek staircase or ladder. The minimalist bathroom "
+    "includes a shower. Emphasize large windows with natural light, clean lines, and neutral tones "
+    "with subtle accents. Annotate key dimensions and furniture placements. Professional architectural rendering style."
+)
+
+# UI
+st.title("🧠 OpenAI Image Generator")
+st.markdown("Generate images using DALL·E 3 based on your prompt.")
+
+# Input with optional sample prompt
+user_prompt = st.text_area("Enter your prompt:", height=200)
+
+if st.button("Insert Sample Prompt"):
+    st.session_state['prompt'] = sample_prompt
+    st.experimental_rerun()
+
+# Maintain prompt in session state
+user_prompt = st.session_state.get('prompt', user_prompt)
+
+# Display updated prompt
+st.text_area("Current prompt:", user_prompt, height=200, disabled=True)
+
+if st.button("Generate Image") and user_prompt.strip():
+    with st.spinner("Generating image..."):
+        try:
+            response = client.images.generate(
+                model="dall-e-3",
+                prompt=user_prompt,
+                size="1024x1024",
+                quality="standard",
+                n=1
+            )
+            image_url = response.data[0].url
+            st.image(image_url, caption="Generated Image", use_column_width=True)
+            st.markdown(f"[Download Image]({image_url})", unsafe_allow_html=True)
+        except Exception as e:
+            st.error(f"Error: {e}")
+else:
+    st.info("Enter a prompt or insert a sample to get started.")