Update app.py

app.py

@@ -284,24 +284,34 @@ def tryon():
     })
 
     
-def combine_images_with_masks(tops_image, bottoms_image, mask, is_checked_crop, crop_size):
+@app.route('/generate_mask', methods=['POST'])
+def generate_mask():
+    data = request.json
+    base64_image = data.get('image')
+    categorie = data.get('categorie', 'upper_body')
+    
+    # Décodage de l'image
+    human_img = decode_image_from_base64(base64_image).convert("RGB")
+    human_img_resized = human_img.resize((384, 512))
+    
     try:
-        # Logique de combinaison des images de haut et de bas
-        if is_checked_crop:
-            tops_image = tops_image.resize(crop_size)
-            bottoms_image = bottoms_image.resize(crop_size)
-            combined_image = Image.new('RGB', (tops_image.width, tops_image.height))
-            combined_image.paste(tops_image, (0, 0))
-            combined_image.paste(bottoms_image, (0, tops_image.height // 2))
-        else:
-            combined_image = Image.new('RGB', (tops_image.width, tops_image.height))
-            combined_image.paste(tops_image, (0, 0))
-            combined_image.paste(bottoms_image, (0, tops_image.height // 2))
-
-        return combined_image
-
+        # Génération des points clés et du masque
+        keypoints = openpose_model(human_img_resized)
+        model_parse, _ = parsing_model(human_img_resized)
+        mask, _ = get_mask_location('hd', categorie, model_parse, keypoints)
+        
+        # Redimensionnement du masque à la taille d'origine de l'image
+        mask_resized = mask.resize(human_img.size)
+        
+        # Encodage du masque en base64
+        mask_base64 = encode_image_to_base64(mask_resized)
+        
+        return jsonify({
+            'mask_image': mask_base64
+        })
     except Exception as e:
-        raise ValueError(f"Error combining images with masks: {e}")
+        logging.error(f"Error generating mask: {e}")
+        return jsonify({'error': str(e)}), 500
 
 
 if __name__ == "__main__":