Update app.py

app.py

@@ -314,6 +314,33 @@ def tryon():
         'mask_image': mask_base64
     })
 
+@app.route('/get_mask', methods=['POST'])
+def get_mask():
+    try:
+        # Récupérer l'image du corps à partir de la requête
+        img_file = request.files['image']
+        img = Image.open(img_file.stream).convert("RGB").resize((384, 512))
+        categorie = request.form.get('categorie', 'upper_body')  # Paramètre avec valeur par défaut
+
+        # Appliquer la détection des points clés
+        keypoints = openpose_model(img)  # Utilise votre modèle
+        model_parse, _ = parsing_model(img)  # Utilise votre modèle
+
+        # Obtenir le masque
+        mask, mask_gray = get_mask_location('hd', categorie , model_parse, keypoints)
+        
+        # Convertir le masque en image (si nécessaire)
+        mask_gray = (1 - transforms.ToTensor()(mask_gray)) * tensor_transfrom(img)
+        mask_gray = to_pil_image((mask_gray + 1.0) / 2.0)
+
+        # Convertir l'image en base64 si besoin pour le retour
+        img_byte_arr = io.BytesIO()
+        mask_gray.save(img_byte_arr, format='PNG')
+        img_byte_arr.seek(0)
+        return jsonify({'mask': img_byte_arr.getvalue().decode('latin1')})  # Utiliser une méthode appropriée pour l'encodage
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
 
 # Route index
 @app.route('/', methods=['GET'])