Update app.py

app.py

@@ -39,12 +39,35 @@ def generate_images(prompt): #def generate_images(image_description, num_images)
     image_url = response.data[0].url
     return image_url
 
-prompt = st.text_area("Enter a description for the image you want to generate")
+# prompt = st.text_area("Enter a description for the image you want to generate")
+
+# #create a button
+# if st.button("Generate Images"):
+#     with st.spinner("Analyzing the image ..."):
+#         generate_image=generate_images(prompt)  #generate_image=generate_images(prompt, num_of_images)
+#         st.image(generate_image)
+
+# 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")
 
-#create a button
 if st.button("Generate Images"):
     with st.spinner("Analyzing the image ..."):
-        generate_image=generate_images(prompt)  #generate_image=generate_images(prompt, num_of_images)
-        st.image(generate_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.')