add: minor fixes nto

app.py

@@ -29,4 +29,4 @@ app.add_middleware(
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
-)
+)

src/api/mannequin_to_model_api.py

@@ -6,6 +6,7 @@ author  @ github.com/ishworrsubedii
 import base64
 import json
 import os
+import time
 
 import requests
 from fastapi.routing import APIRouter
@@ -34,7 +35,6 @@ def run_mto(input):
 
 def read_return(url):
     try:
-        logger.info(f"Fetching image from URL: {url}")
         res = requests.get(url)
         logger.info("Image fetched successfully")
         return res.content
@@ -50,7 +50,9 @@ async def mto_image(image: UploadFile = File(...), store_name: str = Form(...),
                     body_structure: str = Form(...),
                     skin_complexion: str = Form(...),
                     facial_structure: str = Form(...), ):
+    start_time = time.time()
     try:
+
         logger.info(f"Starting MTO image process for store: {store_name}")
 
         if body_structure == "medium":
@@ -61,7 +63,7 @@ async def mto_image(image: UploadFile = File(...), store_name: str = Form(...),
         logger.info("Source image read successfully")
 
         mannequin_image_url = f"https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/ClothingTryOn/{store_name}/{clothing_category}/{product_id}/{product_id}_{skin_complexion}_{facial_structure}_{body_structure}.webp"
-        logger.info(f"Fetching mannequin image from: {mannequin_image_url}")
+        logger.info(f"Fetching mannequin image")
 
         reference_image_bytes = read_return(mannequin_image_url)
         if reference_image_bytes is None:
@@ -81,8 +83,17 @@ async def mto_image(image: UploadFile = File(...), store_name: str = Form(...),
             logger.error("Face swap process failed")
             return JSONResponse({"error": "Face swap process failed"}, status_code=500)
 
-        logger.info("MTO image process completed successfully")
-        return JSONResponse(content={"output": output}, status_code=200)
+        try:
+            logger.info("MTO image process completed successfully")
+            return JSONResponse(content={
+                "output": str(output['image']),
+                "status": "success",
+                "inferece_time": time.time() - start_time
+            }, status_code=200)
+
+        except Exception as e:
+            logger.error(f"Error converting output to base64: {str(e)}")
+            return JSONResponse({"error": "Error processing output image"}, status_code=500)
 
     except Exception as e:
         logger.error(f"Error in MTO image process: {str(e)}")