Update app.py

app.py

@@ -113,11 +113,10 @@ def generate_image_callback():
     prompt = st.session_state.get("prompt_input", "")
     if prompt:
         st.session_state["prompt"] = prompt
-        # Add a spinner while generating the image
-        with st.spinner('Generating image...'):
-            image = generate_image(prompt)
-            if image:
-                st.session_state["image"] = image
+        # The spinner is handled outside this function
+        image = generate_image(prompt)
+        if image:
+            st.session_state["image"] = image
     else:
         st.warning("Please enter a prompt.")
 
@@ -128,7 +127,7 @@ def reset_callback():
     st.session_state["image"] = None
 
 def main():
-    st.title("MiImage Creator | Image Generation")
+    st.title("Stxtement | Image Generation")
 
     # Input field for the prompt, tied to session state
     st.text_input(
@@ -137,8 +136,18 @@ def main():
         key="prompt_input"
     )
 
-    # Generate Image button with callback
-    st.button("Generate Image", on_click=generate_image_callback)
+    # Create a placeholder for the spinner
+    spinner_placeholder = st.empty()
+
+    # Generate Image button
+    generate_button_clicked = st.button("Generate Image")
+    if generate_button_clicked:
+        if st.session_state.get("prompt_input", ""):
+            with spinner_placeholder:
+                with st.spinner('Generating image...'):
+                    generate_image_callback()
+        else:
+            st.warning("Please enter a prompt.")
 
     # Show the image if it exists in the session state
     if st.session_state.get("image"):