Update app.py

app.py

@@ -50,6 +50,8 @@ async def predict(query):
     normalized_query_list = (
         [normalizer.clean_text(query)]
     )
+
+    normalize_query_time = time.time() - start_time
     
     # Base URL for the search API
     base_url = "https://api.omaline.dev/search/product/search"
@@ -61,6 +63,7 @@ async def predict(query):
     url = f"{base_url}?limit={str(50)}&sortBy=''&{query_string}"
     
     # Make request to the API and handle exceptions
+    request_start_time = time.time()
     try:
         request_json = make_request(url)
     except HTTPException as e:
@@ -68,7 +71,7 @@ async def predict(query):
     except Exception as e:
         return {"error": f"An error occurred while making the request: {e}"}
     request_end_time = time.time()
-    request_time = request_end_time - start_time
+    request_time = request_end_time - request_start_time
     # Translate product representations to English
     normalization_start_time = time.time()
     tasks = []
@@ -154,7 +157,8 @@ async def predict(query):
         process_time_taken = process_end_time - process_time
         time_taken = time.time() - start_time
 
-        hits = {"results": results, "time_taken": time_taken, 
+        hits = {"results": results, "time_taken": time_taken,
+                "normalize_query_time": normalize_query_time,
                 "request_time": request_time, "normalization_time": normalization_time,
                 "translation_time": translation_time, "categorize_time": categorize_time, 
                 "tokenization_time": tokenization_time, "encode_time": encode_time,