Update app.py

add quality、size and style

app.py

@@ -1,7 +1,7 @@
 import requests
 import gradio as gr
 
-def generate_image(prompt, api_key, request_url):
+def generate_image(prompt, quality, size, style, api_key, request_url):
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
@@ -9,8 +9,9 @@ def generate_image(prompt, api_key, request_url):
     data = {
         "prompt": prompt,
         "model": "dall-e-3",
-        "quality": "hd",
-        "size":"1792x1024",
+        "quality": quality,
+        "size": size,
+        "style": style,
         "num_images": 1,
     }
     response = requests.post(request_url, headers=headers, json=data)
@@ -21,23 +22,22 @@ def generate_image(prompt, api_key, request_url):
     
     return image_url
 
-css = """
-footer {visibility: hidden !important;}
-"""
-
 # Create the interface using Gradio
-with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
+with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("# <center> OpenAI dall-e-3 API with Gradio </center>")
-        gr.Markdown("This demo uses the OpenAI dall-e-3 API to generate an image from text. You get free session key from https://dongsiqie.me/sess")
+        gr.Markdown("This demo uses the OpenAI dall-e-3 API to generate an image from text. You get a free session key from https://dongsiqie.me/sess")
     with gr.Row():
         api_key_input = gr.Textbox(label="API Key", type="password")
         request_url_input = gr.Textbox(label="Request URL", value="https://api.openai.com/v1/images/generations")
+        quality_input = gr.Dropdown(label="Quality", choices=["standard", "hd"], value="standard")
+        size_input = gr.Dropdown(label="Size", choices=["1792x1024", "1024x1024", "1024x1792"], value="1792x1024")
+        style_input = gr.Dropdown(label="Style", choices=["vivid", "natural"], value="vivid")
     with gr.Row():
-        prompt_input = gr.Textbox(label="Image Description",value="A furry, long-haired cat with distinctive orange and white stripes.")
+        prompt_input = gr.Textbox(label="Image Description", value="A furry, long-haired cat with distinctive orange and white stripes.")
     submit_btn = gr.Button("Generate Image", variant='primary')
     image_output = gr.Image(label="Generated Image")
     
-    submit_btn.click(fn=generate_image, inputs=[prompt_input, api_key_input, request_url_input], outputs=image_output)
+    submit_btn.click(fn=generate_image, inputs=[prompt_input, quality_input, size_input, style_input, api_key_input, request_url_input], outputs=image_output)
 
 demo.launch()