notification improved

app.py

@@ -10,8 +10,6 @@ from openai import OpenAI
 import openai
 from diffusers import StableDiffusionPipeline
 
-# Placeholder for top-of-page notifications
-top_notification = st.empty()
 
 # Initialize session state variables
 if 'simplified_text' not in st.session_state:
@@ -77,7 +75,7 @@ def generate_caption(image_path):
 
 
 # Create a Streamlit app
-st.title("ARTSPEAK simplifier")
+st.title("ARTSPEAK s i m p l i f i e r")
 
 # Display an image from the local file system
 logo_path = 'logo_artspeak.png'  # Replace with your image path
@@ -200,9 +198,9 @@ if st.session_state['message_content_from_simplified_text']:
 
 st.markdown("---")
 
-#####
-##Diffusor
-#####
+############
+##Diffusor##
+############
 # Load Stable Diffusion model 
 def load_diffusion_model():
     pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
@@ -210,40 +208,46 @@ def load_diffusion_model():
     return pipe
 
 # Function to generate an image and show a notification while processing
-def generate_image(pipe, prompt):
-    top_notification.text('Generating image...')
-    image = pipe(prompt).images[0]
-    top_notification.empty()  # Clear the notification after the process is done
+def generate_image(pipe, prompt, notification_placeholder):
+    with notification_placeholder.container():
+        st.text('Generating image...')
+        image = pipe(prompt).images[0]
+        st.empty()  # Clear the notification after the process is done
     return image
 
-# Example with "Generate Image from New Caption" button
+# Button and notification for generating image from new caption
 if st.button("Generate Image from New Caption"):
+    notification_placeholder = st.empty()
     if st.session_state['new_caption']:
         pipe = load_diffusion_model()
         prompt_caption = f"contemporary art of {st.session_state['new_caption']}"
-        st.session_state['image_from_caption'] = generate_image(pipe, prompt_caption)
+        st.session_state['image_from_caption'] = generate_image(pipe, prompt_caption, notification_placeholder)
 
 # Display the image generated from new caption
 if st.session_state['image_from_caption'] is not None:
     st.image(st.session_state['image_from_caption'], caption="Image from New Caption", use_column_width=True)
 
+
 # Button to generate image from simplified text
 if st.button("Generate Image from Simplified Text"):
+    notification_placeholder = st.empty()
     if st.session_state['simplified_text']:
         pipe = load_diffusion_model()
         prompt_summary = f"contemporary art of {st.session_state['simplified_text']}"
-        st.session_state['image_from_simplified_text'] = generate_image(pipe, prompt_summary)
+        st.session_state['image_from_simplified_text'] = generate_image(pipe, prompt_summary, notification_placeholder)
 
 # Display the image generated from simplified text
 if st.session_state['image_from_simplified_text'] is not None:
     st.image(st.session_state['image_from_simplified_text'], caption="Image from Simplified Text", use_column_width=True)
 
+
 # Button to generate image from press text
 if st.button("Generate Image from Press Text"):
+    notification_placeholder = st.empty()
     if st.session_state['message_content_from_simplified_text']:
         pipe = load_diffusion_model()
         prompt_press_text = f"contemporary art of {st.session_state['message_content_from_simplified_text']}"
-        st.session_state['image_from_press_text'] = generate_image(pipe, prompt_press_text)
+        st.session_state['image_from_press_text'] = generate_image(pipe, prompt_press_text, notification_placeholder)
 
 # Display the image generated from press text
 if st.session_state['image_from_press_text'] is not None: