Update app.py

app.py

@@ -1,87 +1,31 @@
 import streamlit as st
-import torch
-import io
 from diffusers import StableDiffusionPipeline
+import torch
 from PIL import Image
 
-# Disable warnings and legacy configuration
-import warnings
-warnings.filterwarnings("ignore")
-
-# Force CPU-only mode
-torch.backends.cudnn.enabled = False
-torch.cuda.is_available = lambda: False
-
-@st.cache_resource(show_spinner=False)
-def load_model():
-    return StableDiffusionPipeline.from_pretrained(
-        "CompVis/stable-diffusion-v1-4",
+@st.cache_resource
+def load_pipeline():
+    model_id = "runwayml/stable-diffusion-v1-5"
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id,
         torch_dtype=torch.float32,
-        use_safetensors=False,
-        use_auth_token=False,
-        safety_checker=None,
-        revision="fp16",
-        local_files_only=False
-    ).to("cpu")
-
-def main():
-    st.set_page_config(page_title="Poetry to Image", page_icon="🎨")
-    st.title("✨ Romantic Text to Image Converter")
-    
-    # Session state management
-    if 'image' not in st.session_state:
-        st.session_state.image = None
-    if 'error' not in st.session_state:
-        st.session_state.error = None
+        use_safetensors=True
+    )
+    return pipe.to("cpu")  # if you're using CPU on Hugging Face Spaces
 
-    # Sidebar controls
-    with st.sidebar:
-        st.header("Settings")
-        steps = st.slider("Inference Steps", 15, 50, 25)
-        guidance = st.slider("Guidance Scale", 5.0, 15.0, 7.5)
+pipe = load_pipeline()
 
-    # Main interface
-    prompt = st.text_area(
-        "Enter your romantic text:",
-        height=150,
-        placeholder="Example: 'Your eyes shine like stars in the midnight sky...'"
-    )
+st.set_page_config(page_title="Love Text to Image", layout="centered")
+st.title("💌 Love Text to Image Generator")
 
-    if st.button("Generate Image", type="primary"):
-        st.session_state.image = None
-        st.session_state.error = None
-        
-        if prompt.strip():
-            try:
-                with st.spinner("Creating artwork (5-10 minutes)..."):
-                    pipe = load_model()
-                    image = pipe(
-                        prompt,
-                        num_inference_steps=steps,
-                        guidance_scale=guidance
-                    ).images[0]
-                    st.session_state.image = image
-            except Exception as e:
-                st.session_state.error = f"Error: {str(e)}"
-        else:
-            st.session_state.error = "Please enter some text!"
+st.markdown("Write a poem or love message, and see it transformed into an image!")
 
-    # Display results
-    if st.session_state.error:
-        st.error(st.session_state.error)
-    
-    if st.session_state.image:
-        st.image(st.session_state.image, use_column_width=True)
-        
-        # Download functionality
-        buf = io.BytesIO()
-        st.session_state.image.save(buf, format="PNG")
-        st.download_button(
-            "Download Image",
-            buf.getvalue(),
-            "poetry_image.png",
-            "image/png"
-        )
+user_input = st.text_area("Enter your love poem, song, or message here...", height=150)
 
-if __name__ == "__main__":
-    main()
+if st.button("Generate Image"):
+    if user_input.strip() == "":
+        st.warning("Please write something romantic to generate an image!")
+    else:
+        with st.spinner("Generating image..."):
+            image = pipe(user_input).images[0]
+            st.image(image, caption="Generated from your text 💖", use_column_width=True)