Update app.py

app.py

@@ -13,7 +13,7 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
 models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL LoRA", "epiCRealism", "PixelArt XL", "NewReality XL"]
 
-def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
+def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None, width, height):
     language = detect(prompt)
     key = random.randint(0, 999)
 
@@ -54,7 +54,9 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
         "is_negative": is_negative,
         "steps": steps,
         "cfg_scale": cfg_scale,
-        "seed": seed if seed is not None else random.randint(-1, 2147483647)
+        "seed": seed if seed is not None else random.randint(-1, 2147483647),
+        "width": width,
+        "height": height
         }
 
     image_bytes = requests.post(API_URL, headers=headers, json=payload).content
@@ -76,12 +78,15 @@ with gr.Blocks(css=css) as dalle:
 
     with gr.Tab("Расширенные настройки"):
         negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Чего не должно быть на изображении", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness", lines=3, elem_id="negative-prompt-text-input")
+        width = gr.Slider(lavel="Ширина", value=512, minimum=15, maximum=1023)
+        height = gr.Slider(lavel="Высота", value=512, minimum=15, maximum=1023)
+
 
     with gr.Row():
         text_button = gr.Button("Генерация", variant='primary', elem_id="gen-button")
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, width, height], outputs=image_output)
 
 dalle.launch(show_api=False)