Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from PIL import Image
 import os
 
+# Load all hairstyle images from folder
 def load_hairstyles():
     folder = "hairstyles"
     if not os.path.exists(folder):
@@ -13,46 +14,46 @@ def load_hairstyles():
 
 hairstyles = load_hairstyles()
 
-def apply_hairstyle(user_img, style_index, x_offset, y_offset, scale):
+# Automatically apply hairstyle to uploaded image
+def apply_hairstyle(user_img, style_index):
     if user_img is None or not hairstyles:
         return None
 
     user_img = user_img.convert("RGBA")
     base_w, base_h = user_img.size
-
     hairstyle = hairstyles[style_index]
 
-    # Resize the hairstyle based on scale
-    new_size = (int(base_w * scale), int(hairstyle.height * (base_w * scale / hairstyle.width)))
-    hairstyle = hairstyle.resize(new_size)
+    # Resize hairstyle to match width of head image
+    scale_factor = 0.9  # Adjust if needed
+    new_width = int(base_w * scale_factor)
+    new_height = int(hairstyle.height * (new_width / hairstyle.width))
+    resized_hairstyle = hairstyle.resize((new_width, new_height))
+
+    # Automatically center it near the top
+    paste_x = (base_w - new_width) // 2
+    paste_y = int(base_h * 0.05)  # Adjust this offset if needed
 
-    # Create a blank transparent image to position the hairstyle
     composite = Image.new("RGBA", user_img.size)
-    paste_x = int((base_w - new_size[0]) / 2 + x_offset)
-    paste_y = int(y_offset)
-    composite.paste(hairstyle, (paste_x, paste_y), hairstyle)
+    composite.paste(resized_hairstyle, (paste_x, paste_y), resized_hairstyle)
 
-    # Overlay it
-    result = Image.alpha_composite(user_img, composite)
-    return result.convert("RGB")
+    final = Image.alpha_composite(user_img, composite)
+    return final.convert("RGB")
 
+# Gradio Interface (no sliders shown to user)
 with gr.Blocks() as demo:
-    gr.Markdown("## 💇 Salon Virtual Hairstyle Try-On (Adjustable)")
+    gr.Markdown("## 💇 Salon Virtual Hairstyle Try-On (Auto Fit – No Adjustment Needed)")
 
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(type="pil", label="📷 Upload an Image")
+            image_input = gr.Image(type="pil", label="📷 Upload Your Image")
             style_slider = gr.Slider(0, max(len(hairstyles)-1, 0), step=1, label="🎨 Select Hairstyle")
-            x_offset = gr.Slider(-200, 200, value=0, step=1, label="⬅️➡️ Move Left / Right")
-            y_offset = gr.Slider(-200, 200, value=0, step=1, label="⬆️⬇️ Move Up / Down")
-            scale = gr.Slider(0.3, 2.0, value=1.0, step=0.05, label="📏 Scale Hairstyle")
             apply_btn = gr.Button("✨ Apply Hairstyle")
         with gr.Column():
             result_output = gr.Image(label="🔍 Result Preview")
 
     apply_btn.click(
         fn=apply_hairstyle,
-        inputs=[image_input, style_slider, x_offset, y_offset, scale],
+        inputs=[image_input, style_slider],
         outputs=result_output
     )