Update app.py

app.py

@@ -12,6 +12,13 @@ import html
 import PIL
 from PIL import Image
 import re
+import cv2
+import numpy as np
+
+def upscale_image(input_image, radio_input):
+    upscale_factor = radio_input
+    output_image = cv2.resize(input_image, None, fx = upscale_factor, fy = upscale_factor, interpolation = cv2.INTER_CUBIC)
+    return [output_image], output_image
 
 def send_prompt(prompt):
     return prompt
@@ -342,6 +349,7 @@ with gr.Blocks(css=css) as demo:
                         i2i_cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
                         i2i_denoising = gr.Slider(label="Denoising Strength", minimum=0, maximum=1, value=0.7, step=0.1)
                         i2i_seed = gr.Number(label="Seed", value=-1)
+                        
 
                 with gr.Column(scale=2):
                     i2i_image_output = gr.Gallery(show_label=False, rows=2)
@@ -468,6 +476,21 @@ with gr.Blocks(css=css) as demo:
             send_to_img2img_prompt.click(send_prompt, inputs=prompt_full, outputs=i2i_prompt)
             idea.submit(prompt_gen, inputs=idea, outputs=prompt_full)
             prompt_button.click(prompt_gen, inputs=idea, outputs=prompt_full)
+        with gr.Tab("Upscaler"):
+            gr.HTML(value="""
+            <h1><center>Upscaler</center></h1>
+            """)
+            with gr.Row():
+                us_input = gr.Image(show_label=False, interactive=True)
+                us_radio = gr.Radio(label="Upscale Levels", choices=[2, 4, 6, 8, 10], value=2)
+            us_button = gr.Button(value="Generate with Upscaler", variant="primary")
+            with gr.Row():
+                us_output = gr.Gallery(show_label=False, rows=3, preview=True)
+                send_to_img2img_us = gr.Button(value="Send UPSCALED IMAGE to img2img")
+                us_past = gr.Image(interactive=False)
+        
+            us_button.click(upscale_image, inputs=[us_input, us_radio], outputs=[us_output, us_past])
+            send_to_img2img_us.click(send_to_img2img_def, inputs=us_past, outputs=i2i_image_input)
 
         with gr.Tab("BLIP"):
             with gr.Tab("Base"):