Abdul-Ib commited on
Commit
9c7a9cb
1 Parent(s): 0b978f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -13
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
- for query in normalized_query_list:
155
- product_list = copy.deepcopy(request_json)
156
- keyword_scores = await check_validity(query, keyword_search)
157
- semantic_scores = await semantic_search(query, doc_embeddings)
158
- hybrid_scores = hybrid_search(keyword_scores, semantic_scores)
159
- product_list = is_cheapest(query=query, request_json=product_list)
160
- # for product in product_list:
161
- # 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']
162
- results[query] = rerank_results(
163
- request_json=product_list, hybrid_scores=hybrid_scores
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
- # return print_results(results)
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}"