Update app.py

app.py

@@ -60,29 +60,29 @@ def generate_images(prompt): #def generate_images(image_description, num_images)
         image_url = response.data[0].url
         return image_url
 
-    # Use a unique key for the text area to avoid conflicts with other session_state usages
-    image_description_prompt = st.text_area("Enter a description for the image you want to generate", key="image_description")
-
-    if st.button("Generate Images"):
-        with st.spinner("Analyzing the image ..."):
-            # Generate the image and store its URL in session_state
-            st.session_state['generated_image_url'] = generate_images(image_description_prompt)
-
-    # Check if the 'generated_image_url' key exists in session_state and display the image and download button
-    if 'generated_image_url' in st.session_state and st.session_state['generated_image_url']:
-        st.image(st.session_state['generated_image_url'], caption="Generated Image")
-
-        # Fetch the image from the URL to enable downloading
-        response = requests.get(st.session_state['generated_image_url'])
-        if response.status_code == 200:
-            img_bytes = BytesIO(response.content)
-            st.download_button(
-                label="Download Image",
-                data=img_bytes,
-                file_name="generated_image.png",
-                mime="image/png"
-            )
-        st.success('Powered by MTSS GPT. AI can make mistakes. Consider checking important information.')
+# Use a unique key for the text area to avoid conflicts with other session_state usages
+image_description_prompt = st.text_area("Enter a description for the image you want to generate", key="image_description")
+
+if st.button("Generate Images"):
+    with st.spinner("Analyzing the image ..."):
+        # Generate the image and store its URL in session_state
+        st.session_state['generated_image_url'] = generate_images(image_description_prompt)
+
+# Check if the 'generated_image_url' key exists in session_state and display the image and download button
+if 'generated_image_url' in st.session_state and st.session_state['generated_image_url']:
+    st.image(st.session_state['generated_image_url'], caption="Generated Image")
+
+    # Fetch the image from the URL to enable downloading
+    response = requests.get(st.session_state['generated_image_url'])
+    if response.status_code == 200:
+        img_bytes = BytesIO(response.content)
+        st.download_button(
+            label="Download Image",
+            data=img_bytes,
+            file_name="generated_image.png",
+            mime="image/png"
+        )
+    st.success('Powered by MTSS GPT. AI can make mistakes. Consider checking important information.')