import gradio as gr
from model.face_enhancer import enhance_face
from model.dress_enhancer import enhance_clothes

def enhance_image(input_image):
    face_enhanced = enhance_face(input_image)
    final_image = enhance_clothes(face_enhanced)
    return final_image

demo = gr.Interface(
    fn=enhance_image,
    inputs=gr.Image(type="pil", label="Upload an image"),
    outputs=gr.Image(type="pil", label="Enhanced Image"),
    title="Face + Dress Image Enhancer",
    description="Enhances facial details using GFPGAN and clothing using SwinIR.",
)

if __name__ == "__main__":
    demo.launch()