feat: Add image resizing functionality and reduce factor option

app.py

@@ -1,8 +1,10 @@
 import os
+import uuid
 import gradio as gr
 import json
 from gradio_client import Client, handle_file
 from gradio_imageslider import ImageSlider
+from PIL import Image
 
 with open('loras.json', 'r') as f:
     loras = json.load(f)
@@ -113,8 +115,21 @@ def update_selection(evt: gr.SelectData):
         evt.index
     )
 
+def resize_image(image_path, reduction_factor):
+    image = Image.open(image_path)
+    width, height = image.size
+    new_size = (width // reduction_factor, height // reduction_factor)
+    resized_image = image.resize(new_size)
+    return resized_image
 
-def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name):
+
+def save_image(image):
+    unique_filename = f"resized_image_{uuid.uuid4().hex}.png"
+    image.save(unique_filename)
+    return unique_filename
+
+
+def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name, reduce_factor):
     global client_tile_upscaler
 
     # if client_tile_upscaler is None:
@@ -125,6 +140,10 @@ def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidanc
         print(f"Failed to load custom model: {e}")
         client_custom_model = None
         raise gr.Error("Failed to load client for " + tile_upscaler_url)
+    
+    if (reduce_factor > 1):
+        image = resize_image(image, reduce_factor)
+        image = save_image(image)
 
     try:
         result = client_tile_upscaler.predict(
@@ -264,6 +283,7 @@ with gr.Blocks(css=css) as demo:
             output_slider = ImageSlider(label="Before / After", type="numpy", show_download_button=False)
 
     with gr.Accordion("Advanced Options", open=False):
+        upscale_reduce_factor = gr.Slide(minimum=1, maximum=10)
         upscale_resolution = gr.Slider(minimum=128, maximum=2048, value=1024, step=128, label="Resolution")
         upscale_num_inference_steps = gr.Slider(minimum=1, maximum=50, value=20, step=1, label="Number of Inference Steps")
         upscale_strength = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.01, label="Strength")
@@ -297,7 +317,7 @@ with gr.Blocks(css=css) as demo:
         outputs=[output_slider]
     ).then(
         upscale_image, 
-        [generated_image, upscale_resolution, upscale_num_inference_steps, upscale_strength, upscale_hdr, upscale_guidance_scale, upscale_controlnet_strength, upscale_scheduler_name],
+        [generated_image, upscale_resolution, upscale_num_inference_steps, upscale_strength, upscale_hdr, upscale_guidance_scale, upscale_controlnet_strength, upscale_scheduler_name, upscale_reduce_factor],
         output_slider
     )