| from typing import Any | |
| import gradio as gr | |
| from models.interface_types import InterfaceType | |
| from state import get_settings, get_context | |
| from backend.upscale.upscaler import upscale_image | |
| from backend.models.upscale import UpscaleMode | |
| from paths import FastStableDiffusionPaths | |
| app_settings = get_settings() | |
| previous_width = 0 | |
| previous_height = 0 | |
| previous_model_id = "" | |
| previous_num_of_images = 0 | |
| def create_upscaled_image( | |
| source_image, | |
| upscale_mode, | |
| ) -> Any: | |
| context = get_context(InterfaceType.WEBUI) | |
| scale_factor = 2 | |
| if upscale_mode == "SD": | |
| mode = UpscaleMode.sd_upscale.value | |
| elif upscale_mode == "AURA-SR": | |
| mode = UpscaleMode.aura_sr.value | |
| scale_factor = 4 | |
| else: | |
| mode = UpscaleMode.normal.value | |
| upscaled_filepath = FastStableDiffusionPaths.get_upscale_filepath( | |
| None, | |
| scale_factor, | |
| app_settings.settings.generated_images.format, | |
| ) | |
| image = upscale_image( | |
| context=context, | |
| src_image_path=source_image, | |
| dst_image_path=upscaled_filepath, | |
| upscale_mode=mode, | |
| ) | |
| return image | |
| def get_upscaler_ui() -> None: | |
| with gr.Blocks(): | |
| with gr.Row(): | |
| with gr.Column(): | |
| input_image = gr.Image(label="Image", type="filepath") | |
| with gr.Row(): | |
| upscale_mode = gr.Radio( | |
| ["EDSR", "SD", "AURA-SR"], | |
| label="Upscale Mode (2x) | AURA-SR v2 (4x)", | |
| info="Select upscale method(For SD Scale GGUF mode is not supported)", | |
| value="EDSR", | |
| ) | |
| generate_btn = gr.Button( | |
| "Upscale", | |
| elem_id="generate_button", | |
| scale=0, | |
| ) | |
| input_params = [ | |
| input_image, | |
| upscale_mode, | |
| ] | |
| with gr.Column(): | |
| output = gr.Gallery( | |
| label="Generated images", | |
| show_label=True, | |
| elem_id="gallery", | |
| columns=2, | |
| height=512, | |
| ) | |
| generate_btn.click( | |
| fn=create_upscaled_image, | |
| inputs=input_params, | |
| outputs=output, | |
| ) | |