feat: Refactorizar función de refinamiento de imágenes en app.py

app.py

@@ -96,7 +96,7 @@ def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width
         used_seed = result[1]             # Esto puede necesitar ser ajustado basado en la estructura real de result
         used_prompt = prompt              # El prompt usado es simplemente el prompt procesado
 
-        return generated_image_path, used_seed, used_prompt 
+        return [generated_image_path, generated_image_path], used_seed, used_prompt 
 
     except Exception as e:
         gr.Warning("Error: " + str(e))
@@ -136,7 +136,7 @@ def save_image(image):
 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:
+    image = image[1]
     try:
         client_tile_upscaler = Client(tile_upscaler_url)
         print(f"Loaded custom model from {tile_upscaler_url}")
@@ -175,7 +175,7 @@ def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_infer
     
     client = InferenceClient()
     refined_image = client.image_to_image(
-        image, 
+        image[0], 
         prompt=prompt, 
         negative_prompt=negative_prompt, 
         num_inference_steps=num_inference_steps, 
@@ -184,7 +184,7 @@ def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_infer
         model=model,
         strength=strength
     )
-    return refined_image
+    return [image[0], refined_image]
 
 
 css="""
@@ -337,7 +337,7 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         with gr.Column():
-            generated_image = gr.Image(label="Input Image", type="filepath")
+            generated_image = gr.ImageSlider(label="Image / Refined Image", type="filepath")
             enhace_button = gr.Button("Enhance Image")
 
         with gr.Column():