Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -149,18 +149,17 @@ async def predict(query):
|
|
149 |
|
150 |
# Perform hybrid search for each query
|
151 |
# this will result in a dictionary of re-ranked search results for each query
|
152 |
-
results = {}
|
153 |
process_time = time.time()
|
154 |
-
|
155 |
-
|
156 |
-
|
157 |
-
|
158 |
-
|
159 |
-
|
160 |
-
|
161 |
-
|
162 |
-
|
163 |
-
|
164 |
)
|
165 |
|
166 |
process_end_time = time.time()
|
@@ -174,8 +173,8 @@ async def predict(query):
|
|
174 |
# "calculate_interrelations_time": calculate_interrelations_time,
|
175 |
# "process_time": process_time_taken}
|
176 |
|
177 |
-
return results
|
178 |
-
|
179 |
|
180 |
except Exception as e:
|
181 |
error_message = f"An error occurred during processing: {e}"
|
|
|
149 |
|
150 |
# Perform hybrid search for each query
|
151 |
# this will result in a dictionary of re-ranked search results for each query
|
|
|
152 |
process_time = time.time()
|
153 |
+
|
154 |
+
product_list = copy.deepcopy(request_json)
|
155 |
+
keyword_scores = await check_validity(query, keyword_search)
|
156 |
+
semantic_scores = await semantic_search(query, doc_embeddings)
|
157 |
+
hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
|
158 |
+
product_list = is_cheapest(query=query, request_json=product_list)
|
159 |
+
for product in product_list:
|
160 |
+
del product['categoryName'], product['providerName'], product['brandName'], product['productId'], product['key'], product['productOldPrice'], product['imageUrl'], product['currency'], product['providerLogo'], product['productUrl'], product['productRatingCount'], product['productRating'], product['productType']
|
161 |
+
results = rerank_results(
|
162 |
+
request_json=product_list, hybrid_scores=hybrid_scores
|
163 |
)
|
164 |
|
165 |
process_end_time = time.time()
|
|
|
173 |
# "calculate_interrelations_time": calculate_interrelations_time,
|
174 |
# "process_time": process_time_taken}
|
175 |
|
176 |
+
# return results
|
177 |
+
return print_results(results)
|
178 |
|
179 |
except Exception as e:
|
180 |
error_message = f"An error occurred during processing: {e}"
|