Update app.py

app.py

@@ -17,8 +17,8 @@ from fake_useragent import UserAgent
 import random
 
 
-def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
-    result = {"prompt": prompt,"negative_prompt": negative_prompt,"task": task,"steps": steps,"sampler": sampler,"cfg_scale": cfg_scale,"seed": seed}
+def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed, width, height):
+    result = {"prompt": prompt,"negative_prompt": negative_prompt,"task": task,"steps": steps,"sampler": sampler,"cfg_scale": cfg_scale,"seed": seed, "width":width, "height":height}
     print(result)
     try:
         language = detect(prompt)
@@ -176,6 +176,8 @@ def mirror(image_output, scale_by, method, gfpgan, codeformer):
     return img
 
 css = """
+.gradio-container {background-color: MediumAquaMarine}
+footer{display:none !important}
 #generate {
     width: 100%;
     background: #e253dd !important;
@@ -216,6 +218,10 @@ with gr.Blocks(css=css) as demo:
             cfg_scale = gr.Slider(show_label=True, label="CFG Scale:", minimum=1, maximum=20, value=7, step=1)
         with gr.Row():
             seed = gr.Number(show_label=True, label="Seed:", minimum=-1, maximum=1000000, value=-1, step=1)
+        with gr.Row():
+            width = gr.Slider(label="Width", minimum=512, maximum=2048, step=8, value=896,)
+        with gr.Row():
+            height = gr.Slider(label="Height", minimum=512, maximum=2048, step=8, value=1152,)
     
     with gr.Tab("Upscaling Settings"):
         with gr.Column():
@@ -233,10 +239,10 @@ with gr.Blocks(css=css) as demo:
         text_button = gr.Button("Generate image", variant='primary', interactive=True, elem_id="generate")
     with gr.Column():
         image_output = gr.Image(show_download_button=True, interactive=False, label='Результат:', elem_id='image_output', type='filepath')
-        text_button.click(flip_text, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed], outputs=image_output, concurrency_limit=48)
+        text_button.click(flip_text, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed, width, height], outputs=image_output, concurrency_limit=48)
         
         img2img_b = gr.Button("Increase the image", variant='secondary')
         image_i2i = gr.Image(show_label=True, label='Increased image:')
         img2img_b.click(mirror, inputs=[image_output, scale_by, method, gfpgan, codeformer], outputs=image_i2i, concurrency_limit=48)
     
-demo.launch()
+demo.launch()