Update helper_functions.py

helper_functions.py

@@ -259,33 +259,43 @@ def check_validity(query: str, keyword_search: BM25L) -> np.ndarray:
             detail=f"An error occurred during query validity check: {e}",
         )
 
-def is_cheapest(queries: list, request_json: list) -> None:
+def is_cheapest(query: str, request_json: list) -> list:
     """
     Check which product is the cheapest within the same category as
     each input query.
     Args:
-        queries (list): List of input queries
+        query (str): The input query
         request_json (list): List of products
     """
     try:
-        for query in queries:
-            query_categories = [
-                category_map[category]
-                for category in categorizer.predict(query, k=3, threshold=0.5)[0]
-            ]
-
-            min_idx = 0
-            min_price = float('inf')  # Initialize min_price as positive infinity
-            for idx, product in enumerate(request_json):
-                if (
-                    product["Inferred Category"] in query_categories
-                    and product["price"] <= min_price
-                ):
-                    min_idx = idx
-                    min_price = product["price"]  # Update min_price if a cheaper product is found
-            for product in request_json:
-                product["cheapest"] = False  # Reset "cheapest" field for all products
-            request_json[min_idx]["cheapest"] = True  # Mark the cheapest product for the current query
+        query_categories = [
+            category_map[category][0]
+            for category in categorizer.predict(query, k=3, threshold=0.5)[0]
+        ]
+
+        # print(f"Query {query} categories: {query_categories}")
+
+        min_idx = 0
+        min_price = float("inf")  # Initialize min_price as positive infinity
+        for idx, product in enumerate(request_json):
+            if (
+                product["Inferred Category"] in query_categories
+                and product["price"] <= min_price
+            ):
+                min_idx = idx
+                min_price = product[
+                    "price"
+                ]  # Update min_price if a cheaper product is found
+        for product in request_json:
+            product["cheapest"] = False  # Reset "cheapest" field for all products
+
+        # print(f"Cheapest product: {request_json[min_idx]['name']}, Price: {request_json[min_idx]['price']}")
+        request_json[min_idx][
+            "cheapest"
+        ] = True  # Mark the cheapest product for the current query
+
+        # print(request_json)
+        return request_json
     except Exception as e:
         raise HTTPException(
             status_code=500,