Update app.py

app.py

@@ -119,15 +119,17 @@ def main():
         st.session_state["prompt"] = ""
 
     # Input field for the prompt
-    prompt = st.text_input("Enter a prompt for image generation:", value="")
+    prompt = st.text_input("Enter a prompt for image generation:", value=st.session_state["prompt"])
 
     if st.button("Generate Image"):
         if prompt:
-            image = generate_image(prompt)
-            if image:
-                st.session_state["image"] = image  # Store generated image in session state
-                st.image(image, caption="Generated Image")
-                st.session_state["prompt"] = prompt  # Store the prompt in session state
+            # Add a spinner while generating the image
+            with st.spinner('Generating image...'):
+                image = generate_image(prompt)
+                if image:
+                    st.session_state["image"] = image  # Store generated image in session state
+                    st.image(image, caption="Generated Image")
+                    st.session_state["prompt"] = prompt  # Store the prompt in session state
         else:
             st.warning("Please enter a prompt.")
 
@@ -149,8 +151,8 @@ def main():
             # Clear session state variables
             st.session_state["image"] = None
             st.session_state["prompt"] = ""
-            
-            # Clear UI by updating query params (this will force rerun of the app)
+
+            # Clear UI by resetting query params
             st.query_params = {}
 
 #------------------------------------------------------------------------