refactor: json response title and description generator

app.py

@@ -23,8 +23,3 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
-
-if __name__ == '__main__':
-    import uvicorn
-
-    uvicorn.run(app)

src/api/image_prep_api.py

@@ -5,6 +5,7 @@ author  @ github.com/ishworrsubedii
 """
 import base64
 import os
+import time
 from io import BytesIO
 
 import cv2
@@ -195,12 +196,26 @@ async def remove_background_color_extraction(image: UploadFile = File(...), hex_
 
 @preprocessing_router.post("/title_description_generator")
 async def product_title_description_generator(image: UploadFile = File(...)):
+    start_time = time.time()
     image_bytes = await image.read()
     image = Image.open(BytesIO(image_bytes)).convert("RGB")
     try:
         result = product_listing_obj.gen_title_desc(image=image)
+        print(result)
+
+        title = result.split("Title:")[1].split("Description:")[0]
+        description = result.split("Description:")[1]
+        inference_time = time.time() - start_time
+
+        result = {
+            "code": 200,
+            "title": title,
+            "description": description,
+            "inference_time": inference_time
+        }
 
         return JSONResponse(content=result, status_code=200)
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to process image: {e}")
+        raise HTTPException(status_code=500,
+                            detail=f"Please make sure the image is clear and necklaces are visible{str(e)}")