import gradio as gr
from preprocess import preprocess_single_image  # Importamos la función de preprocesamiento

def process_image(img):
    """
    Recibe una imagen del usuario y la devuelve en dos versiones:
    - Imagen normalizada
    - Imagen con ruido agregado
    """
    img_clean, img_noisy = preprocess_single_image(img)
    return img_clean, img_noisy  # Devuelve ambas imágenes

# --- Interfaz de Gradio ---
gr.Interface(fn=process_image,
             inputs=gr.Image(type="pil"),
             outputs=[gr.Image(type="numpy", label="Imagen Normalizada"),
                      gr.Image(type="numpy", label="Imagen con Ruido")],
             examples=["example1.jpg", "example2.jpg"]  # Opcional
).launch()