Create ui.py

ui.py

@@ -0,0 +1,19 @@
+import gradio as gr
+
+def create_interface(apply_fn, hairstyles):
+    with gr.Blocks() as demo:
+        gr.Markdown("## 💇 Salon Virtual Hairstyle Try-On (Face-Aligned, No Manual Adjustment Needed)")
+        with gr.Row():
+            with gr.Column():
+                img_input = gr.Image(type="pil", label="📤 Upload Your Image")
+                style_choice = gr.Slider(0, max(len(hairstyles)-1, 0), step=1, label="🎨 Select Hairstyle")
+                apply_btn = gr.Button("✨ Apply Hairstyle")
+            with gr.Column():
+                img_output = gr.Image(label="🔍 Result Preview")
+
+        apply_btn.click(
+            fn=apply_fn,
+            inputs=[img_input, style_choice],
+            outputs=img_output
+        )
+    return demo