Update app.py

app.py

@@ -69,7 +69,29 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             continue
         if status == 'failed':
             return None
+
+
+def mirror(image_output, scale_by, method, gfpgan, codeformer):
+
+    url_up = os.getenv("url_up")
+    url_up_f = os.getenv("url_up_f")
+
+    scale_by = int(scale_by)
+    gfpgan = int(gfpgan)
+    codeformer = int(codeformer)
     
+    with open(image_output, "rb") as image_file:
+        encoded_string2 = base64.b64encode(image_file.read())
+        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,512,512,None,method,"None",1,False,[],"",""],"session_hash":""}
+    print(data)
+    r = requests.post(f"{url_up}", json=data, timeout=100)
+    print(r.text)
+    ph = f"{url_up_f}" + str(r.json()['data'][0][0]['name'])
+    return ph
+
 css = """
 #generate {
     width: 100%;
@@ -89,6 +111,10 @@ display: flex;
 justify-content: center;
 }
 footer {visibility: hidden !important;}
+
+#image_output {
+height: 100% !important;
+}
 """
 
 with gr.Blocks(css=css) as demo:
@@ -113,12 +139,31 @@ 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.Tab("Настройки апскейлинга"):
+        with gr.Column():
+            with gr.Row():
+                scale_by = gr.Number(show_label=True, label="Во сколько раз увеличить:", minimum=1, maximum=4, value=2, step=1)
+            with gr.Row():
+                method = gr.Dropdown(show_label=True, value="ESRGAN_4x", label="Алгоритм увеличения", choices=["ScuNET GAN", "SwinIR 4x", "ESRGAN_4x", "R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B"])
+        with gr.Column():
+            with gr.Row():
+                gfpgan = gr.Slider(show_label=True, label="Эффект GFPGAN", minimum=0, maximum=1, value=0, step=0.1)
+            with gr.Row():
+                codeformer = gr.Slider(show_label=True, label="Эффект CodeFormer", minimum=0, maximum=1, value=0, step=0.1)
+
+    
     with gr.Column():
         text_button = gr.Button("Сгенерировать изображение", variant='primary', elem_id="generate")
-    with gr.Column(scale=2):
-        image_output = gr.Image(show_label=True, label='Результат:', elem_id='image_output')
-
+    with gr.Column():
+        image_output = gr.Image(show_label=True, 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)
+
+        img2img_b = gr.Button("Увеличить изображение", variant='secondary')
+        image_i2i = gr.Image(show_label=True, label='Увеличенное изображение:')
+        img2img_b.click(mirror, inputs=[image_output, scale_by, method, gfpgan, codeformer], outputs=image_i2i)
+        
+        
     
 demo.queue(concurrency_count=12)
 demo.launch()