Update app.py

app.py

@@ -58,7 +58,7 @@ def detect_and_segment_persons(image, clothes):
 
     bboxes, kpss = detector.detect(img)
     if bboxes.shape[0] == 0:  # Aucun visage détecté
-        return [encode_image_to_base64(image)]
+        return []
 
     height, width, _ = img.shape  # Get image dimensions
 
@@ -80,13 +80,18 @@ def detect_and_segment_persons(image, clothes):
         pil_img = Image.fromarray(person_img[:, :, ::-1])  # BGR -> RGB
 
         # Segment clothing for the detected person
-        segmented_result = segment_clothing(pil_img, clothes)
+        img_rm_background = remove_background(pil_img)
+        segmented_result = segment_clothing(img_rm_background, clothes)
 
         # Combine the segmented images for all persons
         all_segmented_images.extend(segmented_result)
 
     return all_segmented_images
 
+@app.route('/', methods=['GET'])
+def welcome():
+   return "Welcome to Clothing Segmentation API"
+
 @app.route('/api/detect', methods=['POST'])
 def detect():
     try:
@@ -104,4 +109,4 @@ def detect():
         return jsonify({'error': str(e)}), 500
 
 if __name__ == "__main__":
-    app.run(debug=True, host="0.0.0.0", port=7860)
+    app.run(debug=True, host="0.0.0.0", port=7860)