feat: simplified image generator with optimized settings

.gitignore

@@ -1,4 +1,11 @@
 __pycache__/
 *.py[cod]
-generated_image.png
+*.so
+.Python
+env/
+build/
+dist/
+eggs/
+*.egg-info/
 .env
+generated_image.png

app.py

@@ -5,89 +5,68 @@ import gc
 
 @st.cache_resource
 def load_model():
-    # Use smaller model better suited for CPU
-    model_id = "CompVis/stable-diffusion-v1-4"
+    # Use RunwayML's smaller model
+    model_id = "runwayml/stable-diffusion-v1-5"
     pipe = StableDiffusionPipeline.from_pretrained(
         model_id,
         torch_dtype=torch.float32,
-        safety_checker=None,  # Disable safety checker for memory efficiency
-        requires_safety_checker=False
+        safety_checker=None,
+        requires_safety_checker=False,
+        use_safetensors=True
     ).to("cpu")
     
-    # Memory optimizations
+    # Optimized settings
     pipe.enable_attention_slicing(slice_size=1)
     pipe.enable_vae_slicing()
-    
-    # Clear memory
-    gc.collect()
+    pipe.enable_sequential_cpu_offload()
+    torch.set_num_threads(6)  # Optimal thread count for most systems
     
     return pipe
 
-st.title("🎨 Fast Image Generator (CPU-Optimized)")
-st.write("Using Stable Diffusion v1.4 optimized for CPU")
+st.title("🎨 AI Image Generator")
+st.write("Create images from text descriptions")
 
 # Initialize model
 pipeline = load_model()
 
-# Adjust settings for faster generation
-with st.sidebar:
-    st.header("Generation Settings")
-    num_steps = st.slider("Steps", 20, 50, 30)
-    guidance_scale = st.slider("CFG Scale", 5.0, 15.0, 7.5)
-    width = st.select_slider("Width", options=[256, 384, 512], value=384)
-    height = st.select_slider("Height", options=[256, 384, 512], value=384)
-    
-    st.info("💡 Generation time: 2-5 minutes on CPU")
-
+# Simple interface
 prompt = st.text_area(
     "Describe your image:",
-    "A professional photograph of a serene landscape at sunset, golden hour lighting, 8k uhd, highly detailed"
-)
-
-negative_prompt = st.text_area(
-    "Negative prompt (what to avoid):",
-    "ugly, blurry, low quality, distorted, disfigured, poor details, bad anatomy, watermark"
+    "A professional photograph of a serene landscape at sunset, golden hour lighting, highly detailed"
 )
 
 if st.button("Generate Image"):
-    with st.spinner("Creating your image (2-5 minutes)..."):
+    with st.spinner("Creating your image... (about 1 minute)"):
         try:
-            gc.collect()
-            
+            # Pre-optimized settings
             image = pipeline(
                 prompt=prompt,
-                negative_prompt=negative_prompt,
-                num_inference_steps=num_steps,
-                guidance_scale=guidance_scale,
-                width=width,
-                height=height,
+                negative_prompt="ugly, blurry, low quality, distorted, disfigured",
+                num_inference_steps=20,
+                guidance_scale=7.0,
+                width=320,
+                height=320,
             ).images[0]
             
-            # Clear memory after generation
-            gc.collect()
-            
-            # Display image
+            # Display and download
             st.image(image, caption=prompt, use_column_width=True)
             
             # Save and offer download
             image.save("generated_image.png")
             with open("generated_image.png", "rb") as file:
                 st.download_button(
-                    label="Download HD Image",
+                    label="Download Image",
                     data=file,
                     file_name="generated_image.png",
                     mime="image/png"
                 )
             
         except Exception as e:
-            st.error(f"Error generating image: {str(e)}")
-            st.info("Try reducing image size or number of steps if you encounter memory issues")
+            st.error(f"Error: {str(e)}")
 
 st.markdown("""
-### Tips for best results:
-- Be specific in your descriptions
-- Include details about style, lighting, and perspective
-- Use artistic terms like "highly detailed", "professional", "8k"
-- Adjust steps for quality vs. speed tradeoff
-- Use negative prompts to avoid unwanted elements
+### Tips for good results:
+- Be specific in your description
+- Include details about style and lighting
+- Keep descriptions concise
 """)