fix

app.py

@@ -11,13 +11,13 @@ logging.basicConfig(level=logging.INFO)
 
 
 def sanitize_filename(prompt):
-    sanitized = re.sub(r'[^\w\-]', '_', prompt)
-    sanitized = re.sub(r'_+', '_', sanitized)
-    sanitized = sanitized.strip('_')
+    sanitized = re.sub(r"[^\w\-]", "_", prompt)
+    sanitized = re.sub(r"_+", "_", sanitized)
+    sanitized = sanitized.strip("_")
     return sanitized[:50]
 
 
-def generate_image(api_key, prompt, file_format):
+def generate_image(api_key, prompt, file_format, image_size):
     if not api_key:
         raise gr.Error("API key is required. Please enter your OpenAI API key.")
 
@@ -33,7 +33,7 @@ def generate_image(api_key, prompt, file_format):
         response = client.images.generate(
             model="dall-e-3",
             prompt=prompt,
-            size="1024x1024",
+            size=image_size,
             quality="standard",
             n=1,
         )
@@ -74,9 +74,8 @@ with gr.Blocks() as demo:
 
     **Please Note:**
     1. Image generation typically takes around 30 seconds per image.
-    2. Each image generated will be charged at $0.040 per image (1024x1024 resolution, standard quality).
-    3. For the most up-to-date pricing information, please visit the [OpenAI Pricing Page](https://openai.com/api/pricing/).
-    4. We are not responsible for any failures in image generation. Use this application at your own risk.
+    2. Pricing varies based on image size. For the most up-to-date pricing information, please visit the [OpenAI Pricing Page](https://openai.com/api/pricing/).
+    3. We are not responsible for any failures in image generation. Use this application at your own risk.
     """)
 
     with gr.Row():
@@ -86,6 +85,11 @@ with gr.Blocks() as demo:
                 label="Enter your prompt", placeholder="e.g., a white siamese cat"
             )
             file_format = gr.Radio(["webp", "png"], label="File Format", value="webp")
+            image_size = gr.Radio(
+                ["1024x1024", "1024x1792", "1792x1024"],
+                label="Image Size",
+                value="1024x1024",
+            )
             submit_button = gr.Button("Generate Image")
 
         with gr.Column(scale=1):
@@ -93,7 +97,7 @@ with gr.Blocks() as demo:
 
     submit_button.click(
         generate_image,
-        inputs=[api_key_input, prompt_input, file_format],
+        inputs=[api_key_input, prompt_input, file_format, image_size],
         outputs=image_output,
     )