Update app.py

app.py

@@ -25,66 +25,20 @@ openai.api_key = st.secrets["openai_api_key"]
 # openai_api_key = os.getenv('OPENAI_API_KEY')
 # if not openai_api_key:
 #     raise ValueError("OPENAI_API_KEY not set in environment variables")
-# # 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
-
-# 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)
-
+# 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
-
-# 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.')
-
-
-
+    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")