Update app.py

app.py

@@ -5,8 +5,10 @@ import numpy as np
 from PIL import Image
 import base64
 import io
+from flask_cors import CORS
 
 app = Flask(__name__)
+CORS(app)  # Allow cross-origin requests from Flutter app
 
 # MediaPipe Pose
 mp_pose = mp.solutions.pose
@@ -19,29 +21,36 @@ def overlay_dress(frame, dress, landmarks):
         h, w, _ = frame.shape
         def to_pixel(lm):
             return int(lm.x * w), int(lm.y * h)
+
         left_shoulder = to_pixel(landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value])
         right_shoulder = to_pixel(landmarks[mp_pose.PoseLandmark.RIGHT_SHOULDER.value])
         left_hip = to_pixel(landmarks[mp_pose.PoseLandmark.LEFT_HIP.value])
         right_hip = to_pixel(landmarks[mp_pose.PoseLandmark.RIGHT_HIP.value])
+
+        # Dress size & position
         dress_width = int(np.linalg.norm(np.array(left_shoulder)-np.array(right_shoulder))*1.8)
         top_shoulder_y = min(left_shoulder[1], right_shoulder[1])
         bottom_hip_y = max(left_hip[1], right_hip[1])
         dress_height = int((bottom_hip_y - top_shoulder_y)*1.2)
+
         center_x = (left_shoulder[0]+right_shoulder[0])//2
-        x1 = max(center_x - dress_width//2,0)
-        y1 = max(top_shoulder_y - 30,0)
+        x1 = max(center_x - dress_width//2, 0)
+        y1 = max(top_shoulder_y - 30, 0)
         x2 = min(x1 + dress_width, w)
         y2 = min(y1 + dress_height, h)
+
+        # Resize and blend dress
         dress_resized = cv2.resize(dress, (x2-x1, y2-y1), interpolation=cv2.INTER_AREA)
-        if dress_resized.shape[2]==4:
-            alpha_s = dress_resized[:,:,3]/255.0
+        if dress_resized.shape[2] == 4:  # has alpha channel
+            alpha_s = dress_resized[:, :, 3] / 255.0
             alpha_l = 1.0 - alpha_s
             for c in range(3):
-                frame[y1:y2,x1:x2,c] = alpha_s*dress_resized[:,:,c] + alpha_l*frame[y1:y2,x1:x2,c]
+                frame[y1:y2, x1:x2, c] = alpha_s*dress_resized[:, :, c] + alpha_l*frame[y1:y2, x1:x2, c]
+
     return frame
 
-# Simple home route for HF Spaces detection
-@app.route("/")
+# Home route for HF Spaces detection
+@app.route("/", methods=["GET"])
 def home():
     return "<h1>Flask API is running!</h1>"
 
@@ -49,8 +58,11 @@ def home():
 @app.route("/tryon", methods=["POST"])
 def tryon():
     try:
-        data = request.json["image"]       # Base64 string of user webcam
-        dress_data = request.json["dress"] # Base64 string of uploaded dress
+        data = request.json.get("image")   # User webcam image in base64
+        dress_data = request.json.get("dress")  # Dress image in base64
+
+        if not data or not dress_data:
+            return jsonify({"error": "Missing image or dress data"}), 400
 
         # Decode user image
         img_bytes = base64.b64decode(data.split(",")[1])
@@ -72,13 +84,14 @@ def tryon():
         if results.pose_landmarks:
             mp_drawing.draw_landmarks(frame, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
 
-        # Encode output
+        # Encode output to base64
         _, buffer = cv2.imencode(".jpg", frame)
         img_base64 = "data:image/jpeg;base64," + base64.b64encode(buffer).decode()
 
         return jsonify({"image": img_base64})
+
     except Exception as e:
-        return jsonify({"error": str(e)})
+        return jsonify({"error": str(e)}), 500
 
-if __name__=="__main__":
+if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000)