Update app.py

app.py

@@ -15,15 +15,13 @@ color_scheme = st.text_input("Color Scheme", placeholder="Preferred color scheme
 additional_elements = st.text_input("Additional Elements", placeholder="Any specific elements to include in the image (e.g., Christmas tree, snowflakes)")
 
 # Dropdown for selecting image dimensions
-video_format = st.selectbox("Video Format", ["16:9 (Standard HD - 1920x1080)", "16:9 (4K - 3840x2160)", "1:1 (Square - 1080x1080)", "4:5 (Portrait - 1080x1350)", "9:16 (Tall - 1080x1920)"])
+video_format = st.selectbox("Video Format", ["16:9 (Standard HD - 1920x1080)", "1:1 (Square - 1080x1080)", "4:5 (Portrait - 1080x1350)"])
 
 # Mapping formats to DALL-E API size parameters
 format_size_mapping = {
-    "16:9 (Standard HD - 1920x1080)": "1024x576",  # Closest available size in DALL-E for 1080p
-    "16:9 (4K - 3840x2160)": "1792x1024",  # Closest available size in DALL-E for 4K
+    "16:9 (Standard HD - 1920x1080)": "1024x576",  # Reduced size for 1080p
     "1:1 (Square - 1080x1080)": "1024x1024",
-    "4:5 (Portrait - 1080x1350)": "1024x1792",  # Closest available vertical size
-    "9:16 (Tall - 1080x1920)": "1024x1792"  # Closest available vertical size
+    "4:5 (Portrait - 1080x1350)": "1024x768",  # Reduced size for portrait
 }
 
 if st.button('Generate Image'):
@@ -36,8 +34,8 @@ if st.button('Generate Image'):
             model="dall-e-3",  # Specify DALL-E 3 model
             prompt=prompt,
             n=1,
-            size=format_size_mapping[video_format],  # Set image dimensions based on selected video format
-            quality="hd"  # Set image quality to high definition
+            size=format_size_mapping[video_format],  # Set reduced image dimensions based on selected video format
+            quality="standard"  # Set image quality to standard for lower cost
         )
 
         # Assuming the response contains a URL to the image