Update app.py

app.py

@@ -3,21 +3,23 @@ import gradio as gr
 from openai import OpenAI
 client = OpenAI(api_key=os.getenv("API_KEY"))
 
-def generate_image(prompt, code):
+def generate_image(prompt, code, size_512):
     if code != os.getenv("code"):
         raise gr.Error("❗ Не верный ключ!")
         return None
+    size = "512x512" if size_512 else "1024x1024"
     response = client.images.generate(
-    model="dall-e-3",
-    prompt=prompt,
-    size="1024x1024",
-    quality="hd",
-    n=1,
+        model="dall-e-3",
+        prompt=prompt,
+        size=size,
+        quality="hd",
+        n=1,
     )
 
     image_url = response.data[0].url
     print(image_url)
     return image_url
+
 css = """
 footer {visibility: hidden !important;}
 """
@@ -29,10 +31,12 @@ 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="512x512", 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], outputs=image_output)
+    submit_btn.click(fn=generate_image, inputs=[prompt_input, code, size_512_checkbox], outputs=image_output)
 
 demo.launch()