Update app.py

app.py

@@ -3,22 +3,21 @@ import gradio as gr
 from openai import OpenAI
 client = OpenAI(api_key=os.getenv("API_KEY"))
 
-def generate_image(prompt, code, size_512):
+def generate_image(prompt, code):
     if code != os.getenv("code"):
         raise gr.Error("❗ Не верный ключ!")
         return None
-    size = "256x256" if size_512 else "1024x1024"
     response = client.images.generate(
-        model="dall-e-3",
-        prompt=prompt,
-        size=size,
-        n=1,
+    model="dall-e-3",
+    prompt=prompt,
+    size="1024x1024",
+    quality="hd",
+    n=1,
     )
 
     image_url = response.data[0].url
     print(image_url)
     return image_url
-
 css = """
 footer {visibility: hidden !important;}
 """
@@ -30,12 +29,10 @@ with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
                 code = gr.Textbox(label="Ключ доступа", type="password")
             with gr.Row():
                 prompt_input = gr.Textbox(label="Описание изображения", lines=3)
-            with gr.Row():
-                size_512_checkbox = gr.Checkbox(label="256x256", value=False)
             submit_btn = gr.Button("Генерация", variant='primary')
         with gr.Column():
             image_output = gr.Image(label="Изображение")
     
-    submit_btn.click(fn=generate_image, inputs=[prompt_input, code, size_512_checkbox], outputs=image_output)
+    submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)
 
 demo.launch()