import gradio as gr


def build_ui(process_fn):
with gr.Blocks() as demo:
gr.Markdown("# 🌟 Salon Virtual Hairstyle Try-On (Manual Adjustments Enabled)")


with gr.Row():
with gr.Column():
uploaded_image = gr.Image(label="Upload Your Image")
hairstyle_slider = gr.Slider(0, 10, step=1, label="Select Hairstyle")
x_offset = gr.Slider(-200, 200, value=0, step=1, label="X Offset")
y_offset = gr.Slider(-200, 200, value=0, step=1, label="Y Offset")
scale = gr.Slider(0.2, 3.0, value=1.0, step=0.1, label="Scale")
submit_btn = gr.Button("Apply Hairstyle")


with gr.Column():
output_image = gr.Image(label="Result Preview")


submit_btn.click(fn=process_fn,
inputs=[uploaded_image, hairstyle_slider, x_offset, y_offset, scale],
outputs=output_image)
return demo