Update app.py

app.py

@@ -17,6 +17,13 @@ if api_key:
     # Prompt input field
     prompt = st.text_area("Enter your prompt:")
 
+    # Quality selection dropdown
+    quality = st.selectbox(
+        "Select image quality:",
+        options=["standard", "hd"],
+        index=0  # Default to "standard"
+    )
+
     # Variables to store the generated image
     image_url = None
     image_data = None
@@ -24,7 +31,7 @@ if api_key:
     # Button to generate image
     if st.button("Generate Image"):
         if prompt:
-            st.write("Generating image with DALL-E 3...")
+            st.write(f"Generating {quality}-quality image with DALL-E 3...")
             try:
                 # Call the OpenAI API to generate the image
                 response = openai.Image.create(
@@ -32,7 +39,7 @@ if api_key:
                     prompt=prompt,
                     n=1,  # Number of images to generate
                     size="1024x1024",  # Available sizes: 256x256, 512x512, or 1024x1024
-                    quality="standard"  # Specify image quality: "standard" or "hd"
+                    quality=quality  # Use selected quality
                 )
                 image_url = response['data'][0]['url']
 
@@ -42,7 +49,6 @@ if api_key:
 
                 # Display the image
                 st.image(image_url, caption="Generated Image", use_column_width=True)
-                st.write("Image URL: " + image_url)
             except Exception as e:
                 st.error(f"An error occurred: {e}")
         else:
@@ -53,7 +59,7 @@ if api_key:
         st.download_button(
             label="Download Image",
             data=image_data,
-            file_name="generated_image.png",
+            file_name=f"generated_image_{quality}.png",
             mime="image/png"
         )
 else: