Update app.py

app.py

@@ -13,9 +13,6 @@ mp_pose = mp.solutions.pose
 pose = mp_pose.Pose(static_image_mode=False, min_detection_confidence=0.5)
 mp_drawing = mp.solutions.drawing_utils
 
-# Load dress
-dress_img = cv2.imread("dress.png", cv2.IMREAD_UNCHANGED)
-
 def overlay_dress(frame, dress, landmarks):
     if landmarks is not None:
         h, w, _ = frame.shape
@@ -44,15 +41,32 @@ def overlay_dress(frame, dress, landmarks):
 
 @app.route("/tryon", methods=["POST"])
 def tryon():
-    data = request.json["image"]
-    img_bytes = base64.b64decode(data.split(",")[1])
-    img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
-    frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+    # Expect JSON with keys: 'image' (webcam) and 'dress' (uploaded dress)
+    data = request.json
+    if "image" not in data or "dress" not in data:
+        return jsonify({"error": "Both 'image' and 'dress' are required"}), 400
+
+    # Decode webcam image
+    webcam_bytes = base64.b64decode(data["image"].split(",")[1])
+    webcam_img = Image.open(io.BytesIO(webcam_bytes)).convert("RGB")
+    frame = cv2.cvtColor(np.array(webcam_img), cv2.COLOR_RGB2BGR)
+
+    # Decode uploaded dress image
+    dress_bytes = base64.b64decode(data["dress"].split(",")[1])
+    dress_img = cv2.cvtColor(np.array(Image.open(io.BytesIO(dress_bytes)).convert("RGBA")), cv2.COLOR_RGBA2BGRA)
+
+    # Pose detection
     results = pose.process(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
     landmarks = results.pose_landmarks.landmark if results.pose_landmarks else None
+
+    # Overlay dress
     frame = overlay_dress(frame, dress_img, landmarks)
+
+    # Draw pose landmarks
     if results.pose_landmarks:
         mp_drawing.draw_landmarks(frame, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
+
+    # Encode result
     _, buffer = cv2.imencode(".jpg", frame)
     img_base64 = "data:image/jpeg;base64," + base64.b64encode(buffer).decode()
     return jsonify({"image": img_base64})