Update app.py

app.py

@@ -22,7 +22,7 @@ app = FastAPI()
 @app.get("/")
 
 def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cfg_scale=7, seed=-1, width=896, height=1152):
-    result = {prompt:"prompt", negative_prompt:"negative_prompt", task:"task", steps:"steps", sampler:"sampler", cfg_scale:"cfg_scale", width:"width" , height:"height" }
+    result = {prompt:"prompt", negative_prompt:"negative_prompt", task:"task", steps:"steps", sampler:"sampler", cfg_scale:print(result)
     print(result)
     try:
         language = detect(prompt)
@@ -33,15 +33,13 @@ def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cf
         pass
 
     prompt = re.sub(r'[^a-zA-Zа-яА-Я\s]', '', prompt)
-
     
     cfg = int(cfg_scale)
     steps = int(steps)
     seed = int(seed)
-    width = int(width)
-    height = int(height)
-    style_name = int(style_name)
-    progress=gr.Progress(track_tqdm=True)
+
+    width = 1024
+    height = 1024
 
     if task == "Playground v2":
         ua = UserAgent()
@@ -115,7 +113,7 @@ def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cf
                     'user-agent': f'{ua.random}'
                 }
                 client = Client("Nymbo/SD-XL", headers=headers)
-                result = client.predict(prompt,negative_prompt,"","",True,False,False,0,896,1152,7,1,25,25,False,api_name="/run")
+                result = client.predict(prompt,negative_prompt,"","",True,False,False,0,1024,1024,7,1,25,25,False,api_name="/run")
                 return result
             except:
                 try:
@@ -171,7 +169,7 @@ def mirror(image_output, scale_by, method, gfpgan, codeformer):
         encoded_string2 = str(encoded_string2).replace("b'", '')
 
     encoded_string2 = "data:image/png;base64," + encoded_string2
-    data = {"fn_index":81,"data":[0,0,encoded_string2,None,"","",True,gfpgan,codeformer,0,scale_by,896,1152,None,method,"None",1,False,[],"",""],"session_hash":""}
+    data = {"fn_index":81,"data":[0,0,encoded_string2,None,"","",True,gfpgan,codeformer,0,scale_by,512,512,None,method,"None",1,False,[],"",""],"session_hash":""}
     r = requests.post(url_up, json=data, timeout=100)
     print(r.text)
     print(r.json()['data'][0][0]['name'])
@@ -197,49 +195,61 @@ footer {
 """
 
 with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
+
+with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
+
     with gr.Tab("Basic Settings"):
-        with gr.Column(scale=1):
-            with gr.Row():
-                prompt = gr.Textbox(placeholder="Enter the image description...", show_label=True, label='Image Prompt ✍️', lines=3, scale=6, show_copy_button = True)
-            with gr.Row():
-                task = gr.Radio(interactive=True, value="Stable Diffusion XL 1.0", show_label=True, label="Model of neural network 🤖 ", choices=['Stable Diffusion XL 1.0', 'Crystal Clear XL', 
-                                                                                                                  'Juggernaut XL', 'DreamShaper XL',
-                                                                                                              'SDXL Niji', 'Cinemax SDXL', 'NightVision XL'])
+        with gr.Row():
+            prompt = gr.Textbox(placeholder="Enter the image description...", show_label=True, label='Description of the image:', lines=3)
+        with gr.Row():
+            task = gr.Radio(interactive=True, value="Stable Diffusion XL 1.0", show_label=True, label="Model of neural network:", choices=['Stable Diffusion XL 1.0', 'Crystal Clear XL', 
+                                                                                                              'Juggernaut XL', 'DreamShaper XL',
+                                                                                                              'SDXL Niji', 'Cinemax SDXL', 'NightVision XL',
+                                                                                                              'Playground v2', 'Artigen v3'])
             gr.Examples(
                 examples = examples,    
                 inputs = [prompt], 
             )
-    
+
+            
     with gr.Tab("Extended settings"):
-        with gr.Column(scale=3):
-            with gr.Row():
-                negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=True, label='Negative Prompt:', scale=6, lines=3, value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry")
-        with gr.Group():    
-            with gr.Row():
-                sampler = gr.Dropdown(value="DPM++ S", show_label=True, label="Sampling Method:", choices=[
-                    "DPM++ 2M Karras", "DPM++ 2S a Karras", "DPM2 a Karras", "DPM2 Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "dpmpp_2s_ancestral", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "LMS Karras", "PLMS", "UniPC", "UniPC BH2"])
-            with gr.Row():
-                steps = gr.Slider(show_label=True, label="Sampling Steps:", minimum=1, maximum=50, value=35, step=1)
+        with gr.Row():
+            negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=True, label='Negative Prompt:', lines=3, value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry")
+        with gr.Row():
+            sampler = gr.Dropdown(value="DPM++ SDE Karras", show_label=True, label="Sampling Method:", choices=[
+                "Euler", "Euler a", "Heun", "DPM++ 2M", "DPM++ SDE", "DPM++ 2M Karras", "DPM++ SDE Karras", "DDIM"])
+        with gr.Row():
+            steps = gr.Slider(show_label=True, label="Sampling Steps:", minimum=1, maximum=50, value=35, step=1)
+        with gr.Row():
+            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.Tab("Upscaling Settings"):
+        with gr.Column():
             with gr.Row():
-                cfg_scale = gr.Slider(show_label=True, label="CFG Scale:", minimum=1, maximum=20, value=7, step=1)
+                scale_by = gr.Number(show_label=True, label="How many times to increase:", minimum=1, maximum=2, value=2, step=1)
             with gr.Row():
-                seed = gr.Number(show_label=True, label="Seed:", minimum=-1, maximum=1000000, value=-1, step=1)
+                method = gr.Dropdown(show_label=True, value="ESRGAN_4x", label="Algorithm to enlarge", choices=["ScuNET GAN", "SwinIR 4x", "ESRGAN_4x", "R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B"])
+        with gr.Column():
             with gr.Row():
-                width = gr.Slider(label="Width", minimum=512, maximum=2048, step=8, value=896, interactive=True,)
+                gfpgan = gr.Slider(show_label=True, label="Эффект GFPGAN (to improve the face)", minimum=0, maximum=1, value=0, step=0.1)
             with gr.Row():
-                height = gr.Slider(label="Height", minimum=512, maximum=2048, step=8, value=1152,interactive=True,)
-                
+                codeformer = gr.Slider(show_label=True, label="Effect CodeFormer (to improve the face)", minimum=0, maximum=1, value=0, step=0.1)
     
-    with gr.Column(scale=3):
-        text_button = gr.Button("Generate image", variant="primary", interactive=True, elem_id="generate")
-    with gr.Column(scale=0):
-        image_output = gr.Image(show_download_button=True, interactive=False, label='Generated Image 🌄', show_share_button=False, show_fullscreen_button=True, format="png", elem_id="gallery")
-        
-        text_button.click(query, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed, width, height], outputs=image_output, concurrency_limit=48)
+    with gr.Column():
+        text_button = gr.Button("Generate image", variant='primary', 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)
         clear_prompt =gr.Button("Clear 🗑️",variant="primary", elem_id="clear_button")
         clear_prompt.click(lambda: (None, None), None, [prompt, image_output], queue=False, show_api=False)
         
-
+        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)
+    
 app.queue(default_concurrency_limit=200, max_size=200)  # <-- Sets up a queue with default parameters
 if __name__ == "__main__":
     app.launch()
+