test

app.py

@@ -9,6 +9,7 @@ import numpy as np
 from PIL import Image
 from gradio_client import Client, handle_file
 import io
+import base64
 
 app = FastAPI()
 
@@ -166,16 +167,21 @@ async def try_on_api(human_image: UploadFile = File(...), garment_image: UploadF
         # Process the images
         result = process_image(human_path, garment_path)
         
-        # Convert result to bytes
+        # Convert result to base64
         img_byte_arr = io.BytesIO()
         result.save(img_byte_arr, format='PNG')
         img_byte_arr = img_byte_arr.getvalue()
+        base64_image = base64.b64encode(img_byte_arr).decode('utf-8')
         
         # Clean up temporary files
         os.remove(human_path)
         os.remove(garment_path)
         
-        return {"status": "success", "image": img_byte_arr}
+        return {
+            "status": "success",
+            "image": base64_image,
+            "format": "base64"
+        }
     except Exception as e:
         return {"status": "error", "message": str(e)}