Upload 2 files

app.py

@@ -0,0 +1,77 @@
+import streamlit as st
+from diffusers import StableDiffusionPipeline
+from PIL import Image
+
+# Initialize the Stable Diffusion pipeline (do this ONCE)
+@st.cache_resource  # Cache the pipeline for efficiency
+def load_model():
+    pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1")
+    return pipe
+
+pipe = load_model()
+
+# Pre-generated prompts (you can customize these)
+PRE_GENERATED_PROMPTS = [
+    "A photorealistic image of a majestic lion in a savanna at sunset",
+    "A vibrant abstract painting with swirling colors",
+    "A futuristic cityscape with flying cars and neon lights",
+    "A cute cartoon cat wearing a top hat and monocle",
+    "A fantasy landscape with floating islands and waterfalls",
+    "A surreal dreamscape with melting clocks and distorted figures",
+    "A portrait of a person with glowing eyes and intricate tattoos",
+    "A science fiction scene with a spaceship landing on a distant planet",
+    "A whimsical illustration of a fairy tale character",
+    "A hyperrealistic image of a delicious-looking pizza",
+]
+
+st.title("Stable Diffusion 2.1 Image Generator")
+
+# Layout with columns for better organization
+col1, col2 = st.columns([1, 1])  # Adjust column ratios as needed
+
+with col1:
+    st.header("Image Generation")
+
+    prompt = st.text_area("Prompt", height=150)  # Larger text area
+    negative_prompt = st.text_area("Negative Prompt (Optional)", height=100)
+
+    # Prompt selection dropdown
+    selected_prompt = st.selectbox("Or choose a pre-generated prompt:", [""] + PRE_GENERATED_PROMPTS) # Add empty option
+    if selected_prompt:
+        prompt = selected_prompt # If a prompt is selected, update the prompt text area
+
+    num_images = st.slider("Number of Images", min_value=1, max_value=4, value=1)
+    #seed = st.number_input("Seed (Optional, for reproducibility)", value=None, placeholder="Enter an integer seed")  # Make seed optional
+    seed = st.text_input("Seed (Optional, for reproducibility)", placeholder="Enter an integer seed")  # Make seed optional
+
+    if st.button("Generate"):
+        if not prompt:  # Basic validation
+            st.error("Please enter a prompt.")
+        else:
+            try:
+                images = []
+                for i in range(num_images):
+                  if seed:
+                      try:
+                        seed = int(seed)
+                      except ValueError:
+                        st.error("Seed must be an integer.")
+                        break
+                  else:
+                    seed = None # if seed input is empty, make seed None so it can be random
+
+                  image = pipe(prompt, negative_prompt=negative_prompt, seed=seed).images[0]
+                  images.append(image)
+
+                for image in images:
+                    st.image(image, use_column_width=True) # Use column width for better display
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+
+with col2:
+    st.header("Prompt Examples")
+    # Display pre-generated prompts in a more visually appealing way
+    for p in PRE_GENERATED_PROMPTS:
+        st.markdown(f"- *{p}*")  # Use markdown for italics and bullet points
+    st.markdown("---")  # Separator
+    st.markdown("You can use these prompts as inspiration or modify them to create your own.")

requirements.txt

@@ -0,0 +1 @@
+