Update app.py

app.py

@@ -236,6 +236,10 @@ def decode_image_from_base64(image_data):
     image = Image.open(io.BytesIO(image_data))
     return image
 
+def encode_image_to_base64(image):
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
 @app.route('/tryon', methods=['POST'])
 def tryon():
@@ -257,13 +261,8 @@ def tryon():
 
     output_image, mask_image = start_tryon(human_dict, garment_image, description, use_auto_mask, use_auto_crop, denoise_steps, seed)
 
-    output_bytes = BytesIO()
-    output_image.save(output_bytes, format='PNG')
-    output_bytes = output_bytes.getvalue()
-
-    mask_bytes = BytesIO()
-    mask_image.save(mask_bytes, format='PNG')
-    mask_bytes = mask_bytes.getvalue()
+    output_base64 = encode_image_to_base64(output_image)
+    mask_base64 = encode_image_to_base64(mask_image)
 
     return jsonify({
         'output_image': output_bytes.hex(),