Update app.py

app.py

@@ -282,7 +282,7 @@ def tryon():
         'mask_image': mask_base64
     })
 
-
+    
 def combine_images_with_masks(tops_image, bottoms_image, mask, is_checked_crop, crop_size):
     try:
         # Logique de combinaison des images de haut et de bas
@@ -329,10 +329,15 @@ def start_tryon_full_body(tops_image, bottoms_image, model_parse_tops, model_par
             human_img = human_img_orig.resize((768, 1024))
 
         if is_checked:
-            keypoints = openpose_model(human_img.resize((384, 512)))
-            model_parse, _ = parsing_model(human_img.resize((384, 512)))
-            mask, mask_gray = get_mask_location('hd', "full_body", model_parse, keypoints)
-            mask = mask.resize((768, 1024))
+            try:
+                keypoints = openpose_model(human_img.resize((384, 512)))
+                model_parse, _ = parsing_model(human_img.resize((384, 512)))
+                if 'bodies' not in keypoints or 'subset' not in keypoints['bodies'] or len(keypoints['bodies']['subset']) == 0:
+                    raise ValueError("No pose keypoints found in the model output.")
+                mask, mask_gray = get_mask_location('hd', "full_body", model_parse, keypoints)
+                mask = mask.resize((768, 1024))
+            except Exception as e:
+                raise ValueError(f"Error processing keypoints or mask: {e}")
         else:
             mask = pil_to_binary_mask(model_parse_tops.convert("RGB").resize((768, 1024)))
         mask_gray = (1 - transforms.ToTensor()(mask)) * tensor_transfrom(human_img)
@@ -425,6 +430,9 @@ def start_tryon_full_body(tops_image, bottoms_image, model_parse_tops, model_par
 
 
 
+
+
+
 @app.route('/tryon-full', methods=['POST'])
 def tryon_full():
     try: