Update app.py

app.py

@@ -28,26 +28,61 @@ openai.api_key = st.secrets["openai_api_key"]
 # # Set the OpenAI API key
 # openai.api_key = openai_api_key
 
-def generate_images(prompt): #def generate_images(image_description, num_images):
-    response = openai.images.generate(
-        model="dall-e-3",
-        prompt = prompt,
-        size="1024x1024",
-        quality="standard",
-        n = 1,
-    )
-    image_url = response.data[0].url
-    return image_url
+# def generate_images(prompt): #def generate_images(image_description, num_images):
+#     response = openai.images.generate(
+#         model="dall-e-3",
+#         prompt = prompt,
+#         size="1024x1024",
+#         quality="standard",
+#         n = 1,
+#     )
+#     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)
 
-#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)
 
+def generate_images(prompt): #def generate_images(image_description, num_images):
+        response = openai.images.generate(
+            model="dall-e-3",
+            prompt = prompt,
+            size="1024x1024",
+            quality="standard",
+            n = 1,
+        )
+        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.')